test_gpt.py

#!/usr/bin/env python3
"""
Script de prueba para verificar que el GPT Local funciona
"""

import sys
from pathlib import Path

# Agregar el directorio raíz al path
sys.path.append(str(Path(__file__).parent))

try:
    from models.model_loader import ModelLoader
    from models.text_generator import TextGenerator
    
    print("🚀 Probando GPT Local...")
    
    # Inicializar componentes
    model_loader = ModelLoader()
    text_generator = TextGenerator(model_loader)
    
    # Cargar modelo pequeño
    print("📦 Cargando modelo DialoGPT-small...")
    success = model_loader.load_model("microsoft/DialoGPT-small")
    
    if success:
        print("✅ Modelo cargado exitosamente")
        
        # Prueba básica
        test_message = "Hola"
        print(f"🧪 Probando con: '{test_message}'")
        
        response = text_generator.generate_response(test_message, max_length=50)
        print(f"🤖 Respuesta: {response}")
        
        print("\n✅ ¡GPT Local funcionando correctamente!")
        
    else:
        print("❌ Error al cargar el modelo")
        
except Exception as e:
    print(f"❌ Error: {str(e)}")
    import traceback
    traceback.print_exc()