#!/usr/bin/env python3
"""Bootstrap: downloads source from HF repo and runs the full build."""
import subprocess, sys, os
subprocess.check_call([sys.executable, "-m", "pip", "install", "-q",
    "huggingface_hub", "jax", "jaxlib", "flax", "optax", "numpy"])
from huggingface_hub import hf_hub_download
repo = "gamer729/cognithread-ssm"
files = ["cognithread_full.py"]
for f in files:
    path = hf_hub_download(repo_id=repo, filename=f, repo_type="model")
    os.system(f"cp {path} /app/{f}")
    print(f"Downloaded {f}")
os.system(f"{sys.executable} /app/cognithread_full.py")