#!/bin/bash

# Heavy 2.0 - Setup Script
# This script creates a virtual environment and installs all dependencies

echo "🤖 Heavy 2.0 - Setup"
echo "===================="
echo ""

# Get the directory where the script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

echo "📍 Working directory: $SCRIPT_DIR"
echo ""

# Create virtual environment
echo "📦 Creating virtual environment..."
python3 -m venv venv

if [ $? -ne 0 ]; then
    echo "❌ Failed to create virtual environment"
    exit 1
fi

echo "✅ Virtual environment created"
echo ""

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

echo "✅ Virtual environment activated"
echo ""

# Upgrade pip
echo "⬆️  Upgrading pip..."
pip install --upgrade pip

echo ""

# Install dependencies
echo "📚 Installing dependencies from requirements.txt..."
echo "   This may take a few minutes..."
echo ""

pip install -r requirements.txt

if [ $? -ne 0 ]; then
    echo ""
    echo "❌ Failed to install dependencies"
    exit 1
fi

echo ""
echo "✅ All dependencies installed successfully!"
echo ""
echo "🎉 Setup complete!"
echo ""
echo "📝 Next steps:"
echo "   1. Activate the virtual environment:"
echo "      source venv/bin/activate"
echo ""
echo "   2. Run the application:"
echo "      python app.py"
echo ""
echo "   3. Open your browser to: http://localhost:7860"
echo ""
echo "🔑 Don't forget to get your API keys:"
echo "   - OpenRouter: https://openrouter.ai/keys"
echo "   - Tavily (optional): https://tavily.com"
echo ""