Update app.py

app.py

@@ -48,6 +48,66 @@ def run_command(command, desc=None):
         print(f"❌ Error during {desc}: {e}")
         raise
 
+def robust_download(url, dest_dir, filename):
+    dest_path = os.path.join(dest_dir, filename)
+    if os.path.exists(dest_path):
+        print(f"✅ {filename} already exists.")
+        return
+
+    print(f"⬇️ Downloading {filename}...")
+    
+    # Method 1: Try aria2c (fastest)
+    try:
+        # Check if aria2c is installed
+        subprocess.run(["aria2c", "--version"], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        run_command(f"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {url} -d {dest_dir} -o {filename}", f"Downloading {filename} (aria2c)")
+        return
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        print("⚠️ aria2c not found or failed, falling back to huggingface_hub...")
+    
+    # Method 2: huggingface_hub (reliable)
+    try:
+        from huggingface_hub import hf_hub_download
+        
+        # Parse Repo ID and Filename from URL
+        # URL format: https://huggingface.co/USER/REPO/resolve/main/PATH/TO/FILE
+        parts = url.replace("https://huggingface.co/", "").split("/resolve/main/")
+        if len(parts) == 2:
+            repo_id = parts[0]
+            subfolder = os.path.dirname(parts[1]) if "/" in parts[1] else None
+            remote_filename = os.path.basename(parts[1])
+            
+            # Download
+            print(f"⏳ Downloading via HF Hub: {repo_id}/{remote_filename}")
+            downloaded_path = hf_hub_download(
+                repo_id=repo_id,
+                filename=parts[1], # Pass full path as filename argument usually handles directory structure?
+                                   # Actually hf_hub_download 'filename' argument is relative path in repo.
+                local_dir=dest_dir,
+                local_dir_use_symlinks=False
+            )
+            
+            # Rename if necessary/ensure it matches what we expect
+            # hf_hub_download with local_dir preserves structure usually.
+            # check where it landed.
+            expected_path = os.path.join(dest_dir, remote_filename)
+            # If subfolders are involved, it might be deep.
+            # Simpler: just move it if name doesn't match
+            
+            # Re-verification handled by existence check on next run
+            return
+    except Exception as e:
+        print(f"❌ Fallback download failed: {e}")
+        # Method 3: Requests (slowest fallback)
+        import requests
+        print(f"⚠️ Trying simple requests download...")
+        with requests.get(url, stream=True) as r:
+            r.raise_for_status()
+            with open(dest_path, 'wb') as f:
+                for chunk in r.iter_content(chunk_size=8192):
+                    f.write(chunk)
+        return
+
 def setup_environment():
     print("🚀 Starting Setup Environment...")
     
@@ -76,20 +136,7 @@ def setup_environment():
     ]
 
     for url, dest_dir, filename in download_list:
-        dest_path = os.path.join(dest_dir, filename)
-        if not os.path.exists(dest_path):
-            print(f"⬇️ Downloading {filename}...")
-            # Use aria2c if available, else wget/curl, or fallback to python
-            # Since we installed aria2 in Dockerfile, try that first
-            try:
-                run_command(f"aria2c --console-log-level=error -c -x 16 -s 16 -k 1M {url} -d {dest_dir} -o {filename}", f"Downloading {filename}")
-            except:
-                print("⚠️ Aria2 failed, falling back to basic download methods could be added here if needed.")
-                # Basic fallback using huggingface_hub or wget could go here
-                from huggingface_hub import hf_hub_download
-                # This is a bit complex since URLs are direct, but for now assuming aria2/git works or manual download
-        else:
-            print(f"✅ {filename} already exists.")
+        robust_download(url, dest_dir, filename)
 
     print("🎉 Environment Setup Complete!")
 
@@ -97,6 +144,13 @@ def setup_environment():
 setup_environment()
 
 # --- ComfyUI Imports ---
+# Configure Execution Arguments for ComfyUI
+if not torch.cuda.is_available():
+    print("⚠️ CUDA not available, forcing CPU mode for ComfyUI...")
+    # Inject --cpu argument so ComfyUI optimizes for CPU instead of crashing on CUDA init
+    if "--cpu" not in sys.argv:
+        sys.argv.append("--cpu")
+
 # These must happen AFTER setup because ComfyUI folder might not exist before
 try:
     import nodes