Update app.py

app.py

@@ -114,7 +114,7 @@ def rerank_with_bm25(docs, query):
     tokenized_query = clean_and_tokenize(query, lang)
     scores = bm25.get_scores(tokenized_query)
     
-    top_indices = sorted(range(len(scores)), key=lambda i: scores[i], reverse=True)[:3]
+    top_indices = sorted(range(len(scores)), key=lambda i: scores[i], reverse=True)[:2]
     return [docs[i] for i in top_indices]
 
 
@@ -140,7 +140,7 @@ def retrieve_from_rag(user_query):
     print("Searching in RAG with history context...")
 
     query_embedding = embedding_model.encode(user_query)
-    results = collection.query(query_embeddings=[query_embedding], n_results=5)  # Get top 5 first
+    results = collection.query(query_embeddings=[query_embedding], n_results=3)  # Get top 5 first
 
     if not results or not results.get('documents'):
         return None