Update GhostPackDemo/install.py

GhostPackDemo/install.py

@@ -1,4 +1,6 @@
-# install.py
+# ==========================================================
+# FILE: install.py
+# ==========================================================
 #!/usr/bin/env python3
 import os, sys, subprocess, shutil, platform, venv
 
@@ -6,19 +8,19 @@ ENV_DIR   = "venv"
 CACHE_DIR = "hf_download"
 MODEL_ID  = "hunyuanvideo-community/HunyuanVideo"
 
-def run(cmd: list[str]):
+def run(cmd: list[str]) -> None:
     print("❯", *cmd, flush=True)
     subprocess.check_call(cmd)
 
-def ensure_python():
+def ensure_python() -> None:
     if sys.version_info < (3, 8):
-        sys.exit("❌ Python 3.8 or newer required.")
+        sys.exit("Python 3.8 or newer required.")
 
-def cuda_info():
+def cuda_info() -> None:
     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).")
+        print("⚠️  CUDA GPU not detected – falling back to CPU (slow).")
 
 def create_venv() -> tuple[str, str]:
     if not os.path.isdir(ENV_DIR):
@@ -32,14 +34,14 @@ def create_venv() -> tuple[str, str]:
         py  = os.path.join(ENV_DIR, "bin", "python")
     return pip, py
 
-def install_deps(pip: str):
+def install_reqs(pip: str) -> None:
     run([pip, "install", "--upgrade", "pip"])
     run([pip, "install", "-r", "requirements.txt"])
 
-def hf_login(py: str):
+def hf_login(py: str) -> None:
     run([py, "-c", "from diffusers_helper.hf_login import login; login()"])
 
-def download_model(py: str):
+def cache_model(py: str) -> None:
     os.makedirs(CACHE_DIR, exist_ok=True)
     code = (
         f"from huggingface_hub import snapshot_download;"
@@ -47,18 +49,18 @@ def download_model(py: str):
     )
     run([py, "-c", code])
 
-def main():
+def main() -> None:
     ensure_python()
     cuda_info()
     pip, py = create_venv()
-    install_deps(pip)
+    install_reqs(pip)
     hf_login(py)
-    download_model(py)
-    print("\n✅ Setup complete. Activate with:")
+    cache_model(py)
     if platform.system() == "Windows":
-        print(f"   {ENV_DIR}\\Scripts\\Activate.ps1 && python ghostpack_main_fixed.py --inbrowser")
+        print(f"\nActivate: {ENV_DIR}\\Scripts\\Activate.ps1")
     else:
-        print(f"   source {ENV_DIR}/bin/activate && python ghostpack_main_fixed.py --inbrowser")
+        print(f"\nActivate: source {ENV_DIR}/bin/activate")
+    print("Run:      python ghostpack.py --inbrowser")
 
 if __name__ == "__main__":
     main()