#!/usr/bin/env bash
set -e

# 0) Make sure cache dir exists (Space volume mounted at runtime)
mkdir -p "${HF_HOME:-/data/.huggingface}"

# 1) One-shot warm-up (skipped after first boot)
python - <<'PY'
from huggingface_hub import snapshot_download
for repo in ("nari-labs/Dia-1.6B", "descriptinc/descript-audio-codec"):
    snapshot_download(repo, local_files_only=False)   # honours HF_HOME
PY

# 2) Start the Gradio app
exec python app.py