#!/bin/bash
set -e

echo "========================================="
echo "Writing Studio - Setup Script"
echo "========================================="
echo ""

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

# Create virtual environment
echo ""
echo "Creating virtual environment..."
if [ ! -d "venv" ]; then
    python3 -m venv venv
    echo "Virtual environment created."
else
    echo "Virtual environment already exists."
fi

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

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

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

# Install in development mode
echo ""
echo "Installing package in development mode..."
pip install -e ".[dev]"

# Copy environment file if not exists
echo ""
if [ ! -f ".env" ]; then
    echo "Creating .env file from .env.example..."
    cp .env.example .env
    echo ".env file created. Please review and update as needed."
else
    echo ".env file already exists."
fi

# Setup pre-commit hooks
echo ""
echo "Setting up pre-commit hooks..."
pre-commit install
echo "Pre-commit hooks installed."

# Create necessary directories
echo ""
echo "Creating necessary directories..."
mkdir -p logs models

echo ""
echo "========================================="
echo "Setup complete!"
echo "========================================="
echo ""
echo "Next steps:"
echo "1. Review and update .env file with your settings"
echo "2. Activate the virtual environment: source venv/bin/activate"
echo "3. Run the application: make run"
echo "   Or: python -m writing_studio.main"
echo ""
echo "For Docker deployment:"
echo "  docker-compose up"
echo ""
echo "For more information, see README.md"
echo ""