Update app.py

app.py

@@ -75,5 +75,18 @@ def test_rag_reranking(query,ranker):
 
 ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type='colbert')
 
-iface = gr.Chatbot()
+# Función de respuesta simple usando test_rag
+def chat_function(message, history):
+    respuesta, _ = test_rag(message)
+    return respuesta
+
+# Interfaz de chat
+iface = gr.ChatInterface(
+    fn=chat_function,
+    chatbot=gr.Chatbot(),
+    title="RAG Bot",
+    description="Hazme preguntas sobre el contenido del PDF.",
+)
+
+# Lanzar interfaz
 iface.launch()