Update app.py

app.py

@@ -114,9 +114,21 @@ def download_required_models():
     import urllib.error
     
     models_to_download = [
+        {
+            'name': 'inswapper_128_fp16.onnx',
+            'url': 'https://huggingface.co/ninjawick/webui-faceswap-unlocked/resolve/main/inswapper_128_fp16.onnx',
+            'path': ASSETS_DIR / 'inswapper_128_fp16.onnx',
+            'description': 'InSwapper FP16 face swap model'
+        },
+        {
+            'name': 'inswapper_128.onnx',
+            'url': 'https://huggingface.co/xingren23/comfyflow-models/resolve/main/insightface/inswapper_128.onnx',
+            'path': ASSETS_DIR / 'inswapper_128.onnx',
+            'description': 'InSwapper face swap model'
+        },
         {
             'name': 'GFPGANv1.4.pth',
-            'url': 'https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth',
+            'url': 'https://huggingface.co/gmk123/GFPGAN/resolve/main/GFPGANv1.4.pth',
             'path': ASSETS_DIR / 'GFPGANv1.4.pth',
             'description': 'GFPGAN face enhancement model'
         }
@@ -137,6 +149,7 @@ def download_required_models():
         
         try:
             print(f"📥 Downloading {model_name}...")
+            print(f"    Description: {model['description']}")
             print(f"    URL: {model_url}")
             print(f"    Path: {model_path}")
             
@@ -147,11 +160,28 @@ def download_required_models():
             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 % 100 == 0:  # Update every 100 blocks to avoid spam
+                    if block_num % 200 == 0:  # Update every 200 blocks to avoid spam
                         print(f"    Progress: {percent:.1f}%")
             
-            # Download with progress
-            urllib.request.urlretrieve(model_url, str(model_path), progress_callback)
+            # Download with progress and proper headers for HuggingFace
+            req = urllib.request.Request(model_url)
+            req.add_header('User-Agent', 'Mozilla/5.0 (compatible; FaceSwapStudio/1.0)')
+            
+            with urllib.request.urlopen(req) as response:
+                total_size = int(response.headers.get('Content-Length', 0))
+                downloaded = 0
+                
+                with open(model_path, 'wb') as f:
+                    while True:
+                        chunk = response.read(8192)
+                        if not chunk:
+                            break
+                        f.write(chunk)
+                        downloaded += len(chunk)
+                        
+                        if total_size > 0 and downloaded % (8192 * 100) == 0:  # Progress every ~800KB
+                            percent = (downloaded * 100) / total_size
+                            print(f"    Progress: {percent:.1f}%")
             
             # Verify download
             if model_path.exists() and model_path.stat().st_size > 1024:
@@ -168,7 +198,7 @@ def download_required_models():
         except Exception as e:
             print(f"❌ Error downloading {model_name}: {e}")
 
-# Download models at startup
+# Download models at startup - BEFORE web interface
 print("\n🔄 Checking required model files...")
 try:
     download_required_models()