Expand picker: 24 animals (lookalike-grouped) + 16 places

- Animals +12: mouse, wolf, bear, pig, cow, pony, chicken, otter, octopus,
bee, ladybug, dragon — reordered into lookalike groups (pets · canines/bears ·
big cats · farm · water/amphibian · bugs · mythical)
- Places +6: canyon, castle, theme park, tent, disco ball, circus
- Kept ANIMAL_EMOJIS/PLACE_EMOJIS (app.py picker) and ANIMAL_MAP/PLACE_MAP
(image_generator prompts) in identical key + order — verified no fallbacks

The style LoRA generalizes to the new animals/places (trained on the original
12×10 but it's a style adapter, so new subjects still render in the cozy look).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

app.py

@@ -18,8 +18,18 @@ from image_generator import generate_reward_image
 
 ANSWERS_PER_LEVEL = 5
 
-ANIMAL_EMOJIS = ["🐶", "🐱", "🐰", "🦊", "🐼", "🐨", "🦁", "🐯", "🐸", "🐧", "🦋", "🦄"]
-PLACE_EMOJIS  = ["🌊", "🏔️", "🌸", "🌈", "🌙", "⭐", "🌴", "🏡", "🌺", "🍄"]
+# Animals — ordered by lookalike groups (pets · canines/bears · big cats · farm ·
+# water/amphibian · bugs · mythical) so similar creatures sit together in the picker.
+ANIMAL_EMOJIS = [
+    "🐶", "🐱", "🐰", "🐭", "🦊", "🐺",
+    "🐻", "🐼", "🐨", "🦁", "🐯", "🐷",
+    "🐮", "🐴", "🐔", "🐸", "🦦", "🐧",
+    "🐙", "🦋", "🐝", "🐞", "🦄", "🐉",
+]
+PLACE_EMOJIS  = [
+    "🌊", "🏔️", "🌸", "🌈", "🌙", "⭐", "🌴", "🏡", "🌺", "🍄",
+    "🏜️", "🏰", "🎢", "⛺", "🪩", "🎪",
+]
 
 MATH_LEVEL_NAMES = {1: "Additions", 2: "Subtractions", 3: "Multiplications", 4: "Mix", 5: "Mix +"}
 

image_generator.py

@@ -163,13 +163,19 @@ else:
 # Emoji → descriptive text maps (fed into FLUX prompt)
 # ---------------------------------------------------------------------------
 
+# Keep keys + order in sync with ANIMAL_EMOJIS in app.py (grouped by lookalike).
 ANIMAL_MAP: dict[str, str] = {
-    "🐶": "puppy",        "🐱": "kitten",       "🐰": "bunny",
-    "🦊": "baby fox",     "🐼": "baby panda",   "🐨": "baby koala",
-    "🦁": "baby lion",    "🐯": "baby tiger",   "🐸": "baby frog",
-    "🐧": "baby penguin", "🦋": "butterfly",    "🦄": "unicorn",
+    "🐶": "puppy",         "🐱": "kitten",        "🐰": "bunny",
+    "🐭": "baby mouse",    "🦊": "baby fox",      "🐺": "baby wolf",
+    "🐻": "baby bear",     "🐼": "baby panda",    "🐨": "baby koala",
+    "🦁": "baby lion",     "🐯": "baby tiger",    "🐷": "baby pig",
+    "🐮": "baby cow",      "🐴": "pony",          "🐔": "chicken",
+    "🐸": "baby frog",     "🦦": "baby otter",    "🐧": "baby penguin",
+    "🐙": "baby octopus",  "🦋": "butterfly",     "🐝": "bee",
+    "🐞": "ladybug",       "🦄": "unicorn",       "🐉": "baby dragon",
 }
 
+# Keep keys + order in sync with PLACE_EMOJIS in app.py.
 PLACE_MAP: dict[str, str] = {
     "🌊": "on a sunny beach with ocean waves",
     "🏔️": "on a snowy mountain top",
@@ -181,6 +187,12 @@ PLACE_MAP: dict[str, str] = {
     "🏡": "in a cosy cottage garden",
     "🌺": "in a field of tropical flowers",
     "🍄": "in an enchanted mushroom forest",
+    "🏜️": "in a red rock canyon",
+    "🏰": "in a fairytale castle",
+    "🎢": "at a fun theme park",
+    "⛺": "in a cosy camping tent",
+    "🪩": "under a sparkling disco ball",
+    "🎪": "at a colourful circus",
 }
 
 # Shared style suffix — must stay in sync with STYLE in scripts/generate_dataset.py.