#!/bin/bash
# Simple deployment script

set -e

echo "🚀 Defect Detection Deployment"
echo "=============================="

# Check for model
if [ ! -f "models/best_model.pth" ] && [ ! -f "models/toy_model.pth" ]; then
    echo "⚠️  No model found. Training toy model for testing..."
    python train.py --toy --epochs 2
fi

# Select deployment method
echo "Select deployment method:"
echo "1) Vercel (Serverless)"
echo "2) Docker (Container)"
echo "3) Local API"
echo "4) Local Web UI"

read -p "Choice [1-4]: " choice

case $choice in
    1)
        echo "📦 Deploying to Vercel..."
        vercel --prod
        ;;
    2)
        echo "🐳 Building Docker image..."
        docker build -t defect-detection .
        echo "✅ Run with: docker run -p 8000:8000 defect-detection"
        ;;
    3)
        echo "🚀 Starting API server..."
        python app.py --mode api
        ;;
    4)
        echo "🌐 Starting Web UI..."
        python app.py --mode web
        ;;
    *)
        echo "Invalid choice"
        exit 1
        ;;
esac