Update app.py

app.py

@@ -1178,10 +1178,15 @@ class ProfessionalCartoonFilmGenerator:
                     download_info = self.create_download_url(final_video, "final_cartoon_film")
                     print(f"✅ Professional cartoon film generation complete!")
                     print(download_info)
-                    return final_video, script_data, f"✅ Professional cartoon film generated successfully! ({file_size:.1f} MB)"
+                    
+                    # Prepare character and background files for galleries
+                    char_files = list(character_images.values()) if character_images else []
+                    bg_files = list(background_images.values()) if background_images else []
+                    
+                    return final_video, script_data, f"✅ Professional cartoon film generated successfully! ({file_size:.1f} MB)", char_files, bg_files
                 else:
                     print("⚠️ Video merging failed")
-                    return None, script_data, "⚠️ Video merging failed"
+                    return None, script_data, "⚠️ Video merging failed", [], []
             else:
                 print("❌ No videos to merge - video generation failed")
                 print("🔄 Creating emergency fallback video...")
@@ -1196,12 +1201,12 @@ class ProfessionalCartoonFilmGenerator:
                         download_info = self.create_download_url(emergency_video, "emergency_fallback_video")
                         print(f"✅ Emergency fallback video created")
                         print(download_info)
-                        return emergency_video, script_data, f"⚠️ Emergency fallback video created ({file_size:.1f} MB)"
+                        return emergency_video, script_data, f"⚠️ Emergency fallback video created ({file_size:.1f} MB)", [], []
                     else:
-                        return None, script_data, "❌ No videos generated - all methods failed"
+                        return None, script_data, "❌ No videos generated - all methods failed", [], []
                 except Exception as e:
                     print(f"❌ Emergency fallback also failed: {e}")
-                    return None, script_data, "❌ No videos generated - all methods failed"
+                    return None, script_data, "❌ No videos generated - all methods failed", [], []
                 
         except Exception as e:
             print(f"❌ Generation failed: {e}")
@@ -1214,7 +1219,7 @@ class ProfessionalCartoonFilmGenerator:
                 "scenes": [],
                 "style": "Error occurred during generation"
             }
-            return None, error_info, f"❌ Generation failed: {str(e)}"
+            return None, error_info, f"❌ Generation failed: {str(e)}", [], []
 
 # Initialize professional generator
 generator = ProfessionalCartoonFilmGenerator()
@@ -1230,7 +1235,7 @@ def create_professional_cartoon_film(script):
             "scenes": [],
             "style": "Please enter a script"
         }
-        return None, empty_response, "❌ Please enter a script"
+        return None, empty_response, "❌ Please enter a script", [], []
     
     return generator.generate_professional_cartoon_film(script)
 
@@ -1321,6 +1326,21 @@ The more details you provide about characters, setting, and emotion, the better
                 height=500
             )
             
+            # Add file galleries for generated content
+            with gr.Accordion("📁 Generated Files (Click to Download)", open=False):
+                character_gallery = gr.Gallery(
+                    label="🎭 Character Images",
+                    columns=2,
+                    height=200,
+                    allow_preview=True
+                )
+                background_gallery = gr.Gallery(
+                    label="🏞️ Background Images", 
+                    columns=2,
+                    height=200,
+                    allow_preview=True
+                )
+                
             status_output = gr.Textbox(
                 label="📊 Generation Status",
                 lines=3
@@ -1335,7 +1355,7 @@ The more details you provide about characters, setting, and emotion, the better
     generate_btn.click(
         fn=create_professional_cartoon_film,
         inputs=[script_input],
-        outputs=[video_output, script_details, status_output],
+        outputs=[video_output, script_details, status_output, character_gallery, background_gallery],
         show_progress=True
     )