Update app.py

app.py

@@ -50,10 +50,17 @@ class PDFChatbot:
         if self.index is None:
             return "Please upload a PDF first."
         
+        # query_embedding = embedding_model.encode([query], convert_to_numpy=True)
+        # _, indices = self.index.search(query_embedding, 1)  # Get top match
+        # retrieved_text = self.text_chunks[indices[0][0]]
+        # response = agent_response(query, retrieved_text)
+
         query_embedding = embedding_model.encode([query], convert_to_numpy=True)
-        _, indices = self.index.search(query_embedding, 1)  # Get top match
-        retrieved_text = self.text_chunks[indices[0][0]]
-        response = agent_response(query, retrieved_text)
+        _, indices = self.index.search(query_embedding, 5)  # Get top 5 matches
+        retrieved_texts = [self.text_chunks[idx] for idx in indices[0]]
+        retrieved_text_combined = "\n\n".join(retrieved_texts)
+        print('Retrieved_texts:',retrieved_text_combined)
+        response = agent_response(query, retrieved_text_combined)
         return response
 
 # Instantiate chatbot