#!/bin/bash

################################################################################
# UPDATE SCRIPT
# Actualiza código, modelos y dependencias
################################################################################

set -e

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
VENV_DIR="$PROJECT_DIR/venv"

log() { echo -e "${BLUE}[$(date '+%H:%M:%S')]${NC} $@"; }
log_success() { echo -e "${GREEN}✅ $@${NC}"; }
log_error() { echo -e "${RED}❌ $@${NC}"; exit 1; }

log "Actualizando LLM Server..."

# Pull latest from git
log "Descargando cambios de git..."
cd "$PROJECT_DIR"
git pull origin main || log "Git pull completado con warnings"

# Update Python dependencies
log "Actualizando dependencias Python..."
source "$VENV_DIR/bin/activate"
pip install --upgrade pip
pip install -r requirements.txt --upgrade

# Update Ollama models (opcional)
read -p "¿Actualizar modelos Ollama? (y/n): " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    log "Actualizando Phi..."
    ollama pull phi:latest
    
    log "Actualizando DeepSeek..."
    ollama pull deepseek-coder:33b
fi

# Restart services
log "Reiniciando servicios..."
sudo systemctl restart ollama
sudo systemctl restart llm-api

log_success "Actualización completada"