Upload app.py

app.py

@@ -30,7 +30,8 @@ def load_vector_db(zip_file_path, extract_path):
 
 # Function to augment prompt
 def augment_prompt(query, vectordb):
-    results = vectordb.similarity_search(query, k=10)
+    # Max Marginal Relevance for better diversity in info retrieval
+    results = vectordb.max_marginal_relevance_search(query, k=10, fetch_k=50)
     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. 
@@ -46,7 +47,7 @@ def augment_prompt(query, vectordb):
 
 # Function to handle chat with OpenAI
 def chat_with_openai(query, vectordb, openai_api_key):
-    chat = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key=openai_api_key)
+    chat = ChatOpenAI(model_name="gpt-5-mini", openai_api_key=openai_api_key, temperature=0.1)
     augmented_query = augment_prompt(query, vectordb)
     prompt = HumanMessage(content=augmented_query)
     messages = [
@@ -201,4 +202,4 @@ if prompt := st.chat_input("Enter your query"):
 #             if isinstance(message, HumanMessage):
 #                 st.write(f"You: {message.content}")
 #             else:
-#                 st.write(f"AI: {message.content}")------
+#                 st.write(f"AI: {message.content}")