#!/bin/bash
set -e

echo "🚀 AutoTrain Advanced - Optimized Deployment"
echo "============================================="

# Check for HF_TOKEN
if [ -z "$HF_TOKEN" ]; then
    echo "⚠️  Warning: HF_TOKEN not set"
    echo "   Set it with: export HF_TOKEN=your_token_here"
fi

# Build
echo "📦 Building optimized Docker image..."
docker build -t autotrain-advanced-optimized .

# Run locally for testing
echo "🧪 Starting local server..."
echo "   URL: http://localhost:7860"
echo "   Press Ctrl+C to stop"
echo ""

docker run -it --rm \
    -p 7860:7860 \
    -e WORKERS=2 \
    -e HF_HUB_ENABLE_HF_TRANSFER=1 \
    -e HF_TOKEN="${HF_TOKEN}" \
    -v "$(pwd)/data:/data" \
    autotrain-advanced-optimized