Update app.py

app.py

@@ -53,10 +53,11 @@ TEXT:
 """
 
 IMAGE_PROMPT_TEMPLATE = """
-Based on the following story, craft ONE vivid scene description suitable as a prompt for a text‑to‑image AI.
-Do NOT restate the entire story—write only the visual description.
-Include setting, mood, key characters, and distinctive details.
-Return a single sentence, nothing else.
+Based on the following story, write %d distinct vivid scene descriptions, one per line.
+Each line should begin with a dash (-) followed by a detailed image-worthy scene.
+Include setting, mood, characters, and visual cues.
+
+Return ONLY the list of scenes, each on its own line.
 
 Story:
 \"\"\"%s\"\"\"
@@ -85,17 +86,20 @@ def extract_entities(text: str):
 # ────────────────────────────────
 # Build visual prompt
 # ────────────────────────────────
-def generate_image_prompt(story_text: str):
+def generate_image_prompts(story_text: str, count=1):
     try:
-        prompt_msg = IMAGE_PROMPT_TEMPLATE % story_text
+        prompt_msg = IMAGE_PROMPT_TEMPLATE % (count, story_text)
         resp = together_client.chat.completions.create(
             model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
             messages=[{"role": "user", "content": prompt_msg}],
-            max_tokens=120,
+            max_tokens=200,
         )
-        return resp.choices[0].message.content.strip()
+        raw_output = resp.choices[0].message.content.strip()
+        prompts = [line.strip("-• ").strip() for line in raw_output.split("\n") if line.strip()]
+        return prompts[:count]  # just in case LLM gives more than needed
     except Exception as e:
-        return e
+        print("⚠️ LLM scene prompt generation failed:", e)
+        return []