Update app.py

app.py

@@ -124,10 +124,25 @@ qa_chain = ConversationalRetrievalChain.from_llm(
     output_key = 'answer'
 )
 
-def chat_interface(question,history):
-
+def chat_interface(question, history):
+    # Invoke the QA chain to get the latest answer
     result = qa_chain.invoke({"question": question})
-    return result['answer']  # If the result is a string, return it directly
+    
+    # Access the history stored in the memory
+    all_messages = memory.get_all_messages()
+    
+    # Filter out all but the latest question and answer
+    latest_question = None
+    latest_answer = None
+    for message in all_messages[::-1]:
+        if message['output_key'] == 'answer':
+            latest_answer = message['output']
+        elif message['input_key'] == 'question' and latest_question is None:
+            latest_question = message['input']
+        if latest_question is not None and latest_answer is not None:
+            break
+    
+    return latest_answer  # Return the answer to the latest question
 
 chatbot_gradio_app = gr.ChatInterface(
     fn=chat_interface,