#!/bin/bash
#
# Download AI Models
# Wrapper for Ansible models tag
#

set -e

cd "$(dirname "$0")/.."

echo "========================================="
echo "  Downloading AI Models (~37GB)"
echo "========================================="
echo ""

# Source .env if it exists
if [ -f .env ]; then
    set -a
    source .env
    set +a
fi

# Check HF_TOKEN
if [ -z "$HF_TOKEN" ]; then
    echo "Error: HF_TOKEN not set"
    echo "Add HF_TOKEN to .env file"
    exit 1
fi

# Run Ansible with models tag
ansible-playbook playbook.yml --tags models

echo ""
echo "========================================="
echo "  Model download complete!"
echo "========================================="