app.py

import os
import sys
import subprocess
import importlib.util

local_bin = os.path.expanduser("~/bin")
os.makedirs(local_bin, exist_ok=True)

# Build download URL
download_url = (
    "https://github.com/replicate/pget/releases/latest/download/pget_Linux_x86_64"
)
local_pget_path = os.path.join(local_bin, "pget")

# Download pget to ~/bin/pget
subprocess.run(["curl", "-L", "-o", local_pget_path, download_url], check=True)

# Make it executable
subprocess.run(["chmod", "+x", local_pget_path], check=True)

# Optional: add ~/bin to PATH (for subprocesses only)
os.environ["PATH"] = f"{local_bin}:" + os.environ["PATH"]


subprocess.run(
    ["git", "clone", "--recursive", os.environ["REPO_URL"], os.environ["REPO_DIR"]],
    check=True,
)
os.chdir(os.environ["REPO_DIR"])

os.chmod("./scripts/deploy_gradio_inference.sh", 0o755)
subprocess.run(["./scripts/deploy_gradio_inference.sh", "image"], check=True)

os.chdir("inference")
subprocess.run(["pip", "install", "-r", "requirements.txt"], check=True)

sys.path.append(os.getcwd())

# Add the *parent* directory of inference to sys.path so that we can inference.*
sys.path.append(os.path.abspath(".."))

spec = importlib.util.spec_from_file_location("__main__", "app.py")
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)