Update app.py

app.py

@@ -283,8 +283,12 @@ def retrieve_bot_answer(query, loaded_documents):
     """
     qa_bot_instance = create_retrieval_qa_bot(loaded_documents)
     bot_response = qa_bot_instance({"question": query})
-    answer = bot_response['answer']
-    return answer
+    # Check if the 'answer' key exists in the bot_response dictionary
+    if 'answer' in bot_response:
+        answer = bot_response['answer']
+        return answer
+    else:
+        raise KeyError("Expected 'answer' key in bot_response, but it was not found.")
 
 
 # from your_module import load_model, set_custom_prompt, set_custom_prompt_condense, create_vector_database, retrieve_bot_answer