Update app.py

app.py

@@ -78,14 +78,21 @@ class DocumentRetrievalAndGeneration:
         distances, indices = self.gpu_index.search(np.array([query_embedding]), k=5)
 
         content = ""
-        for idx in indices[0]:
+        for idx, distance in zip(indices[0], distances[0]):
             content += "-" * 50 + "\n"
             content += self.all_splits[idx].page_content + "\n"
-            distance=distances[0][i]
             print("CHUNK", idx)
-            print("Distance :",distance)
+            print("Distance:", distance)
             print(self.all_splits[idx].page_content)
             print("############################")
+        # for idx in indices[0]:
+        #     content += "-" * 50 + "\n"
+        #     content += self.all_splits[idx].page_content + "\n"
+        #     distance=distances[0][i]
+        #     print("CHUNK", idx)
+        #     print("Distance :",distance)
+        #     print(self.all_splits[idx].page_content)
+        #     print("############################")
         prompt = f"""<s>
         You are a knowledgeable assistant with access to a comprehensive database. 
         I need you to answer my question and provide related information in a specific format.