lot

untitled/generation.py

@@ -3,19 +3,24 @@ import random
 from untitled import content
 
 
+def generate_trait_sentence(traits):
+    article = "An" if traits["size"][0] in "aeiou" else "A"
+    description = f"{article} {traits["size"]} {traits["color"]} kitten with {traits["eyes"]} eyes who {traits["personality"]}"
+    return description
+
+
 def generate_cat_choice(personality=None):
     size = random.choice(content.CAT_SIZES)
     color = random.choice(content.CAT_COLORS)
     eyes = random.choice(content.CAT_EYE_COLORS)
     personality = personality or random.choice(content.CAT_PERSONALITIES)
-    article = "An" if size[0] in "aeiou" else "A"
-    description = f"{article} {size} {color} kitten with {eyes} eyes who {personality}"
-    return description, {
+    traits = {
         "size": size,
         "color": color,
         "eyes": eyes,
         "personality": personality,
     }
+    return generate_trait_sentence(traits), traits
 
 
 def generate_cat_choices(n=5):