Dataset: FLUX.2-dev + app-aligned prompts with multi-animal/place combos

image_generator.py:
- Extract build_subject(animals, places) → "A cute {animals} {places}"
shared core; build_prompt now = build_subject + ", " + NUMZOO_STYLE
- Add _join() helper for the 1–3 phrase joining

scripts/generate_dataset.py:
- Switch model to FLUX.2-dev (via fal-ai provider, HF Pro credits)
- Rebuild prompts programmatically from the app's exact ANIMAL_MAP /
PLACE_MAP using the shared build_subject — captions are byte-identical
in structure to live app prompts, just with a rich scene-detail clause
- 54 curated scenes: all 12 animals solo, all 10 places, plus deliberate
2–3 animal and 2–3 place combinations mirroring multi-emoji selections

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

image_generator.py

@@ -174,30 +174,27 @@ NUMZOO_STYLE = (
 # Prompt builder
 # ---------------------------------------------------------------------------
 
-def build_prompt(animals: list[str], places: list[str]) -> str:
-    # Use ALL selected animals and places (up to 3 each)
+def _join(parts: list[str]) -> str:
+    """Join 1–3 phrases: 'a', 'a and b', 'a, b and c'."""
+    if len(parts) == 1:
+        return parts[0]
+    if len(parts) == 2:
+        return f"{parts[0]} and {parts[1]}"
+    return f"{parts[0]}, {parts[1]} and {parts[2]}"
+
+
+def build_subject(animals: list[str], places: list[str]) -> str:
+    """The 'A cute {animals} {places}' core — SHARED with scripts/generate_dataset.py
+    so training captions and live prompts use the exact same structure & vocabulary."""
     animal_list = animals[:3] if animals else [random.choice(list(ANIMAL_MAP))]
     place_list  = places[:3]  if places  else [random.choice(list(PLACE_MAP))]
+    animal_text = _join([ANIMAL_MAP.get(a, "bunny") for a in animal_list])
+    place_text  = _join([PLACE_MAP.get(p, "in a magical garden") for p in place_list])
+    return f"A cute {animal_text} {place_text}"
 
-    # Build animal text: "puppy", "puppy and bunny", "puppy, bunny and kitten"
-    animal_names = [ANIMAL_MAP.get(a, "bunny") for a in animal_list]
-    if len(animal_names) == 1:
-        animal_text = animal_names[0]
-    elif len(animal_names) == 2:
-        animal_text = f"{animal_names[0]} and {animal_names[1]}"
-    else:
-        animal_text = f"{animal_names[0]}, {animal_names[1]} and {animal_names[2]}"
-
-    # Build place text: "in a garden", "in a garden and under a rainbow", ...
-    place_texts = [PLACE_MAP.get(p, "in a magical garden") for p in place_list]
-    if len(place_texts) == 1:
-        place_text = place_texts[0]
-    elif len(place_texts) == 2:
-        place_text = f"{place_texts[0]} and {place_texts[1]}"
-    else:
-        place_text = f"{place_texts[0]}, {place_texts[1]} and {place_texts[2]}"
 
-    return f"A cute {animal_text} {place_text}, {NUMZOO_STYLE}"
+def build_prompt(animals: list[str], places: list[str]) -> str:
+    return f"{build_subject(animals, places)}, {NUMZOO_STYLE}"
 
 # ---------------------------------------------------------------------------
 # Pipeline loader (cached globally — survives between ZeroGPU calls)

scripts/generate_dataset.py

@@ -1,33 +1,33 @@
 """
 NumZoo training dataset generator.
 
-Generates 80 images matching the NumZoo aesthetic using FLUX.1-dev via the
-HuggingFace Inference API (fal-ai provider — best quality, free credits on signup).
+Generates LoRA training images matching the NumZoo aesthetic using FLUX.2-dev
+via the HuggingFace Inference API (fal-ai provider, billed to your HF Pro credits).
 
-- Kawaii chibi animals with big sparkling eyes
-- Rich scene backgrounds (no white background)
-- Warm pastel palette, fairy lights, cozy props
-- Square 1024×1024 output to match the app layout
+Alignment with the live app is guaranteed by construction:
+  - The "A cute {animals} {places}" prefix is built with image_generator.build_subject
+    (the SAME function the app uses), from the app's exact ANIMAL_MAP / PLACE_MAP.
+  - The NUMZOO_STYLE suffix is imported from image_generator.
+  - Scenes deliberately mix 1–3 animals and 1–3 places, exactly like the app does
+    when the player selects multiple emojis.
+So every caption looks like a real app prompt, plus a rich scene detail clause.
 
 Output: training/image_001.jpg + training/image_001.txt (caption)
 
 Requirements:
-    pip install huggingface_hub pillow python-dotenv
+    ~/miniforge3/bin/pip install huggingface_hub pillow python-dotenv
 
 Setup (HF Pro — just your existing token):
-    1. Get your HF token at https://huggingface.co/settings/tokens
-       (fine-grained, with "Make calls to Inference Providers" permission)
-    2. Add to .env:
-         HF_TOKEN=hf_...
-
-    With HF Pro your token already has credits on fal-ai, replicate, together etc.
-    No separate provider account needed.
+    1. Accept the FLUX.2-dev license: https://huggingface.co/black-forest-labs/FLUX.2-dev
+    2. Get an HF token (fine-grained, "Make calls to Inference Providers" permission)
+       https://huggingface.co/settings/tokens
+    3. Add to .env:  HF_TOKEN=hf_...
 
 Usage:
-    python scripts/generate_dataset.py              # all 80 via fal-ai (FLUX.1-dev)
-    python scripts/generate_dataset.py --count 5   # first 5 only (test run)
-    python scripts/generate_dataset.py --start 41  # resume from image 41
-    python scripts/generate_dataset.py --provider hf-inference  # HF native (schnell)
+    ~/miniforge3/bin/python3 scripts/generate_dataset.py            # all scenes
+    ~/miniforge3/bin/python3 scripts/generate_dataset.py --count 5  # first 5 (test run)
+    ~/miniforge3/bin/python3 scripts/generate_dataset.py --start 20 # resume from #20
+    ~/miniforge3/bin/python3 scripts/generate_dataset.py --dry-run  # preview prompts
 """
 
 import os
@@ -43,206 +43,112 @@ try:
 except ImportError:
     pass  # dotenv optional — can also export GEMINI_API_KEY manually
 
-# Import the canonical style string from image_generator so training captions
-# and live prompts are always identical.
+# Reuse the app's exact prompt builder + vocabulary so training captions and live
+# prompts share the same structure ("A cute {animals} {places}") and emoji mappings.
 sys.path.insert(0, str(Path(__file__).parent.parent))
-from image_generator import NUMZOO_STYLE as STYLE  # noqa: E402
+from image_generator import (  # noqa: E402
+    build_subject,
+    NUMZOO_STYLE as STYLE,
+    ANIMAL_MAP,
+    PLACE_MAP,
+)
 
 # ---------------------------------------------------------------------------
-# 80 prompts — 8 scene categories × 10 animals
-# STYLE is appended to every prompt so the LoRA learns it as a trigger
+# Scenes: (animal emojis, place emojis, scene-detail clause)
+# Animals/places use the app's exact emoji keys. The "A cute {animals} {places}"
+# prefix is built by image_generator.build_subject; the detail adds rich props,
+# activity and lighting for the cozy aesthetic. Deliberately mixes 1–3 animals
+# and 1–3 places to mirror multi-emoji selections in the app.
 # ---------------------------------------------------------------------------
 
-PROMPTS: list[tuple[str, str]] = [
-    # ── 1. Cozy cabin interior ──────────────────────────────────────────────
-    ("cozy_cabin_01", f"a fluffy bunny curled on an armchair by a stone fireplace inside a wooden cabin, "
-        f"pastel star blanket, string lights, hot cocoa on a tree-stump table, warm amber glow, {STYLE}"),
-    ("cozy_cabin_02", f"a sleepy kitten reading a tiny book in a cozy log cabin, "
-        f"fairy lights strung above wooden shelves with colorful jars, fireplace crackling, "
-        f"pastel pink and lavender tones, {STYLE}"),
-    ("cozy_cabin_03", f"a baby bear in a rainbow scarf sitting at a round wooden table with pancakes "
-        f"inside a cabin, soft morning light through an arched window, {STYLE}"),
-    ("cozy_cabin_04", f"a hedgehog wrapped in a cloud-print blanket watching raindrops on a cabin window, "
-        f"string lights reflected in the glass, soft purple tones, {STYLE}"),
-    ("cozy_cabin_05", f"a baby fox and a bunny sharing hot tea by a fireplace in a treehouse cabin, "
-        f"wooden bookshelves, pastel mugs, twinkle lights, {STYLE}"),
-    ("cozy_cabin_06", f"a tiny owl perched on a branch reading a glowing book inside a lantern-lit cabin, "
-        f"warm golden light, mossy windowsill with flowers outside, {STYLE}"),
-    ("cozy_cabin_07", f"a baby deer sitting on a plaid cushion inside a snug wooden cabin, "
-        f"potted cactus and tiny succulents on shelves, fairy lights overhead, {STYLE}"),
-    ("cozy_cabin_08", f"a raccoon stirring a sparkling potion in a cozy cabin kitchen, "
-        f"mason jars of colorful ingredients, star-shaped window, warm light, {STYLE}"),
-    ("cozy_cabin_09", f"a baby penguin in a striped scarf napping on a pastel sofa next to a glowing fireplace, "
-        f"snow visible through the cabin window, {STYLE}"),
-    ("cozy_cabin_10", f"a squirrel decorating a tiny pine tree inside a cabin with star ornaments and ribbon, "
-        f"fairy lights everywhere, soft pink tones, {STYLE}"),
-
-    # ── 2. Enchanted forest — day ───────────────────────────────────────────
-    ("forest_day_01", f"a bunny in a daisy field inside an enchanted forest, "
-        f"sunbeams through tall trees, floating sparkles, colorful mushrooms, {STYLE}"),
-    ("forest_day_02", f"a kitten playing with a butterfly on a mossy log in a sunlit magical forest, "
-        f"wildflowers in pastel pink and lilac, {STYLE}"),
-    ("forest_day_03", f"a baby bear discovering a tiny glowing door in a tree trunk in an enchanted forest, "
-        f"soft golden afternoon light, flower garlands, {STYLE}"),
-    ("forest_day_04", f"a hedgehog gathering strawberries in a forest clearing, "
-        f"pastel checkered picnic blanket, wicker basket, soft dappled sunlight, {STYLE}"),
-    ("forest_day_05", f"a baby fox sitting on a giant mushroom in a magical forest, "
-        f"fairy dust floating around, colorful birds in branches above, {STYLE}"),
-    ("forest_day_06", f"an owl in a graduation cap perched on a tree branch in a sunlit forest, "
-        f"hanging star lanterns, flowers blooming below, {STYLE}"),
-    ("forest_day_07", f"a baby deer drinking from a sparkling stream in a lush forest, "
-        f"rainbow trout visible in the water, cherry blossoms drifting, {STYLE}"),
-    ("forest_day_08", f"a raccoon painting a tiny canvas in a forest studio made of logs and branches, "
-        f"colorful paint pots, sunlight streaming in, {STYLE}"),
-    ("forest_day_09", f"a baby penguin waddling through a magical spring forest with tulips taller than it, "
-        f"pastel butterflies, {STYLE}"),
-    ("forest_day_10", f"a squirrel and a bunny having a tea party on a giant leaf in the forest, "
-        f"acorn cups, mushroom chairs, sunlit clearing, {STYLE}"),
-
-    # ── 3. Magical forest — night / campfire ────────────────────────────────
-    ("forest_night_01", f"a bunny and a kitten roasting marshmallows over a small campfire in a night forest, "
-        f"crescent moon, fireflies, string lights in the trees, {STYLE}"),
-    ("forest_night_02", f"a baby bear stargazing in a moonlit forest clearing, lying on a cloud-print blanket, "
-        f"glowing constellations above, lantern beside, {STYLE}"),
-    ("forest_night_03", f"a baby fox with a tiny lantern walking a forest path at night, "
-        f"glowing mushrooms lining the path, crescent moon, {STYLE}"),
-    ("forest_night_04", f"a hedgehog sleeping under a luminescent mushroom at night in the forest, "
-        f"fireflies around, purple and blue tones, soft moonlight, {STYLE}"),
-    ("forest_night_05", f"an owl sitting on a branch under the stars playing a tiny ukulele for forest friends "
-        f"gathered below, night forest, fairy lights, {STYLE}"),
-    ("forest_night_06", f"a baby deer watching shooting stars from a mossy hilltop at night, "
-        f"forest silhouette below, soft purple sky, sparkles, {STYLE}"),
-    ("forest_night_07", f"a raccoon cooking s'mores at a campfire in a night forest, "
-        f"camping tent with fairy lights, moon and stars above, {STYLE}"),
-    ("forest_night_08", f"a baby penguin looking at its reflection in a glowing moonlit pond in a night forest, "
-        f"lily pads with tiny lights, {STYLE}"),
-    ("forest_night_09", f"a squirrel telling stories around a campfire to bunny and kitten, night forest, "
-        f"warm firelight, stars through the tree canopy, {STYLE}"),
-    ("forest_night_10", f"a kitten catching fireflies in a jar at the edge of a glowing forest at dusk, "
-        f"silhouette trees, warm golden light, {STYLE}"),
-
-    # ── 4. Treehouse ────────────────────────────────────────────────────────
-    ("treehouse_01", f"a bunny inside a cozy treehouse at sunset, rope bridge entrance, "
-        f"flower window box, warm light glowing inside, {STYLE}"),
-    ("treehouse_02", f"a kitten at the door of a treehouse with a welcome sign, "
-        f"string lights along the railing, forest view at golden hour, {STYLE}"),
-    ("treehouse_03", f"a baby bear and a hedgehog playing board games on the treehouse porch, "
-        f"paper lanterns, leafy canopy above, {STYLE}"),
-    ("treehouse_04", f"a baby fox gazing at the stars through a treehouse telescope at night, "
-        f"spiral staircase, crescent moon, {STYLE}"),
-    ("treehouse_05", f"an owl's cozy treehouse library with tiny glowing books and a reading nook, "
-        f"ivy covered walls, daytime, {STYLE}"),
-    ("treehouse_06", f"a baby bear watering flower boxes on a treehouse balcony, "
-        f"bees and butterflies, golden afternoon light, leafy forest below, {STYLE}"),
-    ("treehouse_07", f"a hedgehog and a squirrel stringing fairy lights around a treehouse at dusk, "
-        f"warm glow, birds watching from nearby branches, {STYLE}"),
-    ("treehouse_08", f"a baby fox sliding down a spiral slide from a treehouse into a pile of leaves, "
-        f"autumn forest, rope ladder, golden light, {STYLE}"),
-    ("treehouse_09", f"a baby deer reading a glowing map inside a treehouse at night, "
-        f"star-shaped windows, moonlight, cozy lanterns, {STYLE}"),
-    ("treehouse_10", f"a raccoon and a bunny cooking soup in a tiny treehouse kitchen, "
-        f"steam rising, herbs hanging, sunlight through small windows, {STYLE}"),
-
-    # ── 5. Garden / meadow — day ────────────────────────────────────────────
-    ("garden_01", f"a bunny watering oversized tulips in a magical garden, "
-        f"pastel blue watering can, butterflies, soft morning light, {STYLE}"),
-    ("garden_02", f"a kitten napping in a flower hammock in a sunny garden, "
-        f"roses and lavender around, bees buzzing, {STYLE}"),
-    ("garden_03", f"a baby bear picking giant strawberries in a garden, "
-        f"ladybugs on leaves, warm afternoon sun, picket fence, {STYLE}"),
-    ("garden_04", f"a hedgehog rolling through a field of sunflowers, "
-        f"pastel sky, birds above, soft breeze, {STYLE}"),
-    ("garden_05", f"a baby fox blowing dandelion seeds in a meadow, "
-        f"seeds glowing like stars, cherry trees in background, {STYLE}"),
-    ("garden_06", f"a baby deer in a rain boots splashing in a puddle in a garden after rain, "
-        f"rainbow in the sky, wet flowers, {STYLE}"),
-    ("garden_07", f"a raccoon painting a garden fence with rainbow stripes, "
-        f"paint buckets, butterflies landing on fresh paint, {STYLE}"),
-    ("garden_08", f"a baby penguin planting seeds in a tiny garden plot, "
-        f"little labeled plant markers, pastel gloves and apron, {STYLE}"),
-    ("garden_09", f"a squirrel collecting colorful autumn leaves in a basket in a golden meadow, "
-        f"maple trees, soft warm light, {STYLE}"),
-    ("garden_10", f"a bunny and kitten flying a star-shaped kite in an open meadow, "
-        f"puffy clouds, wildflowers below, {STYLE}"),
-
-    # ── 6. Beach / waterfront ───────────────────────────────────────────────
-    ("beach_01", f"a bunny building a sand castle on a pastel beach at sunset, "
-        f"sea shells, gentle waves, pink and orange sky, {STYLE}"),
-    ("beach_02", f"a kitten surfing a tiny wave on a pastel board, "
-        f"tropical beach, palm trees, clear turquoise water, {STYLE}"),
-    ("beach_03", f"a baby bear making friends with a crab on the beach, "
-        f"rock pools with anemones, soft summer light, {STYLE}"),
-    ("beach_04", f"a baby fox asleep in a hammock strung between two palm trees on a beach, "
-        f"coconuts, ocean at dusk, {STYLE}"),
-    ("beach_05", f"a baby penguin waddling along a sandy beach collecting glowing starfish at night, "
-        f"bioluminescent waves, crescent moon, {STYLE}"),
-    ("beach_06", f"a bunny and a hedgehog sharing a watermelon slice on a colorful beach towel, "
-        f"sunny beach, parasol, gentle waves, {STYLE}"),
-    ("beach_07", f"a baby fox building a sand boat on the beach, "
-        f"pebble sail, seagulls, golden sunset, {STYLE}"),
-    ("beach_08", f"a baby bear snorkeling in a clear lagoon, colorful fish and coral below, "
-        f"tropical beach, bright midday light, {STYLE}"),
-    ("beach_09", f"a kitten watching the sunset from a dock over calm ocean water, "
-        f"orange and pink sky, lantern beside, soft waves, {STYLE}"),
-    ("beach_10", f"a baby deer and a squirrel collecting shells in tide pools at low tide, "
-        f"pastel sunset, starfish, {STYLE}"),
-
-    # ── 7. Snowy winter ──────────────────────────────────────────────────────
-    ("snow_01", f"a bunny building a snowman in a snowy forest clearing, "
-        f"carrot nose, scarf, snowflakes falling, soft pink sky, {STYLE}"),
-    ("snow_02", f"a kitten ice skating on a frozen pond in a winter forest, "
-        f"snowflakes, pine trees with snow caps, fairy lights on branches, {STYLE}"),
-    ("snow_03", f"a baby bear sledding down a snowy hill at night, "
-        f"moonlight, sparkling snow, cozy village lights below, {STYLE}"),
-    ("snow_04", f"a hedgehog warming up by a bonfire in the snow, "
-        f"hot chocolate in tiny paws, snowflakes floating, pine forest, {STYLE}"),
-    ("snow_05", f"a baby fox wrapped in a giant scarf making snow angels in a winter field, "
-        f"snowflakes, soft blue and white tones, {STYLE}"),
-    ("snow_06", f"an owl delivering a tiny letter through falling snow to a cabin, "
-        f"warm window glow, snow-covered trees, {STYLE}"),
-    ("snow_07", f"a baby deer with snowflakes on its nose in a quiet snowy forest, "
-        f"bare tree branches with icicles, soft grey-blue light, {STYLE}"),
-    ("snow_08", f"a raccoon and a squirrel having a snowball fight in a park, "
-        f"snow-covered benches, winter lanterns, {STYLE}"),
-    ("snow_09", f"a baby penguin sliding down an icy path to the frozen ocean, "
-        f"auroras in the sky, soft blue and green glow, {STYLE}"),
-    ("snow_10", f"a bunny and kitten sharing a pastel umbrella in gently falling snow, "
-        f"fairy lights on trees, warm glow from a café window, {STYLE}"),
-
-    # ── 8. Autumn / rainy day ────────────────────────────────────────────────
-    ("autumn_01", f"a bunny jumping in a pile of colorful autumn leaves in a park, "
-        f"orange and red maples, soft afternoon sun, {STYLE}"),
-    ("autumn_02", f"a kitten at a rainy window with a cup of tea, "
-        f"raindrops on glass, cozy sweater, autumn forest visible outside, {STYLE}"),
-    ("autumn_03", f"a baby bear in a pumpkin patch at golden hour, "
-        f"scarecrow, autumn leaves drifting, {STYLE}"),
-    ("autumn_04", f"a hedgehog foraging mushrooms in an autumn forest, "
-        f"golden light through orange leaves, red berries, {STYLE}"),
-    ("autumn_05", f"a baby fox jumping through autumn leaves in a red-orange forest, "
-        f"bokeh light, warm amber tones, {STYLE}"),
-    ("autumn_06", f"an owl in a cozy raincoat standing under a colorful mushroom in the rain, "
-        f"puddles reflecting fairy lights, {STYLE}"),
-    ("autumn_07", f"a baby deer watching the first autumn rain from under a giant leaf, "
-        f"golden forest background, raindrops sparkle, {STYLE}"),
-    ("autumn_08", f"a raccoon selling hot apple cider from a tiny wooden cart in an autumn market, "
-        f"string lights, orange and red leaves, {STYLE}"),
-    ("autumn_09", f"a squirrel cozy in a hollow tree watching autumn rain, "
-        f"acorn collection visible inside, warm golden glow, {STYLE}"),
-    ("autumn_10", f"a bunny and a baby fox under a giant colorful umbrella in a rainy autumn street, "
-        f"café lights, reflections in puddles, {STYLE}"),
+SCENES: list[tuple[list[str], list[str], str]] = [
+    # ── Solo animal, single place — covers all 12 animals + all 10 places ──
+    (["🐰"], ["🍄"], "sitting on a polka-dot toadstool, fireflies and floating spores drifting around, soft lantern glow"),
+    (["🐱"], ["🌊"], "building a tiny sandcastle with a bucket and shells, gentle waves lapping, warm sunset sky"),
+    (["🐶"], ["🏡"], "napping in a flower-filled wheelbarrow, watering can and butterflies nearby, golden afternoon light"),
+    (["🦊"], ["⭐"], "curled up on a fluffy cloud cradling a tiny glowing star, glittering night sky"),
+    (["🐼"], ["🌸"], "nibbling a dango skewer as petals fall, paper lanterns strung above, soft pink light"),
+    (["🐨"], ["🌴"], "hugging a palm trunk with a coconut drink, striped hammock and a parrot, turquoise sea behind"),
+    (["🦁"], ["🌈"], "wearing a tiny crown at the end of a rainbow, pastel clouds and floating sparkles"),
+    (["🐯"], ["🏔️"], "bundled in a knitted scarf on a snowy peak planting a tiny flag, sparkling snow and faint aurora"),
+    (["🐸"], ["🌺"], "perched on a giant hibiscus bloom, dewdrops glistening, big tropical leaves and warm bokeh"),
+    (["🐧"], ["🌙"], "sitting on the curve of a glowing crescent moon in a knitted hat, scattered twinkling stars"),
+    (["🦋"], ["🌸"], "fluttering through cherry blossoms trailing sparkles, pastel petals swirling in the breeze"),
+    (["🦄"], ["⭐"], "galloping across a starry sky, rainbow mane glowing, a trail of sparkles behind"),
+    (["🐶"], ["🍄"], "exploring beneath a giant mushroom with a tiny lantern, glowing toadstools and soft moss"),
+    (["🐱"], ["🌙"], "curled asleep on a crescent moon wearing a nightcap, twinkling stars all around"),
+    (["🐰"], ["🌊"], "splashing in shallow waves beside a starfish friend, beach pail and spade, pink sunset"),
+    (["🐼"], ["🏡"], "tending a vegetable patch in a straw hat, bees and tall sunflowers, warm sun"),
+    (["🐧"], ["🏔️"], "sliding down a snowy slope on its belly, scarf flying, sparkling powder snow"),
+    (["🦊"], ["🌴"], "lounging in a hammock between two palms with sunglasses, coconuts and calm ocean"),
+    (["🦁"], ["🌺"], "snoozing in a field of tropical flowers, a butterfly on its nose, dappled golden light"),
+    (["🐯"], ["🌸"], "chasing falling cherry petals, paper lanterns above, soft pink and lilac tones"),
+    (["🐸"], ["🍄"], "playing a tiny flute on a lily pad among glowing mushrooms, fireflies and reeds"),
+    (["🦄"], ["🌈"], "standing proudly under a rainbow, flower garland around its neck, pastel clouds"),
+
+    # ── Two animals, single place ──
+    (["🐰", "🐱"], ["🍄"], "roasting marshmallows over a tiny campfire, fireflies and glowing mushrooms, cozy night"),
+    (["🐶", "🦊"], ["🌊"], "building a sandcastle together with shell flags, gentle waves at golden hour"),
+    (["🐼", "🐨"], ["🌸"], "sharing tea under a blooming cherry tree, paper lanterns, drifting petals"),
+    (["🦁", "🐯"], ["🏡"], "tumbling over a ball of yarn in a cottage garden, picket fence and butterflies"),
+    (["🐧", "🐰"], ["🏔️"], "ice skating on a frozen pond atop a snowy mountain, fairy lights, gentle snowfall"),
+    (["🐸", "🦋"], ["🌺"], "resting together on lily pads among tropical flowers, dragonflies and warm bokeh"),
+    (["🐱", "🐶"], ["🌙"], "stargazing from a crescent moon with a tiny brass telescope, soft constellations"),
+    (["🦄", "🐰"], ["🌈"], "trotting side by side under a rainbow, flower garlands and floating sparkles"),
+    (["🦊", "🐼"], ["🍄"], "reading a glowing storybook under a toadstool, a lantern and curious fireflies"),
+    (["🐨", "🐧"], ["🌴"], "sipping coconut drinks on a tropical island, beach umbrella and gentle surf"),
+    (["🐰", "🦄"], ["⭐"], "swinging on a swing hung from the stars, sparkles raining down, deep blue night"),
+    (["🐱", "🐸"], ["🌊"], "collecting shells in tide pools at low tide, a little net and a pastel sunset"),
+
+    # ── Three animals, single place ──
+    (["🐰", "🐱", "🐶"], ["🍄"], "having a picnic on a checkered blanket among glowing mushrooms, lanterns and fireflies"),
+    (["🦊", "🐼", "🐨"], ["🌸"], "a tea party under cherry blossoms with tiny cups, paper lanterns and drifting petals"),
+    (["🦁", "🐯", "🐸"], ["🏡"], "playing tag through flower beds in a cottage garden, butterflies and warm sun"),
+    (["🐧", "🐰", "🐱"], ["🏔️"], "building a snowman on a snowy peak in matching scarves, sparkling snow, aurora above"),
+    (["🦄", "🦋", "🐰"], ["🌈"], "dancing under a rainbow amid sparkles and flower petals, pastel sky"),
+    (["🐶", "🦊", "🐼"], ["🌊"], "surfing tiny waves together with a beach ball, palm trees and sunset glow"),
+    (["🐱", "🐨", "🐸"], ["🌺"], "weaving flower crowns in a field of tropical flowers, butterflies and golden bokeh"),
+    (["🦁", "🐯", "🐰"], ["⭐"], "huddled on a cloud counting sparkling stars under a shared blanket, soft glow"),
+
+    # ── Two places ──
+    (["🐰"], ["🍄", "🌈"], "hopping from a mushroom grove toward a rainbow, sparkles bridging the two, pastel light"),
+    (["🐱", "🐶"], ["🌊", "🌴"], "a beach day between ocean waves and a tropical island, palm shade and scattered shells"),
+    (["🦄"], ["⭐", "🌙"], "soaring past sparkling stars toward a crescent moon, a glowing rainbow trail"),
+    (["🐼"], ["🌸", "🏡"], "wandering from cherry blossoms into a cosy cottage garden, petals and busy bees"),
+    (["🐧", "🐰"], ["🏔️", "⭐"], "watching sparkling stars from a snowy mountain top, fairy lights and soft snow"),
+    (["🦊"], ["🌺", "🌴"], "exploring tropical flowers along a tropical island shore, parrots and warm bokeh"),
+    (["🐸", "🦋"], ["🌸", "🌺"], "drifting between cherry blossoms and tropical flowers, dewdrops and floating petals"),
+    (["🦁"], ["🏡", "🌈"], "lazing in a cottage garden as a rainbow arcs overhead, butterflies and golden light"),
+
+    # ── Three places ──
+    (["🐰", "🐱"], ["🍄", "🌈", "⭐"], "a dreamy journey through a mushroom forest, under a rainbow and beneath sparkling stars, a glowing trail"),
+    (["🦄"], ["🌙", "⭐", "🌈"], "flying past a crescent moon and sparkling stars toward a rainbow, sparkles everywhere"),
+    (["🐶", "🦊", "🐼"], ["🌊", "🌴", "🌺"], "a tropical adventure across a sunny beach, a tropical island and fields of flowers, parrots and surf"),
+    (["🐧", "🐰", "🐱"], ["🏔️", "⭐", "🌙"], "a starry night on a snowy peak under a crescent moon and sparkling stars, fairy lights and aurora"),
 ]
 
-assert len(PROMPTS) == 80, f"Expected 80 prompts, got {len(PROMPTS)}"
+
+def _scene_name(idx: int, animals: list[str], places: list[str]) -> str:
+    """Short readable label for logs, e.g. '03_panda_1a1p'."""
+    first = ANIMAL_MAP[animals[0]].replace("baby ", "")
+    return f"{idx:02d}_{first}_{len(animals)}a{len(places)}p"
+
+
+# Build (name, full_prompt) using the SAME prefix builder + STYLE suffix as the app.
+PROMPTS: list[tuple[str, str]] = [
+    (_scene_name(i + 1, a, p), f"{build_subject(a, p)}, {detail}, {STYLE}")
+    for i, (a, p, detail) in enumerate(SCENES)
+]
 
 # ---------------------------------------------------------------------------
-# Generator using FLUX.1-dev via HuggingFace Inference API
+# Generator using FLUX.2-dev via HuggingFace Inference API
 # ---------------------------------------------------------------------------
 
 # Provider → model routing:
-#   fal-ai       → FLUX.1-dev  (best quality, free credits at fal.ai)
-#   hf-inference → FLUX.1-schnell (HF free tier, lower quality)
+#   fal-ai       → FLUX.2-dev      (best quality; billed to HF Pro credits)
+#   hf-inference → FLUX.1-schnell  (HF native fallback, lower quality)
 PROVIDERS = {
-    "fal-ai":       "black-forest-labs/FLUX.1-dev",
+    "fal-ai":       "black-forest-labs/FLUX.2-dev",
     "hf-inference": "black-forest-labs/FLUX.1-schnell",
 }