Update app.py

app.py

@@ -59,12 +59,24 @@ class ProfessionalCartoonFilmGenerator:
         try:
             # 1. FLUX pipeline for superior image generation
             print("🎨 Loading FLUX pipeline...")
-            self.flux_pipe = FluxPipeline.from_pretrained(
-                "black-forest-labs/FLUX.1-dev",
-                torch_dtype=torch.bfloat16,
-                variant="fp16",
-                use_safetensors=True
-            ).to(self.device)
+            try:
+                self.flux_pipe = FluxPipeline.from_pretrained(
+                    "black-forest-labs/FLUX.1-dev",
+                    torch_dtype=torch.bfloat16,
+                    variant="fp16",
+                    use_safetensors=True
+                ).to(self.device)
+            except Exception as flux_error:
+                if "401" in str(flux_error) or "authentication" in str(flux_error).lower():
+                    print("🔐 FLUX authentication failed - model requires Hugging Face token")
+                    print("💡 To use FLUX, you need to:")
+                    print("   1. Get a Hugging Face token from https://huggingface.co/settings/tokens")
+                    print("   2. Accept the FLUX model license at https://huggingface.co/black-forest-labs/FLUX.1-dev")
+                    print("   3. Set your token: huggingface-cli login")
+                    print("🔄 Falling back to Stable Diffusion...")
+                    raise flux_error
+                else:
+                    raise flux_error
             
             # Load cartoon/anime LoRA for character generation
             print("🎭 Loading cartoon LoRA models...")
@@ -109,13 +121,29 @@ class ProfessionalCartoonFilmGenerator:
             # Fallback to Stable Diffusion
             try:
                 from diffusers import StableDiffusionPipeline
-                self.flux_pipe = StableDiffusionPipeline.from_pretrained(
-                    "runwayml/stable-diffusion-v1-5",
-                    torch_dtype=torch.float16,
-                    use_safetensors=True,
-                    safety_checker=None,
-                    requires_safety_checker=False
-                ).to(self.device)
+                print("🔄 Loading Stable Diffusion fallback model...")
+                
+                # Try a more accessible model first
+                try:
+                    self.flux_pipe = StableDiffusionPipeline.from_pretrained(
+                        "CompVis/stable-diffusion-v1-4",
+                        torch_dtype=torch.float16,
+                        use_safetensors=True,
+                        safety_checker=None,
+                        requires_safety_checker=False
+                    ).to(self.device)
+                    print("✅ Loaded Stable Diffusion v1.4")
+                except Exception as sd_error:
+                    print(f"⚠️ SD v1.4 failed: {sd_error}")
+                    # Try the original model
+                    self.flux_pipe = StableDiffusionPipeline.from_pretrained(
+                        "runwayml/stable-diffusion-v1-5",
+                        torch_dtype=torch.float16,
+                        use_safetensors=True,
+                        safety_checker=None,
+                        requires_safety_checker=False
+                    ).to(self.device)
+                    print("✅ Loaded Stable Diffusion v1.5")
                 
                 # Enable memory optimizations
                 self.flux_pipe.enable_vae_slicing()
@@ -173,6 +201,25 @@ class ProfessionalCartoonFilmGenerator:
             words = prompt.split()
             return " ".join(words[:50])
     
+    def create_download_url(self, file_path: str, file_type: str = "file") -> str:
+        """Create a download URL for generated content"""
+        try:
+            # For Hugging Face Spaces, we can create a simple download link
+            # In a real deployment, this would point to your actual file hosting service
+            file_name = os.path.basename(file_path)
+            
+            # Create a simple download URL format
+            # In production, replace this with your actual file hosting URL
+            download_url = f"📥 Download {file_type}: {file_name}"
+            download_url += f"\n   📁 Local path: {file_path}"
+            download_url += f"\n   📊 File size: {os.path.getsize(file_path) / (1024*1024):.1f} MB"
+            
+            return download_url
+            
+        except Exception as e:
+            print(f"⚠️ Failed to create download URL: {e}")
+            return f"📁 File saved: {file_path}"
+    
     def generate_professional_script(self, user_input: str) -> Dict[str, Any]:
         """Generate a professional cartoon script with detailed character development"""
         
@@ -507,7 +554,11 @@ class ProfessionalCartoonFilmGenerator:
                 char_path = f"{self.temp_dir}/character_{character['name'].replace(' ', '_')}.png"
                 image.save(char_path)
                 character_images[character['name']] = char_path
+                
+                # Create download URL for character
+                download_info = self.create_download_url(char_path, f"character_{character['name']}")
                 print(f"✅ Generated high-quality character: {character['name']}")
+                print(download_info)
                 
                 self.clear_gpu_memory()
                 
@@ -601,7 +652,11 @@ class ProfessionalCartoonFilmGenerator:
                 bg_path = f"{self.temp_dir}/background_scene_{scene['scene_number']}.png"
                 image.save(bg_path)
                 background_images[scene['scene_number']] = bg_path
+                
+                # Create download URL for background
+                download_info = self.create_download_url(bg_path, f"background_scene_{scene['scene_number']}")
                 print(f"✅ Created cinematic background for scene {scene['scene_number']}")
+                print(download_info)
                 
                 self.clear_gpu_memory()
                 
@@ -688,8 +743,11 @@ class ProfessionalCartoonFilmGenerator:
                 
                 if video_path and os.path.exists(video_path):
                     scene_videos.append(video_path)
-                    print(f"✅ Generated professional video for scene {scene_num}: {video_path}")
-                    print(f"📁 Video size: {os.path.getsize(video_path) / (1024*1024):.1f} MB")
+                    
+                    # Create download URL for video
+                    download_info = self.create_download_url(video_path, f"video_scene_{scene_num}")
+                    print(f"✅ Generated professional video for scene {scene_num}")
+                    print(download_info)
                 else:
                     print(f"❌ No video generated for scene {scene_num}")
                 
@@ -1102,8 +1160,11 @@ class ProfessionalCartoonFilmGenerator:
                 
                 if final_video and os.path.exists(final_video):
                     file_size = os.path.getsize(final_video) / (1024*1024)
+                    
+                    # Create download URL for final video
+                    download_info = self.create_download_url(final_video, "final_cartoon_film")
                     print(f"✅ Professional cartoon film generation complete!")
-                    print(f"📁 Final video: {final_video} ({file_size:.1f} MB)")
+                    print(download_info)
                     return final_video, script_data, f"✅ Professional cartoon film generated successfully! ({file_size:.1f} MB)"
                 else:
                     print("⚠️ Video merging failed")
@@ -1117,7 +1178,11 @@ class ProfessionalCartoonFilmGenerator:
                     emergency_video = self._create_emergency_fallback_video(script_data)
                     if emergency_video and os.path.exists(emergency_video):
                         file_size = os.path.getsize(emergency_video) / (1024*1024)
-                        print(f"✅ Emergency fallback video created: {emergency_video} ({file_size:.1f} MB)")
+                        
+                        # Create download URL for emergency video
+                        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)"
                     else:
                         return None, script_data, "❌ No videos generated - all methods failed"