run.sh

#!/bin/bash

# AI Documentation Assistant - Run Script

echo "🤖 AI Documentation Assistant"
echo "=============================="
echo ""

# Check if .env file exists
if [ ! -f .env ]; then
    echo "⚠️  No .env file found. Creating from .env.example..."
    cp .env.example .env
    echo "✅ Created .env file. Please update it with your API keys."
    echo ""
    echo "Required API keys:"
    echo "  - OPENAI_API_KEY"
    echo "  - PINECONE_API_KEY"
    echo "  - PINECONE_ENVIRONMENT"
    echo ""
    echo "After updating .env, run this script again."
    exit 1
fi

# Function to check if command exists
command_exists() {
    command -v "$1" >/dev/null 2>&1
}

# Check Python version
if command_exists python3; then
    PYTHON=python3
elif command_exists python; then
    PYTHON=python
else
    echo "❌ Python not found. Please install Python 3.9 or higher."
    exit 1
fi

echo "📍 Using Python: $($PYTHON --version)"
echo ""

# Create virtual environment if it doesn't exist
if [ ! -d "venv" ]; then
    echo "🔧 Creating virtual environment..."
    $PYTHON -m venv venv
fi

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

# Install/upgrade pip
echo "🔧 Upgrading pip..."
pip install --upgrade pip

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

# Initialize database
echo "🗄️  Initializing database..."
python scripts/init_db.py

# Start services based on argument
case "$1" in
    "api")
        echo "🚀 Starting API server..."
        python -m src.api.main
        ;;
    "ui")
        echo "🎨 Starting Streamlit UI..."
        streamlit run src/ui/app.py
        ;;
    "docker")
        echo "🐳 Starting with Docker Compose..."
        docker-compose up --build
        ;;
    "all")
        echo "🚀 Starting all services..."
        # Start API in background
        python -m src.api.main &
        API_PID=$!
        
        # Wait a bit for API to start
        sleep 3
        
        # Start UI
        streamlit run src/ui/app.py &
        UI_PID=$!
        
        echo ""
        echo "✅ Services started:"
        echo "   - API: http://localhost:8000"
        echo "   - UI: http://localhost:8501"
        echo ""
        echo "Press Ctrl+C to stop all services..."
        
        # Wait for interrupt
        trap "kill $API_PID $UI_PID; exit" INT
        wait
        ;;
    *)
        echo "Usage: ./run.sh [api|ui|docker|all]"
        echo ""
        echo "Options:"
        echo "  api    - Start only the FastAPI backend"
        echo "  ui     - Start only the Streamlit UI"
        echo "  docker - Start with Docker Compose"
        echo "  all    - Start both API and UI locally"
        echo ""
        echo "Example: ./run.sh all"
        ;;
esac