Añadir memoria de conversación a AURA

Hacer que AURA pueda tener un seguimiento de las conversaciones y reccuerde mensajes pasadps

app.py

@@ -3,11 +3,28 @@ from transformers import pipeline
 
 generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
 
+# Variable para almacenar el historial de conversación
+historial = ""
+
 def responder(texto_usuario):
-    prompt = f"Tú eres AURA, un asistente emocional escolar muy amable. Ayuda a los estudiantes a sentirse escuchados sin juzgar. Si alguien dice: '{texto_usuario}', tú respondes:"
-    resultado_completo = generator(prompt, max_length=50)[0]["text"]
-    respuesta_limpia = resultado_completo[len(prompt):].strip()
-    return respuesta_limpia
+    global historial
+    prompt_inicial = "Tú eres AURA, un asistente emocional escolar muy amable. Escucha sin juzgar y responde con empatía."
+
+    # Construir el nuevo prompt con el historial
+    prompt = (
+        f"{prompt_inicial}\n"
+        f"{historial}"
+        f"Usuario: {texto_usuario}\n"
+        f"AURA:"
+    )
+
+    resultado_completo = generator(prompt, max_length=100)[0]["text"]
+    respuesta = resultado_completo[len(prompt):].strip()
+
+    # Actualizar el historial
+    historial += f"Usuario: {texto_usuario}\nAURA: {respuesta}\n"
+
+    return respuesta
 
 
 iface = gr.Interface(