#!/bin/bash

# Script to configure Ollama settings for Docker Compose or external server

echo "Ollama Configuration Helper"
echo "============================"
echo ""
echo "Choose your Ollama setup:"
echo "1) Docker Compose (Ollama runs in container)"
echo "2) External Server (Ollama runs on host machine)"
echo ""
read -p "Enter choice [1-2]: " choice

ENV_FILE="backend/.env"

if [ ! -f "$ENV_FILE" ]; then
    echo "Error: $ENV_FILE not found!"
    exit 1
fi

case $choice in
    1)
        echo "Configuring for Docker Compose..."
        # Update OLLAMA_BASE_URL to use internal Docker network
        if grep -q "OLLAMA_BASE_URL=" "$ENV_FILE"; then
            sed -i.bak 's|OLLAMA_BASE_URL=.*|OLLAMA_BASE_URL=http://ollama:11434|' "$ENV_FILE"
        else
            echo "OLLAMA_BASE_URL=http://ollama:11434" >> "$ENV_FILE"
        fi
        echo "✓ Updated OLLAMA_BASE_URL to http://ollama:11434"
        echo ""
        echo "Next steps:"
        echo "1. Start services: docker-compose up -d"
        echo "2. Wait for model download: docker-compose logs -f ollama-setup"
        echo "3. Test: docker-compose exec crawler python crawler_service.py 1"
        ;;
    2)
        echo "Configuring for external Ollama server..."
        # Update OLLAMA_BASE_URL to use host machine
        if grep -q "OLLAMA_BASE_URL=" "$ENV_FILE"; then
            sed -i.bak 's|OLLAMA_BASE_URL=.*|OLLAMA_BASE_URL=http://host.docker.internal:11434|' "$ENV_FILE"
        else
            echo "OLLAMA_BASE_URL=http://host.docker.internal:11434" >> "$ENV_FILE"
        fi
        echo "✓ Updated OLLAMA_BASE_URL to http://host.docker.internal:11434"
        echo ""
        echo "Next steps:"
        echo "1. Install Ollama: https://ollama.ai/download"
        echo "2. Pull model: ollama pull phi3:latest"
        echo "3. Start Ollama: ollama serve"
        echo "4. Start services: docker-compose up -d"
        ;;
    *)
        echo "Invalid choice!"
        exit 1
        ;;
esac

echo ""
echo "Configuration complete!"