#!/usr/bin/env bash
# Copyright 2026 Umbra Tools (umbratools.com)
# SPDX-License-Identifier: Apache-2.0

# Downloads the Qwen3-TTS model weights required by UmbraAudio.
#
# These models are stock HuggingFace downloads (~9 GB). They are NOT tracked
# in git; run this script once after cloning to reconstruct the models/ tree.
#
# Usage:
#   bash scripts/download_models.sh [models_dir]
#
# Defaults to ./models relative to the UmbraAudio repo root.
set -euo pipefail

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
MODELS_DIR="${1:-$REPO_ROOT/models}"

mkdir -p "$MODELS_DIR"

# Qwen3-TTS models used by the configured engines (see umbraaudio/config.py and .env).
huggingface-cli download "Qwen/Qwen3-TTS-12Hz-0.6B-Base"        --local-dir "$MODELS_DIR/Qwen3-TTS-12Hz-0.6B-Base"
huggingface-cli download "Qwen/Qwen3-TTS-12Hz-0.6B-CustomVoice"  --local-dir "$MODELS_DIR/Qwen3-TTS-12Hz-0.6B-CustomVoice"
huggingface-cli download "Qwen/Qwen3-TTS-12Hz-1.7B-Base"         --local-dir "$MODELS_DIR/Qwen3-TTS-12Hz-1.7B-Base"
huggingface-cli download "Qwen/Qwen3-TTS-12Hz-1.7B-CustomVoice"  --local-dir "$MODELS_DIR/Qwen3-TTS-12Hz-1.7B-CustomVoice"
huggingface-cli download "Qwen/Qwen3-TTS-12Hz-1.7B-VoiceDesign"  --local-dir "$MODELS_DIR/Qwen3-TTS-12Hz-1.7B-VoiceDesign"
huggingface-cli download "Qwen/Qwen3-TTS-Tokenizer-12Hz"         --local-dir "$MODELS_DIR/Qwen3-TTS-Tokenizer-12Hz"

echo "Models downloaded to $MODELS_DIR"