Update app.py

app.py

@@ -92,14 +92,26 @@ class DocumentRetrievalAndGeneration:
         distances, indices = self.cpu_index.search(np.array([query_embedding]), k=5)
 
         content = ""
+        # for idx in indices[0]:
+        #     content += "-" * 50 + "\n"
+        #     content += self.all_splits[idx].page_content + "\n"
+        #     distance=distances[0][idx]
+        #     print("CHUNK", idx)
+        #     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][idx]
-            print("CHUNK", idx)
-            print("Distance :",distance)
-            print(self.all_splits[idx].page_content)
-            print("############################")
+            if idx < len(self.all_splits) and idx < len(distances[0]):
+                content += "-" * 50 + "\n"
+                content += self.all_splits[idx].page_content + "\n"
+                distance = distances[0][idx]
+                print("CHUNK", idx)
+                print("Distance :", distance)
+                print(self.all_splits[idx].page_content)
+                print("############################")
+            else:
+                print(f"Index {idx} is out of bounds. Skipping.")
+
         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.