#!/bin/bash

# Setup script for local development

set -e

echo "🔧 Setting up Image Sampling & Quantization Demo"
echo "==============================================="
echo ""

# Check Python version
echo "🐍 Checking Python version..."
PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}')
echo "   Found Python $PYTHON_VERSION"

REQUIRED_VERSION="3.11"
if [ "$(printf '%s\n' "$REQUIRED_VERSION" "$PYTHON_VERSION" | sort -V | head -n1)" != "$REQUIRED_VERSION" ]; then
    echo "⚠️  Warning: Python $REQUIRED_VERSION or higher is recommended."
fi

echo ""

# Create virtual environment
if [ -d "venv" ]; then
    echo "📦 Virtual environment already exists."
    read -p "   Do you want to recreate it? (y/N): " RECREATE
    if [ "$RECREATE" = "y" ] || [ "$RECREATE" = "Y" ]; then
        echo "   Removing old virtual environment..."
        rm -rf venv
        python3 -m venv venv
    fi
else
    echo "📦 Creating virtual environment..."
    python3 -m venv venv
fi

# Activate virtual environment
echo "🔧 Activating virtual environment..."
source venv/bin/activate

# Upgrade pip
echo "⬆️  Upgrading pip..."
pip install --upgrade pip > /dev/null 2>&1

# Install requirements
echo "📥 Installing requirements..."
pip install -r requirements.txt

echo ""
echo "✅ Setup complete!"
echo ""
echo "📚 Next steps:"
echo "   1. Run './run_simple.sh' to start the demo locally"
echo "   2. Or activate the environment: 'source venv/bin/activate'"
echo "   3. Then run: 'streamlit run app.py'"
echo ""
echo "🚀 To deploy to Hugging Face Spaces:"
echo "   1. Create a Space at https://huggingface.co/new-space"
echo "   2. Run './deploy.sh' and follow the prompts"