Update GhostPackDemo/install.py

GhostPackDemo/install.py

@@ -1,60 +1,64 @@
 # install.py
 #!/usr/bin/env python3
-import os
-import sys
-import subprocess
-import shutil
-
-# 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)}")
+import os, sys, subprocess, shutil, platform, venv
+
+ENV_DIR   = "venv"
+CACHE_DIR = "hf_download"
+MODEL_ID  = "hunyuanvideo-community/HunyuanVideo"
+
+def run(cmd: list[str]):
+    print("❯", *cmd, flush=True)
     subprocess.check_call(cmd)
 
-def main():
-    # Clone or update repo
-    if not os.path.isdir("GhostPack"):
-        run(["git", "clone", REPO_URL])
-    os.chdir(os.path.join("GhostPack", SUBDIR))
-
-    # Python version check
-    if sys.version_info < (3,8):
-        sys.exit("Python 3.8+ required")
-
-    # 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")
-
-    # Virtual env
-    if not os.path.isdir(ENV_DIR):
-        run([sys.executable, "-m", "venv", ENV_DIR])
+def ensure_python():
+    if sys.version_info < (3, 8):
+        sys.exit("❌ Python 3.8 or newer required.")
+
+def cuda_info():
+    if shutil.which("nvidia-smi"):
+        run(["nvidia-smi", "--query-gpu=name,memory.total,driver_version", "--format=csv"])
+    else:
+        print("⚠️  CUDA GPU not found – fall-back to CPU (slow).")
 
-    if os.name == "nt":
+def create_venv() -> tuple[str, str]:
+    if not os.path.isdir(ENV_DIR):
+        print("📦 Creating virtual environment…")
+        venv.create(ENV_DIR, with_pip=True)
+    if platform.system() == "Windows":
         pip = os.path.join(ENV_DIR, "Scripts", "pip.exe")
         py  = os.path.join(ENV_DIR, "Scripts", "python.exe")
     else:
         pip = os.path.join(ENV_DIR, "bin", "pip")
         py  = os.path.join(ENV_DIR, "bin", "python")
+    return pip, py
 
-    # Install dependencies
+def install_deps(pip: str):
     run([pip, "install", "--upgrade", "pip"])
     run([pip, "install", "-r", "requirements.txt"])
 
-    # HF login + cache model
-    os.makedirs(CACHE_DIR, exist_ok=True)
+def hf_login(py: str):
     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.")
-    print(f"cd GhostPack/{SUBDIR} && source {ENV_DIR}/bin/activate && python ghostpack_main_fixed.py --server 0.0.0.0 --port 7860")
+def download_model(py: str):
+    os.makedirs(CACHE_DIR, exist_ok=True)
+    code = (
+        f"from huggingface_hub import snapshot_download;"
+        f"snapshot_download('{MODEL_ID}', cache_dir='{CACHE_DIR}', resume_download=True)"
+    )
+    run([py, "-c", code])
+
+def main():
+    ensure_python()
+    cuda_info()
+    pip, py = create_venv()
+    install_deps(pip)
+    hf_login(py)
+    download_model(py)
+    print("\n✅ Setup complete. Activate with:")
+    if platform.system() == "Windows":
+        print(f"   {ENV_DIR}\\Scripts\\Activate.ps1 && python ghostpack_main_fixed.py --inbrowser")
+    else:
+        print(f"   source {ENV_DIR}/bin/activate && python ghostpack_main_fixed.py --inbrowser")
 
 if __name__ == "__main__":
     main()