Update app.py

app.py

@@ -106,22 +106,41 @@ def generate_images_with_together(story, style, quality, count=1):
         full_prompt = f"{style} style, cinematic lighting, quality {quality}, {base_prompt} [Scene {i + 1}]"
         seed = random.randint(1, 10_000_000)
 
-        response = together_client.images.generate(
-            prompt=full_prompt,
-            model="black-forest-labs/FLUX.1-schnell-Free",
-            seed=seed,
-            width=768,
-            height=512,
-            steps=4
-        )
-
-        if response.data and response.data[0].b64_json:
-            image_b64 = response.data[0].b64_json
-            image_data = base64.b64decode(image_b64)
-            image = Image.open(BytesIO(image_data))
-            images.append(image)
+        try:
+            resp = together_client.images.generate(
+                model="black-forest-labs/FLUX.1-schnell-Free",
+                prompt=full_prompt,
+                seed=seed,
+                width=768,
+                height=512,
+                steps=4
+            )
+        except Exception as e:
+            print("🔥 Together image API error:", e)
+            break
+
+        img = None
+        if resp.data:
+            choice = resp.data[0]
+
+            if getattr(choice, "url", None):
+                try:
+                    img_bytes = requests.get(choice.url, timeout=30).content
+                    img = Image.open(BytesIO(img_bytes))
+                except Exception as e:
+                    print("⚠️ URL fetch failed:", e)
+
+            elif getattr(choice, "b64_json", None):
+                try:
+                    img_bytes = base64.b64decode(choice.b64_json)
+                    img = Image.open(BytesIO(img_bytes))
+                except Exception as e:
+                    print("⚠️ base64 decode failed:", e)
+
+        if img is not None:
+            images.append(img)
         else:
-            print("⚠️ No image returned for scene", i + 1)
+            print(f"⚠️ No image for scene {i+1}")
 
     return images
 
@@ -156,8 +175,8 @@ def generate_assets(prompt, style, quality, num_images, state):
         try:
             images = generate_images_with_together(prompt, style, quality, int(num_images))
         except Exception as e:
-            print("⚠️ Together.ai image generation failed:", e)
-            images = []
+            status = f"⚠️ Image generation failed: {e}"
+            return [], graph_path, status, data
 
     status = "✅ All assets generated." if images else "✅ Graph generated (no images)."
     return images, graph_path, status, data