Update app.py

app.py

@@ -696,24 +696,39 @@ with gr.Blocks(title="Axon v6") as demo:
 print("🔥 Axon v6 starting...")
 print("⬇️ Pre-downloading all models...")
 
-for model_name in MODELS.keys():
-    print(f"📥 Checking {model_name}...")
-    filename = MODELS[model_name]
+download_results = {}
+
+for model_name, filename in MODELS.items():
+    print(f"\n📥 Checking: {model_name}")
+    print(f"   Filename: {filename}")
     model_path = os.path.join(MODELS_DIR, filename)
     
-    if not os.path.exists(model_path):
+    if os.path.exists(model_path):
+        print(f"   ✅ Already cached")
+        download_results[model_name] = "cached"
+    else:
         repo_id = MODEL_REPOS.get(filename)
+        print(f"   Repo: {repo_id}")
+        
         if repo_id:
-            print(f"⬇️ Downloading {filename}...")
             try:
+                print(f"   ⬇️ Downloading...")
                 hf_hub_download(repo_id=repo_id, filename=filename, local_dir=MODELS_DIR)
-                print(f"✅ {filename} downloaded!")
+                print(f"   ✅ Downloaded!")
+                download_results[model_name] = "downloaded"
             except Exception as e:
-                print(f"❌ Failed to download {filename}: {e}")
-    else:
-        print(f"✅ {filename} already cached")
+                print(f"   ❌ FAILED: {e}")
+                download_results[model_name] = f"failed: {e}"
+        else:
+            print(f"   ❌ No repo configured!")
+            download_results[model_name] = "no repo"
 
-print("✅ All models ready!")
-print("🚀 Launching Axon...")
+print("\n" + "="*50)
+print("📊 DOWNLOAD SUMMARY:")
+for model, status in download_results.items():
+    emoji = "✅" if status in ["cached", "downloaded"] else "❌"
+    print(f"   {emoji} {model}: {status}")
+print("="*50 + "\n")
 
+print("🚀 Launching Axon...")
 demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft(primary_hue="indigo"))