loader.py

import os
import shutil
import subprocess
import sys
from huggingface_hub import snapshot_download

PRIVATE_REPO_ID = "HongshengY/NeuralSketch2Surf"
HF_TOKEN = os.environ.get("HF_TOKEN")

APP_DIR = "/tmp/private_app"

def main():
    if not HF_TOKEN:
        print("HF_TOKEN not detected")
        sys.exit(1)

    print(f"Loading latest code from private repository {PRIVATE_REPO_ID}...")

    try:
        if os.path.exists(APP_DIR):
            shutil.rmtree(APP_DIR)

        snapshot_download(
           repo_id=PRIVATE_REPO_ID,
           repo_type="space",
           revision="main",
           local_dir=APP_DIR,
           token=HF_TOKEN,
           force_download=True,
        )

        print("Code and model download complete!")

        os.chdir(APP_DIR)

        if os.path.exists("backend/requirements.txt"):
            print("Installing backend/requirements.txt...")
            subprocess.run(
                [sys.executable, "-m", "pip", "install", "-r", "backend/requirements.txt"],
                check=True
            )

        print("Starting NeuralSketch2Surf Web...")

        cmd = [
            "uvicorn",
            "backend.server:app",
            "--host", "0.0.0.0",
            "--port", "7860"
        ]

        os.execvp("uvicorn", cmd)

    except Exception as e:
        print(f"An error occurred: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()