Update app.py

app.py

@@ -443,13 +443,41 @@ def upscale_video_4k(input_path, output_path):
 # Initialize directories and try to set up Real-ESRGAN
 ensure_directories()
 
-# Try to initialize Real-ESRGAN on startup
-if REALESRGAN_AVAILABLE:
+def force_init_realesrgan():
+    """Force Real-ESRGAN initialization with detailed logging"""
+    log_message("🔧 Force initializing Real-ESRGAN...")
+    
+    if not REALESRGAN_AVAILABLE:
+        log_message("❌ Real-ESRGAN package not available")
+        return False
+    
     try:
-        log_message("🚀 Attempting to initialize Real-ESRGAN on startup...")
-        initialize_realesrgan()
+        # Try to download models first
+        log_message("📥 Checking/downloading Real-ESRGAN models...")
+        download_success = download_realesrgan_models()
+        if not download_success:
+            log_message("❌ Model download failed")
+            return False
+        
+        # Initialize upscaler
+        log_message("🔧 Creating Real-ESRGAN upscaler...")
+        upscaler = initialize_realesrgan()
+        if upscaler:
+            log_message("✅ Real-ESRGAN initialized successfully!")
+            return True
+        else:
+            log_message("❌ Real-ESRGAN initialization returned None")
+            return False
+            
     except Exception as e:
-        log_message(f"⚠️ Could not initialize Real-ESRGAN on startup: {e}")
+        log_message(f"❌ Real-ESRGAN initialization error: {str(e)}")
+        import traceback
+        log_message(f"🔍 Traceback: {traceback.format_exc()}")
+        return False
+
+# Try to initialize Real-ESRGAN on startup
+log_message("🚀 Starting Real-ESRGAN initialization...")
+force_init_realesrgan()
 
 app = Flask(__name__)
 
@@ -484,9 +512,19 @@ def api_system():
         info["pytorch_version"] = torch.__version__
         
         # Real-ESRGAN info
-        info["realesrgan_available"] = REALESRGAN_AVAILABLE and app_state["upscaler"] is not None
+        info["realesrgan_available"] = REALESRGAN_AVAILABLE
+        info["realesrgan_initialized"] = app_state["upscaler"] is not None
         info["current_model"] = app_state.get("current_model", "None")
         
+        # Check if models exist
+        models_status = {}
+        if REALESRGAN_AVAILABLE:
+            models = ['RealESRGAN_x4plus', 'RealESRGAN_x2plus']
+            for model in models:
+                model_path = os.path.join(MODEL_FOLDER, f"{model}.pth")
+                models_status[model] = os.path.exists(model_path)
+        info["models_downloaded"] = models_status
+        
         # Storage info
         try:
             upload_files = os.listdir(UPLOAD_FOLDER) if os.path.exists(UPLOAD_FOLDER) else []