#!/bin/bash
# Setup script for HF Space deployment

echo "Installing HeartMuLa package..."
pip install -e .

echo "Downloading HeartMuLa models..."
mkdir -p ./models

# Download HeartMuLa 3B model (RL-tuned version - recommended)
echo "Downloading HeartMuLa-RL-oss-3B-20260123..."
huggingface-cli download \
  --local-dir "./models/HeartMuLa-oss-3B" \
  HeartMuLa/HeartMuLa-RL-oss-3B-20260123

# Download HeartCodec
echo "Downloading HeartCodec-oss-20260123..."
huggingface-cli download \
  --local-dir "./models/HeartCodec-oss" \
  HeartMuLa/HeartCodec-oss-20260123

# Copy tokenizer and config
echo "Copying tokenizer and config..."
huggingface-cli download \
  --local-dir "./models" \
  HeartMuLa/HeartMuLaGen

echo "Setup complete! Starting Gradio app..."