Update app.py

app.py

@@ -1,21 +1,3 @@
-import os
-import urllib.request
-
-# Target path (relative to your repo root)
-MODEL_DIR = './.assets/models'
-MODEL_PATH = os.path.join(MODEL_DIR, 'GFPGANv1.4.pth')
-MODEL_URL = 'https://huggingface.co/talhaty/GFPGANv1.4/resolve/main/GFPGANv1.4.pth'
-
-# Create directory if it does not exist
-os.makedirs(MODEL_DIR, exist_ok=True)
-
-# Download if not already present
-if not os.path.isfile(MODEL_PATH):
-    print(f'Downloading GFPGANv1.4.pth to {MODEL_PATH}...')
-    urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
-    print('Download complete.')
-else:
-    print(f'GFPGANv1.4.pth already exists at {MODEL_PATH}, skipping download.')
 """
 🎭 Advanced Face Swap Studio - HuggingFace Spaces Optimized
 =========================================================
@@ -75,15 +57,75 @@ BASE_DIR = Path(__file__).parent
 TEMP_DIR = Path(tempfile.mkdtemp(prefix="facefusion_"))
 OUTPUT_DIR = BASE_DIR / "outputs"
 CONVERT_DIR = BASE_DIR / "Convert"
+ASSETS_DIR = BASE_DIR / ".assets" / "models"
 
 # Create directories
 TEMP_DIR.mkdir(exist_ok=True)
 OUTPUT_DIR.mkdir(exist_ok=True)
 CONVERT_DIR.mkdir(exist_ok=True)
+ASSETS_DIR.mkdir(parents=True, exist_ok=True)
 
 print(f"📁 Base directory: {BASE_DIR}")
 print(f"📂 Temp directory: {TEMP_DIR}")
 print(f"📤 Output directory: {OUTPUT_DIR}")
+print(f"🎯 Assets directory: {ASSETS_DIR}")
+
+# Download required model files
+def download_required_models():
+    """Download required model files if not present"""
+    import urllib.request
+    import urllib.error
+    
+    models_to_download = [
+        {
+            'name': 'GFPGANv1.4.pth',
+            'url': 'https://huggingface.co/talhaty/GFPGANv1.4/resolve/main/GFPGANv1.4.pth',
+            'path': ASSETS_DIR / 'GFPGANv1.4.pth',
+            'description': 'GFPGAN face enhancement model'
+        }
+    ]
+    
+    for model in models_to_download:
+        model_path = model['path']
+        model_url = model['url']
+        model_name = model['name']
+        
+        if model_path.exists():
+            file_size = model_path.stat().st_size / (1024 * 1024)  # MB
+            print(f"✅ {model_name} already exists ({file_size:.1f}MB)")
+            continue
+        
+        try:
+            print(f"📥 Downloading {model_name}...")
+            print(f"    URL: {model_url}")
+            print(f"    Path: {model_path}")
+            
+            # Create a progress callback
+            def progress_callback(block_num, block_size, total_size):
+                if total_size > 0:
+                    percent = min(100, (block_num * block_size * 100) / total_size)
+                    if block_num % 50 == 0:  # Update every 50 blocks to avoid spam
+                        print(f"    Progress: {percent:.1f}%")
+            
+            # Download with progress
+            urllib.request.urlretrieve(model_url, str(model_path), progress_callback)
+            
+            # Verify download
+            if model_path.exists() and model_path.stat().st_size > 0:
+                file_size = model_path.stat().st_size / (1024 * 1024)  # MB
+                print(f"✅ {model_name} downloaded successfully ({file_size:.1f}MB)")
+            else:
+                print(f"❌ {model_name} download failed - file not created or empty")
+                
+        except urllib.error.URLError as e:
+            print(f"❌ Network error downloading {model_name}: {e}")
+        except Exception as e:
+            print(f"❌ Error downloading {model_name}: {e}")
+
+# Download models at startup
+print("\n🔄 Checking required model files...")
+download_required_models()
+print("✅ Model check complete\n")
 
 # Global variables
 current_process = None