#!/usr/bin/env bash
# Quick setup script for development

set -e

echo "=================================================="
echo "Autonomous Traffic Control - Development Setup"
echo "=================================================="
echo

# Check Python version
python_version=$(python --version 2>&1 | awk '{print $2}')
echo "✓ Python version: $python_version"

# Create virtual environment
if [ ! -d "venv" ]; then
    echo "Creating virtual environment..."
    python -m venv venv
fi

# Activate virtual environment
echo "Activating virtual environment..."
source venv/bin/activate 2>/dev/null || . venv/Scripts/activate 2>/dev/null

# Upgrade pip
echo "Upgrading pip..."
pip install --upgrade pip --quiet

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

# Install development dependencies
echo "Installing development dependencies..."
pip install black flake8 pytest --quiet

# Create .env from template
if [ ! -f ".env" ]; then
    echo "Creating .env file from template..."
    cp .env.example .env
    echo "⚠  Please edit .env with your API credentials"
fi

echo
echo "=================================================="
echo "Setup Complete!"
echo "=================================================="
echo
echo "Next steps:"
echo "  1. Edit .env with your API credentials"
echo "  2. Run: python inference.py"
echo "  3. Run: streamlit run dashboard_final.py"
echo "  4. Run: python validate.py"
echo