#!/bin/bash

# Script de instalación y configuración para GPT Local

echo "🚀 Configurando GPT Local..."

# Verificar Python
echo "🐍 Verificando Python..."
if ! command -v python3 &> /dev/null; then
    echo "❌ Python3 no encontrado. Instálalo primero."
    exit 1
fi

echo "✅ Python encontrado: $(python3 --version)"

# Crear entorno virtual (opcional)
read -p "¿Crear entorno virtual? (y/N): " create_venv
if [[ $create_venv =~ ^[Yy]$ ]]; then
    echo "📦 Creando entorno virtual..."
    python3 -m venv venv
    source venv/bin/activate
    echo "✅ Entorno virtual activado"
fi

# Instalar dependencias
echo "📦 Instalando dependencias..."
pip3 install -r requirements.txt

# Verificar instalación
echo "🔍 Verificando instalación..."
python3 -c "
import torch, transformers, gradio
print('✅ Todas las dependencias instaladas')
print(f'PyTorch: {torch.__version__}')
print(f'Transformers: {transformers.__version__}')
print(f'Gradio: {gradio.__version__}')
"

# Crear directorios necesarios
echo "📁 Creando directorios..."
mkdir -p models_cache
mkdir -p logs

# Prueba rápida
echo "🧪 Ejecutando prueba..."
python3 test_gpt.py

echo "🎉 ¡Configuración completada!"
echo ""
echo "Para ejecutar:"
echo "  Chat en terminal: python3 chat_terminal.py"
echo "  Interfaz web: python3 main.py"
echo "  Utilidades: python3 utils.py check"