Update app.py

app.py

@@ -27,17 +27,35 @@ def load_vector_db(zip_file_path, extract_path):
     st.success("Vector store loaded")
     return vectordb
 
+# # Function to augment prompt
+# def augment_prompt(query, vectordb):
+#     results = vectordb.similarity_search(query, k=10)
+#     source_knowledge = "\n".join([x.page_content for x in results])
+#     augmented_prompt = f"""
+#     You are an AI assistant. Use the context provided below to answer the question as comprehensively as possible. 
+#     If the answer is not contained within the context, respond politely that you cannot provide that information.
+
+#     Context:
+#     {source_knowledge}
+
+#     Question: {query}
+#     """
+#     return augmented_prompt
+
 # Function to augment prompt
-def augment_prompt(query, vectordb):
-    results = vectordb.similarity_search(query, k=10)
+def augment_prompt(query, vectordb, search_results):
+    results = vectordb.similarity_search(query, k=5)
     source_knowledge = "\n".join([x.page_content for x in results])
     augmented_prompt = f"""
     You are an AI assistant. Use the context provided below to answer the question as comprehensively as possible. 
-    If the answer is not contained within the context, respond politely that you cannot provide that information.
+    If the answer is not contained within the context, respond with "I don't know".
 
     Context:
     {source_knowledge}
 
+    Additional Web Search Results:
+    {search_results}
+
     Question: {query}
     """
     return augmented_prompt