Update app.py

app.py

@@ -1,47 +1,34 @@
 import gradio as gr
 from transformers import pipeline
 
-# Modelo (ligero)
-chatbot = pipeline("text-generation", model="gpt2")
+# Cargar modelo ligero
+generator = pipeline("text-generation", model="gpt2")
 
-def responder(mensaje, historial):
-    historial = historial or []
+def responder(message, history):
+    history = history or []
 
-    # Crear prompt con historial
-    prompt = "Eres Intelarya.ai, una IA educativa que explica paso a paso de forma clara.\n\n"
+    prompt = "Eres Intelarya.ai, una IA educativa que explica claro y paso a paso.\n\n"
 
-    for msg in historial:
-        if msg["role"] == "user":
-            prompt += f"Usuario: {msg['content']}\n"
-        else:
-            prompt += f"Intelarya: {msg['content']}\n"
+    for h in history:
+        prompt += f"Usuario: {h[0]}\nIntelarya: {h[1]}\n"
 
-    prompt += f"Usuario: {mensaje}\nIntelarya:"
+    prompt += f"Usuario: {message}\nIntelarya:"
 
-    respuesta = chatbot(
-        prompt,
-        max_length=200,
-        do_sample=True,
-        temperature=0.7
-    )[0]["generated_text"]
+    result = generator(prompt, max_length=150, do_sample=True, temperature=0.7)
+    response = result[0]["generated_text"].split("Intelarya:")[-1].strip()
 
-    texto = respuesta.split("Intelarya:")[-1].strip()
-
-    # Agregar al historial en formato correcto
-    historial.append({"role": "user", "content": mensaje})
-    historial.append({"role": "assistant", "content": texto})
-
-    return historial, historial
+    history.append((message, response))
+    return history, history
 
 with gr.Blocks() as demo:
     gr.Markdown("# 💎 intelarya.ai")
     gr.Markdown("Aprende más rápido. Entiende mejor. Sin complicaciones.")
 
-    chatbot_ui = gr.Chatbot(type="messages")
-    msg = gr.Textbox(placeholder="Escribe tu tarea aquí...")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
 
-    estado = gr.State([])
+    state = gr.State([])
 
-    msg.submit(responder, [msg, estado], [chatbot_ui, estado])
+    msg.submit(responder, [msg, state], [chatbot, state])
 
 demo.launch()