Update app.py

app.py

@@ -45,8 +45,12 @@ class ProfessionalCartoonFilmGenerator:
         
         # Model configurations for ZeroGPU optimization
         self.models_loaded = False
+        self.using_flux = False
         self.flux_pipe = None
         self.script_enhancer = None
+        self.cartoon_lora = None
+        self.character_lora = None
+        self.sketch_lora = None
         
     @spaces.GPU
     def load_models(self):
@@ -57,7 +61,7 @@ class ProfessionalCartoonFilmGenerator:
         print("🚀 Loading professional-grade models...")
         
         try:
-            # 1. FLUX pipeline for superior image generation
+            # 1. Try FLUX pipeline first (if user has authentication)
             print("🎨 Loading FLUX pipeline...")
             try:
                 self.flux_pipe = FluxPipeline.from_pretrained(
@@ -66,6 +70,8 @@ class ProfessionalCartoonFilmGenerator:
                     variant="fp16",
                     use_safetensors=True
                 ).to(self.device)
+                print("✅ FLUX pipeline loaded successfully!")
+                self.using_flux = True
             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")
@@ -74,11 +80,16 @@ class ProfessionalCartoonFilmGenerator:
                     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
+                    self.using_flux = False
                 else:
-                    raise flux_error
-            
-            # Load cartoon/anime LoRA for character generation
+                    print(f"❌ FLUX loading failed: {flux_error}")
+                    self.using_flux = False
+        except Exception as e:
+            print(f"❌ FLUX pipeline failed: {e}")
+            self.using_flux = False
+        
+        # Load cartoon/anime LoRA for character generation (only if FLUX is available)
+        if self.using_flux:
             print("🎭 Loading cartoon LoRA models...")
             try:
                 # Load multiple LoRA models for different purposes
@@ -98,27 +109,23 @@ class ProfessionalCartoonFilmGenerator:
             except Exception as e:
                 print(f"⚠️ Some LoRA models failed to load: {e}")
             
-            # Enable memory optimizations
-            self.flux_pipe.enable_vae_slicing()
-            self.flux_pipe.enable_vae_tiling()
-            
-            # Enable flash attention if available
-            if FLASH_ATTN_AVAILABLE:
-                try:
-                    self.flux_pipe.enable_xformers_memory_efficient_attention()
-                    print("✅ Flash attention enabled for better performance")
-                except Exception as e:
-                    print(f"⚠️ Flash attention failed: {e}")
-            else:
-                print("ℹ️ Using standard attention (flash attention not available)")
-            
-            print("✅ FLUX pipeline loaded successfully")
-            
-        except Exception as e:
-            print(f"❌ FLUX pipeline failed: {e}")
-            print("🔄 Falling back to Stable Diffusion...")
+            # Enable memory optimizations for FLUX
+            if self.flux_pipe:
+                self.flux_pipe.enable_vae_slicing()
+                self.flux_pipe.enable_vae_tiling()
+                
+                # Enable flash attention if available
+                if FLASH_ATTN_AVAILABLE:
+                    try:
+                        self.flux_pipe.enable_xformers_memory_efficient_attention()
+                        print("✅ Flash attention enabled for better performance")
+                    except Exception as e:
+                        print(f"⚠️ Flash attention failed: {e}")
+                else:
+                    print("ℹ️ Using standard attention (flash attention not available)")
             
-            # Fallback to Stable Diffusion
+        # Load Stable Diffusion fallback if FLUX is not available
+        if not self.using_flux:
             try:
                 from diffusers import StableDiffusionPipeline
                 print("🔄 Loading Stable Diffusion fallback model...")
@@ -204,17 +211,18 @@ class ProfessionalCartoonFilmGenerator:
     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)
+            file_size = os.path.getsize(file_path) / (1024*1024)
             
-            # 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"
+            # For Hugging Face Spaces, create a download info block
+            # Note: Replace 'your-username' and 'your-space-name' with your actual Space info
+            download_info = f"📥 Download {file_type}: {file_name}"
+            download_info += f"\n   📁 Local path: {file_path}"
+            download_info += f"\n   📊 File size: {file_size:.1f} MB"
+            download_info += f"\n   🔗 Space URL: https://huggingface.co/spaces/your-username/your-space-name"
+            download_info += f"\n   💡 To download: Return this file as a Gradio output component"
             
-            return download_url
+            return download_info
             
         except Exception as e:
             print(f"⚠️ Failed to create download URL: {e}")