start.sh

#!/bin/bash
# Quick-start script for the NLP-to-SQL Agentic Pipeline
# Run with: bash start.sh

set -e

echo "🚀 NLP-to-SQL Agentic Pipeline Setup"
echo "===================================="

# Check Python
if ! command -v python3 &> /dev/null; then
    echo "❌ Python 3 not found. Please install Python 3.9+"
    exit 1
fi

# Create virtualenv if it doesn't exist
if [ ! -d ".venv" ]; then
    echo "📦 Creating virtual environment..."
    python3 -m venv .venv
    source .venv/bin/activate
    echo "✓ Virtual environment created"
else
    echo "✓ Virtual environment already exists"
    source .venv/bin/activate
fi

# Install dependencies
echo "📥 Installing dependencies..."
pip install -q -r requirements.txt 2>&1 | tail -5
echo "✓ Dependencies installed"

# Create sample database
if [ ! -f "data.db" ]; then
    echo "🗄️  Creating sample database..."
    python3 setup_sample_db.py
else
    echo "✓ Sample database already exists"
fi

# Check API key
if [ -z "$OPENAI_API_KEY" ]; then
    echo ""
    echo "⚠️  OPENAI_API_KEY not set!"
    echo "Set it with: export OPENAI_API_KEY='sk-...'"
    echo ""
    read -p "Continue without API key? (y/n) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        exit 1
    fi
fi

# Menu
echo ""
echo "What would you like to do?"
echo "  1. Run tests (pytest)"
echo "  2. Run dry-run example"
echo "  3. Run integration examples"
echo "  4. Run Streamlit app"
echo "  5. All of the above"
echo ""
read -p "Choose option (1-5): " choice

case $choice in
    1)
        echo ""
        echo "🧪 Running tests..."
        python3 -m pytest test_agent.py -v
        ;;
    2)
        echo ""
        echo "▶️  Running dry-run example..."
        python3 dry_run.py
        ;;
    3)
        echo ""
        echo "📚 Running integration examples..."
        python3 integration_examples.py
        ;;
    4)
        echo ""
        echo "🎨 Starting Streamlit app..."
        echo "   Open http://localhost:8501"
        streamlit run app.py
        ;;
    5)
        echo ""
        echo "🧪 Running tests..."
        python3 -m pytest test_agent.py -v
        echo ""
        echo "▶️  Running dry-run example..."
        python3 dry_run.py
        echo ""
        echo "📚 Running integration examples..."
        python3 integration_examples.py
        echo ""
        echo "🎨 Starting Streamlit app..."
        streamlit run app.py
        ;;
    *)
        echo "Invalid option"
        exit 1
        ;;
esac

echo ""
echo "✅ Done!"