Update GhostPackDemo/install.py

GhostPackDemo/install.py

@@ -5,57 +5,56 @@ import sys
 import subprocess
 import shutil
 
-ENV_DIR = "venv"
-CACHE_DIR = "hf_download"
-REPO_MODEL = "hunyuanvideo-community/HunyuanVideo"
-
-def check_command(cmd):
-    return shutil.which(cmd) is not None
-
-def run(cmd, **kw):
+# Repo and subdir
+REPO_URL   = "https://huggingface.co/spaces/ghostai1/GhostPack.git"
+SUBDIR     = "GhostPackDemo"
+ENV_DIR    = "venv"
+CACHE_DIR  = "hf_download"
+MODEL_ID   = "hunyuanvideo-community/HunyuanVideo"
+
+def run(cmd):
     print(f"> {' '.join(cmd)}")
-    subprocess.check_call(cmd, **kw)
+    subprocess.check_call(cmd)
 
 def main():
-    # 1. Python version
-    if sys.version_info < (3,8):
-        print("Python 3.8+ is required.")
-        sys.exit(1)
+    # Clone or update repo
+    if not os.path.isdir("GhostPack"):
+        run(["git", "clone", REPO_URL])
+    os.chdir(os.path.join("GhostPack", SUBDIR))
 
-    # 2. CUDA / GPU check
-    if not check_command("nvidia-smi"):
-        print("Warning: nvidia-smi not found. CUDA GPU acceleration will be unavailable.")
-    else:
-        run(["nvidia-smi"])
+    # Python version check
+    if sys.version_info < (3,8):
+        sys.exit("Python 3.8+ required")
 
-    if not check_command("nvcc"):
-        print("Warning: nvcc not found. Install CUDA Toolkit for faster inference.")
+    # CUDA/GPU check
+    if shutil.which("nvidia-smi") is None:
+        print("Warning: nvidia-smi not found; CUDA will be unavailable")
+    if shutil.which("nvcc") is None:
+        print("Warning: nvcc not found; install CUDA toolkit for best performance")
 
-    # 3. create virtual env
+    # Virtual env
     if not os.path.isdir(ENV_DIR):
-        print("Creating virtual environment...")
         run([sys.executable, "-m", "venv", ENV_DIR])
 
     if os.name == "nt":
         pip = os.path.join(ENV_DIR, "Scripts", "pip.exe")
-        python = os.path.join(ENV_DIR, "Scripts", "python.exe")
+        py  = os.path.join(ENV_DIR, "Scripts", "python.exe")
     else:
         pip = os.path.join(ENV_DIR, "bin", "pip")
-        python = os.path.join(ENV_DIR, "bin", "python")
+        py  = os.path.join(ENV_DIR, "bin", "python")
 
-    # 4. upgrade pip & install
+    # Install dependencies
     run([pip, "install", "--upgrade", "pip"])
     run([pip, "install", "-r", "requirements.txt"])
 
-    # 5. HF login & cache model
+    # HF login + cache model
     os.makedirs(CACHE_DIR, exist_ok=True)
-    run([python, "-c", "from diffusers_helper.hf_login import login; login()"])
-    run([python, "-c",
-         f"from huggingface_hub import snapshot_download; snapshot_download('{REPO_MODEL}', cache_dir='{CACHE_DIR}')"])
+    run([py, "-c", "from diffusers_helper.hf_login import login; login()"])
+    run([py, "-c",
+         f"from huggingface_hub import snapshot_download; snapshot_download('{MODEL_ID}', cache_dir='{CACHE_DIR}')"])
 
-    print("\nSetup complete.\n")
-    print("To run GhostPackDemo:")
-    print(f"  source {ENV_DIR}/bin/activate && python ghostpack_main_fixed.py --server 0.0.0.0 --port 7860")
+    print("\nSetup complete.")
+    print(f"cd GhostPack/{SUBDIR} && source {ENV_DIR}/bin/activate && python ghostpack_main_fixed.py --server 0.0.0.0 --port 7860")
 
 if __name__ == "__main__":
     main()