Update app.py

app.py

@@ -32,17 +32,17 @@ def generate_text(context, query):
 def test_rag_reranking(query, ranker):
     docs = vectordb.similarity_search_with_score(query)
     context = []
-    
+
     for doc, score in docs:
-        if score < 7:  
+        if score < 7:
             doc_details = doc.to_json()['kwargs']
             context.append(doc_details['page_content'])
 
     if not context:
         return "No se encontró información suficiente para responder."
 
-    reranked = ranker.rerank(query=query, documents=context, top_k=1)
-    best_context = reranked[0]["content"]
+    reranked = ranker.rank(query=query, documents=context, top_k=1)
+    best_context = reranked[0]["text"]  
 
     return generate_text(best_context, query)