#!/usr/bin/env bash
# Entrypoint for the LTX Director I2V Space.
# Downloads LTX-2.3 distilled fp8 weights on each cold boot (no persistent disk),
# then launches ComfyUI on port 7860.
set -euo pipefail

COMFY_DIR="/home/user/app/ComfyUI"
cd "$COMFY_DIR"

export HF_HUB_ENABLE_HF_TRANSFER=1

mkdir -p models/checkpoints models/text_encoders

echo "==> Downloading LTX-2.3 distilled fp8 checkpoint (~29GB)..."
if [ ! -f "models/checkpoints/ltx-2.3-22b-distilled-fp8.safetensors" ]; then
  hf download Lightricks/LTX-2.3-fp8 \
    ltx-2.3-22b-distilled-fp8.safetensors \
    --local-dir models/checkpoints
else
  echo "    checkpoint already present, skipping."
fi

echo "==> Downloading Gemma text encoder (~8GB)..."
TE_DIR="models/text_encoders/gemma-3-12b-it-qat-q4_0-unquantized"
if [ ! -d "$TE_DIR" ] || [ -z "$(ls -A "$TE_DIR" 2>/dev/null)" ]; then
  hf download google/gemma-3-12b-it-qat-q4_0-unquantized \
    --local-dir "$TE_DIR"
else
  echo "    text encoder already present, skipping."
fi

echo "==> Starting ComfyUI on 0.0.0.0:7860 ..."
exec python main.py --listen 0.0.0.0 --port 7860