Update app.py

app.py

@@ -305,16 +305,33 @@ def generate_all_scenes(prompts_text):
                 image = generate_story_image(scene_prompt)
                 
                 if image is not None:
-                    img_array = np.array(image)
-                    generated_images.append(img_array)
+                    # Convert PIL Image to numpy array with explicit mode conversion
+                    pil_image = image if isinstance(image, Image.Image) else Image.fromarray(image)
+                    pil_image = pil_image.convert('RGB')  # Ensure RGB mode
+                    img_array = np.array(pil_image)
+                    
+                    # Verify array shape and type
+                    if len(img_array.shape) == 3 and img_array.shape[2] == 3:
+                        generated_images.append(img_array)
+                        print(f"Successfully added image for scene {scene_num}")
+                    else:
+                        print(f"Invalid image array shape: {img_array.shape}")
+                else:
+                    print(f"Failed to generate image for scene {scene_num}")
                 
                 clear_memory()
                 
             except Exception as e:
-                print(f"Error generating image: {str(e)}")
+                print(f"Error generating image for scene {scene_num}: {str(e)}")
                 clear_memory()
                 continue
     
+    # Verify we have images before returning
+    if not generated_images:
+        print("No images were successfully generated")
+        return [], "\n\n".join(formatted_prompts)
+    
+    print(f"Successfully generated {len(generated_images)} images")
     return generated_images, "\n\n".join(formatted_prompts)
 
 def overlay_text_on_image(image, text):