gg

app_g.py

@@ -11,17 +11,24 @@ import sys
 from huggingface_hub import snapshot_download
 
 # === Setup Paths ===
-PUSA_ROOT = os.path.abspath("./PusaV1")
+import os
+import sys
+from huggingface_hub import snapshot_download
+
+# === Robust Base Path ===
+# Ensures compatibility inside Hugging Face Spaces (or any container)
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+PUSA_ROOT = os.path.join(BASE_DIR, "PusaV1")
 MODEL_ZOO_DIR = os.path.join(PUSA_ROOT, "model_zoo")
 WAN_SUBFOLDER = "Wan2.1-T2V-14B"
 WAN_MODEL_PATH = os.path.join(MODEL_ZOO_DIR, WAN_SUBFOLDER)
 LORA_PATH = os.path.join(MODEL_ZOO_DIR, "pusa_v1.pt")
 
-# Add PUSA_ROOT to sys.path so Python can find local modules like diffsynth
+# Add PUSA_ROOT to sys.path so Python can import diffsynth
 if PUSA_ROOT not in sys.path:
     sys.path.insert(0, PUSA_ROOT)
 
-# Validate presence of diffsynth
+# === Validate diffsynth ===
 DIFFSYNTH_PATH = os.path.join(PUSA_ROOT, "diffsynth")
 if not os.path.exists(DIFFSYNTH_PATH):
     raise RuntimeError(
@@ -29,8 +36,10 @@ if not os.path.exists(DIFFSYNTH_PATH):
         f"Ensure PusaV1 is correctly cloned and folder structure is intact."
     )
 
-# === Ensure model and weights are downloaded ===
+# === Ensure models exist, skip download if already present ===
 def ensure_model_downloaded():
+    print("🔍 Checking model presence...\n")
+
     if not os.path.exists(MODEL_ZOO_DIR):
         print("Downloading RaphaelLiu/PusaV1 to ./PusaV1/model_zoo ...")
         snapshot_download(
@@ -40,6 +49,8 @@ def ensure_model_downloaded():
             local_dir_use_symlinks=False,
         )
         print("✅ PusaV1 base model downloaded.")
+    else:
+        print("✅ PusaV1 base folder already exists.")
 
     if not os.path.exists(WAN_MODEL_PATH):
         print("Downloading Wan-AI/Wan2.1-T2V-14B to ./PusaV1/model_zoo/Wan2.1-T2V-14B ...")
@@ -50,14 +61,18 @@ def ensure_model_downloaded():
             local_dir_use_symlinks=False,
         )
         print("✅ Wan2.1-T2V-14B model downloaded.")
+    else:
+        print("✅ Wan2.1-T2V-14B folder already exists.")
 
     if not os.path.exists(LORA_PATH):
         raise FileNotFoundError(
-            f"Expected LoRA weights 'pusa_v1.pt' not found at {LORA_PATH}. "
-            f"Please ensure the file exists or is downloaded manually."
+            f"❌ Expected LoRA weights 'pusa_v1.pt' not found at {LORA_PATH}. "
+            f"Please make sure it exists in your repo."
         )
+    else:
+        print("✅ LoRA weights (pusa_v1.pt) found.")
 
-    # ✅ Add this to list all files in modelzoo
+    # === List contents of model_zoo for verification
     print(f"\n📁 Verifying files under: {MODEL_ZOO_DIR}\n")
     for root, dirs, files in os.walk(MODEL_ZOO_DIR):
         for file in files: