Update app.py

app.py

@@ -234,7 +234,7 @@ def create_chain(llm, prompt, CONDENSE_QUESTION_PROMPT, db):
     )
     return chain
 
-def create_retrieval_qa_bot():
+def create_retrieval_qa_bot(loaded_documents):
     # if not os.path.exists(persist_dir):
     #       raise FileNotFoundError(f"No directory found at {persist_dir}")
 
@@ -267,7 +267,7 @@ def create_retrieval_qa_bot():
 
     return qa
 
-def retrieve_bot_answer(query):
+def retrieve_bot_answer(query, loaded_documents):
     """
     Retrieves the answer to a given query using a QA bot.
 
@@ -333,7 +333,7 @@ def main():
                     CONDENSE_QUESTION_PROMPT = set_custom_prompt_condense()
                     db = create_vector_database(loaded_documents)
                     st.write(f"db: {db}") 
-                    response = retrieve_bot_answer(query)
+                    response = retrieve_bot_answer(query,loaded_documents)
 
                     # Display bot response
                     st.write("Bot Response:")