File size: 2,405 Bytes
22ca508 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
#!/usr/bin/env python3
"""
Versión simple del GPT Local sin Gradio
"""
import sys
import logging
from pathlib import Path
# Agregar el directorio raíz al path
sys.path.append(str(Path(__file__).parent))
from models.model_loader import ModelLoader
from models.text_generator import TextGenerator
from config.settings import DEFAULT_MODEL
def setup_logging():
"""Configurar el sistema de logging"""
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
def main():
"""Función principal para el chat en terminal"""
setup_logging()
logger = logging.getLogger(__name__)
print("🤖 GPT Local - Chat en Terminal")
print("="*50)
# Inicializar componentes
model_loader = ModelLoader()
text_generator = TextGenerator(model_loader)
# Cargar modelo por defecto
print(f"📦 Cargando modelo: {DEFAULT_MODEL}")
if not model_loader.load_model(DEFAULT_MODEL):
print("❌ Error al cargar el modelo")
return
print("✅ Modelo cargado exitosamente")
print("💡 Escribe 'salir' para terminar")
print("💡 Escribe 'reset' para reiniciar el chat")
print("-"*50)
try:
while True:
# Obtener entrada del usuario
user_input = input("\n👤 Tú: ").strip()
if user_input.lower() in ['salir', 'exit', 'quit']:
print("👋 ¡Hasta luego!")
break
if user_input.lower() == 'reset':
text_generator.reset_chat_history()
print("🔄 Chat reiniciado")
continue
if not user_input:
continue
# Generar respuesta
print("🤖 GPT: ", end="", flush=True)
try:
response = text_generator.generate_response(
user_input,
temperature=0.7,
max_length=200
)
print(response)
except Exception as e:
print(f"❌ Error: {str(e)}")
except KeyboardInterrupt:
print("\n👋 Chat interrumpido por el usuario")
except Exception as e:
logger.error(f"❌ Error en el chat: {str(e)}")
if __name__ == "__main__":
main()
|