Update app.py

app.py

@@ -31,24 +31,36 @@ chain = llm | StrOutputParser()
 ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type="colbert")
 
 # Función RAG
-def rag_chat(query):
+def rag_chat(message, history):
+    # Solo usamos el mensaje del usuario
+    query = message
+
     results = vectordb.similarity_search_with_score(query)
     context = []
     for doc, score in results:
         if score < 7:
             context.append(doc.page_content)
     if not context:
-        return "No tengo información para responder a esa pregunta."
+        return "No tengo información suficiente para responder a esa pregunta."
 
     ranking = ranker.rank(query=query, docs=context)
     best_context = ranking[0].text
 
-    prompt = hub.pull("rlm/rag-prompt")
-    rag_chain = prompt | llm | StrOutputParser()
+    prompt = f"""Contesta a la siguiente pregunta usando solo el contexto que se proporciona:
+
+    Contexto:
+    {best_context}
 
-    result = rag_chain.invoke({"context": best_context, "question": query})
-    return result
+    Pregunta: {query}
+    Respuesta:"""
 
+    return llm.invoke(prompt)
+    
 # Interfaz Gradio
-iface = gr.ChatInterface(fn=rag_chat, title="Chat Julio Verne - RAG", description="Pregunta lo que quieras sobre *La vuelta al mundo en 80 días* de Julio Verne.")
+iface = gr.ChatInterface(
+    fn=rag_chat,
+    title="Chat Julio Verne - RAG",
+    description="Pregunta lo que quieras sobre *La vuelta al mundo en 80 días* de Julio Verne.",
+    chatbot=gr.Chatbot(type="messages")  # 👈 Esto elimina el warning de formato obsoleto
+)
 iface.launch()