Update app.py

app.py

@@ -13,6 +13,8 @@ import spaces
 import gc
 from huggingface_hub import hf_hub_download
 import threading
+import datetime
+import time
 
 # ZeroGPU-compatible imports
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
@@ -1147,13 +1149,17 @@ class ProfessionalCartoonFilmGenerator:
                     "scenes": [],
                     "style": "Model loading failed"
                 }
-                return None, error_info, "❌ Failed to load AI models", [], []
+                return None, error_info, "❌ Failed to load AI models", [], [], None, None, []
             
             # Step 1: Generate professional script
             print("📝 Creating professional script structure...")
             script_data = self.generate_professional_script(script)
             print(f"✅ Script generated with {len(script_data['scenes'])} scenes")
             
+            # Save script to file
+            print("📄 Saving script to file...")
+            script_file_path = self.save_script_to_file(script_data, script)
+            
             # Step 2: Generate high-quality characters  
             print("🎭 Creating professional character designs...")
             character_images = self.generate_professional_character_images(script_data['characters'])
@@ -1193,10 +1199,22 @@ class ProfessionalCartoonFilmGenerator:
                     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
+                    # Create download links for all files
+                    all_files = {}
+                    if script_file_path:
+                        all_files["script"] = script_file_path
+                    if final_video:
+                        all_files["video"] = final_video
+                    all_files.update(character_images)
+                    all_files.update(background_images)
+                    
+                    download_links = self.create_download_links(all_files)
+                    script_file, video_file = self.get_download_files(all_files)
+                    
+                    return final_video, script_data, f"✅ Professional cartoon film generated successfully! ({file_size:.1f} MB)", char_files, bg_files, script_file, video_file, download_links
                 else:
                     print("⚠️ Video merging failed")
-                    return None, script_data, "⚠️ Video merging failed", [], []
+                    return None, script_data, "⚠️ Video merging failed", [], [], None, None, []
             else:
                 print("❌ No videos to merge - video generation failed")
                 print("🔄 Creating emergency fallback video...")
@@ -1211,12 +1229,25 @@ 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)", [], []
+                        
+                        # Create download links for emergency files
+                        all_files = {}
+                        if script_file_path:
+                            all_files["script"] = script_file_path
+                        if emergency_video:
+                            all_files["video"] = emergency_video
+                        all_files.update(character_images)
+                        all_files.update(background_images)
+                        
+                        download_links = self.create_download_links(all_files)
+                        script_file, video_file = self.get_download_files(all_files)
+                        
+                        return emergency_video, script_data, f"⚠️ Emergency fallback video created ({file_size:.1f} MB)", [], [], script_file, video_file, download_links
                     else:
-                        return None, script_data, "❌ No videos generated - all methods failed", [], []
+                        return None, script_data, "❌ No videos generated - all methods failed", [], [], None, None, []
                 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", [], [], None, None, []
                 
         except Exception as e:
             print(f"❌ Generation failed: {e}")
@@ -1229,7 +1260,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)}", [], [], None, None, []
 
     def _create_lightweight_animated_video(self, scene: Dict, character_images: Dict, background_images: Dict) -> str:
         """Create lightweight animated video with character/background compositing"""
@@ -1402,6 +1433,67 @@ class ProfessionalCartoonFilmGenerator:
             print(f"⚠️ Character overlay failed: {e}")
             return background
 
+    def save_script_to_file(self, script_data: Dict[str, Any], original_script: str) -> str:
+        """Save script data to a JSON file in tmp folder"""
+        try:
+            # Create a comprehensive script file with all data
+            script_file_data = {
+                "original_script": original_script,
+                "generated_script": script_data,
+                "timestamp": str(datetime.datetime.now()),
+                "version": "1.0"
+            }
+            
+            # Save to tmp folder
+            script_path = f"{self.output_dir}/cartoon_script_{int(time.time())}.json"
+            
+            with open(script_path, 'w', encoding='utf-8') as f:
+                json.dump(script_file_data, f, indent=2, ensure_ascii=False)
+            
+            if os.path.exists(script_path):
+                file_size = os.path.getsize(script_path) / 1024  # KB
+                print(f"📝 Script saved: {script_path} ({file_size:.1f} KB)")
+                return script_path
+            else:
+                print(f"❌ Failed to save script: {script_path}")
+                return None
+                
+        except Exception as e:
+            print(f"❌ Error saving script: {e}")
+            return None
+    
+    def create_download_links(self, files_dict: Dict[str, str]) -> List[Dict[str, str]]:
+        """Create download links for files"""
+        download_links = []
+        
+        for file_type, file_path in files_dict.items():
+            if os.path.exists(file_path):
+                file_name = os.path.basename(file_path)
+                file_size = os.path.getsize(file_path) / (1024*1024)  # MB
+                
+                download_links.append({
+                    "name": file_name,
+                    "path": file_path,
+                    "size": f"{file_size:.1f} MB",
+                    "type": file_type
+                })
+        
+        return download_links
+    
+    def get_download_files(self, files_dict: Dict[str, str]) -> tuple:
+        """Get file objects for Gradio download components"""
+        script_file = None
+        video_file = None
+        
+        for file_type, file_path in files_dict.items():
+            if os.path.exists(file_path):
+                if file_type == "script":
+                    script_file = file_path
+                elif file_type == "video":
+                    video_file = file_path
+        
+        return script_file, video_file
+
 # Initialize professional generator
 generator = ProfessionalCartoonFilmGenerator()
 
@@ -1416,7 +1508,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", [], [], None, None, []
     
     # Check if another generation is in progress
     if not generation_lock.acquire(blocking=False):
@@ -1427,7 +1519,7 @@ def create_professional_cartoon_film(script):
             "scenes": [],
             "style": "Please wait for current generation to complete"
         }
-        return None, busy_response, "⏳ Generation already in progress - please wait", [], []
+        return None, busy_response, "⏳ Generation already in progress - please wait", [], [], None, None, []
     
     try:
         return generator.generate_professional_cartoon_film(script)
@@ -1476,6 +1568,8 @@ with gr.Blocks(
     - **Cinematic backgrounds** with professional lighting
     - **Advanced animation effects** based on scene mood
     - **4K video output** with 24fps cinematic quality
+    - **📄 Script downloads** - Full JSON with story analysis
+    - **📁 File management** - All files saved in /tmp with download links
     
     **🎯 Perfect for:**
     - Content creators seeking professional results
@@ -1486,10 +1580,11 @@ with gr.Blocks(
     ---
     
     **⚠️ Current Status:**
-    - ✅ **Storage System:** Fixed for Hugging Face Spaces
+    - ✅ **Storage System:** Fixed for Hugging Face Spaces (/tmp folder)
+    - ✅ **Script Downloads:** JSON files with complete story analysis
+    - ✅ **File Downloads:** Direct download buttons for all generated content
     - ⚠️ **FLUX Models:** Require authentication token (using Stable Diffusion fallback)
     - ⚠️ **Open-Sora:** Using static video fallback for stability
-    - ✅ **File Downloads:** Available through Gradio galleries below
     
     **💡 To unlock full FLUX quality:**
     1. Get token from [Hugging Face Settings](https://huggingface.co/settings/tokens)
@@ -1533,8 +1628,9 @@ The more details you provide about characters, setting, and emotion, the better
             **⚠️ Important Notes:**
             - Only **ONE generation at a time** - multiple clicks will be queued
             - **Processing takes 8-12 minutes** - please be patient
-            - **Files auto-download** through galleries below
-            - **Storage fixed** for Hugging Face Spaces compatibility
+            - **Files saved in /tmp folder** with download links below
+            - **Script saved as JSON** with full story analysis
+            - **Images and videos** available for download
             """)
             
             video_output = gr.Video(
@@ -1557,6 +1653,25 @@ The more details you provide about characters, setting, and emotion, the better
                     allow_preview=True
                 )
                 
+                # Add download buttons for scripts and other files
+                script_download = gr.File(
+                    label="📄 Download Script (JSON)",
+                    file_types=[".json"],
+                    visible=True
+                )
+                
+                video_download = gr.File(
+                    label="🎬 Download Video (MP4)",
+                    file_types=[".mp4"],
+                    visible=True
+                )
+                
+                # Download links display
+                download_links_output = gr.JSON(
+                    label="📥 Download Links",
+                    visible=True
+                )
+                
             status_output = gr.Textbox(
                 label="📊 Generation Status",
                 lines=3
@@ -1571,7 +1686,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, character_gallery, background_gallery],
+        outputs=[video_output, script_details, status_output, character_gallery, background_gallery, script_download, video_download, download_links_output],
         show_progress=True
     )