#!/bin/bash
# One-shot MiniMax H3 setup for a fresh RunPod pod that already has ComfyUI at
# /ComfyUI (see install-comfy.sh). Idempotent.
#
# Usage (from your machine):
#   scp -O -P <port> scripts/pod/{install-minimax-h3.sh,download-weights.sh,download-weights-tmpfs.sh,extra_model_paths.yaml,make-test-images.py} root@<pod>:/root/
#   ssh -p <port> root@<pod> 'bash /root/install-minimax-h3.sh'
set -euo pipefail

HERE="$(cd "$(dirname "$0")" && pwd)"

# deps
pip install -q --break-system-packages hf_transfer 2>/dev/null || pip install -q hf_transfer

# layout + scripts on the persistent volume
mkdir -p /workspace/minimax-h3/models
cp "$HERE"/download-weights.sh "$HERE"/download-weights-tmpfs.sh \
   "$HERE"/make-test-images.py /workspace/minimax-h3/ 2>/dev/null || true
cp "$HERE"/start-comfyui.sh /workspace/minimax-h3/ 2>/dev/null || true

# wire model paths into ComfyUI
cp "$HERE"/extra_model_paths.yaml /ComfyUI/extra_model_paths.yaml

# test/reference images
python3 /workspace/minimax-h3/make-test-images.py

# weights (phase 1 makes ComfyUI usable; later phases are the bigger families)
tmux new -s downloads -d 'bash /workspace/minimax-h3/download-weights.sh' 2>/dev/null \
  || echo "downloads session already exists"

# ComfyUI headless
bash /workspace/minimax-h3/start-comfyui.sh

echo "Done. Watch downloads: tmux attach -t downloads ; tail -f /workspace/minimax-h3/download.log"