Update app.py

app.py

@@ -51,19 +51,21 @@ def qa_bot():
     qa = retrieval_QA_chain(llm, qa_prompt, db)
     return qa
 
-def chatbot_response(query):
+bot = qa_bot()
+
+def chatbot_response(message, history):
     try:
-        qa = qa_bot()
-        response = qa({'query': query})
+        response = bot({'query': message})
         answer = response["result"]
         sources = response["source_documents"]
         if sources:
             answer += f"\nSources:" + str(sources)
         else:
             answer += "\nNo sources found"
-        return answer
+        history.append((message, answer))
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        history.append((message, f"An error occurred: {str(e)}"))
+    return history, history
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()