#!/bin/bash
# Quick script to toggle between local and remote Ollama configuration

ENV_FILE=".env"
BACKUP_FILE=".env.backup"

if [ ! -f "$ENV_FILE" ]; then
    echo "❌ .env file not found!"
    exit 1
fi

# Backup current .env
cp "$ENV_FILE" "$BACKUP_FILE"

# Check current environment
CURRENT_ENV=$(grep "^ENVIRONMENT=" "$ENV_FILE" | cut -d'=' -f2)

if [ "$CURRENT_ENV" = "local" ]; then
    echo "🔄 Switching to REMOTE configuration..."
    
    # Switch to remote
    sed -i 's/^OLLAMA_HOST=localhost/OLLAMA_HOST=10.0.30.63/' "$ENV_FILE"
    sed -i 's/^OLLAMA_MODEL=llama3:latest/OLLAMA_MODEL=llama3.1:8b/' "$ENV_FILE"
    sed -i 's/^OLLAMA_WORK_MODEL=llama3:latest/OLLAMA_WORK_MODEL=llama3.1:8b/' "$ENV_FILE"
    sed -i 's/^OLLAMA_FAMILY_MODEL=llama3:latest/OLLAMA_FAMILY_MODEL=phi3:mini-q4_0/' "$ENV_FILE"
    sed -i 's/^ENVIRONMENT=local/ENVIRONMENT=remote/' "$ENV_FILE"
    
    echo "✅ Switched to REMOTE (10.0.30.63)"
    echo "   Model: llama3.1:8b (work), phi3:mini-q4_0 (family)"
    
else
    echo "🔄 Switching to LOCAL configuration..."
    
    # Switch to local
    sed -i 's/^OLLAMA_HOST=10.0.30.63/OLLAMA_HOST=localhost/' "$ENV_FILE"
    sed -i 's/^OLLAMA_MODEL=llama3.1:8b/OLLAMA_MODEL=llama3:latest/' "$ENV_FILE"
    sed -i 's/^OLLAMA_WORK_MODEL=llama3.1:8b/OLLAMA_WORK_MODEL=llama3:latest/' "$ENV_FILE"
    sed -i 's/^OLLAMA_FAMILY_MODEL=phi3:mini-q4_0/OLLAMA_FAMILY_MODEL=llama3:latest/' "$ENV_FILE"
    sed -i 's/^ENVIRONMENT=remote/ENVIRONMENT=local/' "$ENV_FILE"
    
    echo "✅ Switched to LOCAL (localhost:11434)"
    echo "   Model: llama3:latest"
fi

echo ""
echo "📝 Current configuration:"
grep "^OLLAMA_" "$ENV_FILE" | grep -v "^#"
grep "^ENVIRONMENT=" "$ENV_FILE"