Create install.py

GhostPackDemo/install.py

@@ -0,0 +1,61 @@
+# install.py
+#!/usr/bin/env python3
+import os
+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):
+    print(f"> {' '.join(cmd)}")
+    subprocess.check_call(cmd, **kw)
+
+def main():
+    # 1. Python version
+    if sys.version_info < (3,8):
+        print("Python 3.8+ is required.")
+        sys.exit(1)
+
+    # 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"])
+
+    if not check_command("nvcc"):
+        print("Warning: nvcc not found. Install CUDA Toolkit for faster inference.")
+
+    # 3. create 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")
+    else:
+        pip = os.path.join(ENV_DIR, "bin", "pip")
+        python = os.path.join(ENV_DIR, "bin", "python")
+
+    # 4. upgrade pip & install
+    run([pip, "install", "--upgrade", "pip"])
+    run([pip, "install", "-r", "requirements.txt"])
+
+    # 5. 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}')"])
+
+    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")
+
+if __name__ == "__main__":
+    main()