Update app.py

app.py

@@ -184,7 +184,11 @@ class LangChain_Document_QA:
         message = response.choices[0].text.strip()
         if  ":" in message:
           message = re.sub(r'^.*:', '', message)
-        return message.strip()
+
+        response = message.strip()
+        history[-1][1] = response
+        history_state.value = history
+        return history
       except:
         return "How can I help you?"
 
@@ -270,8 +274,8 @@ class LangChain_Document_QA:
 
           txt_msg = txt.submit(self._add_text, [chatbot, txt], [chatbot, txt])
           txt_msg.then(lambda: gr.update(interactive=True), None, [txt])
-          txt.submit(self._suggested_answer,txt,txt3)
-          button.click(self._agent_text, [chatbot,txt3], chatbot)
+          txt.submit(self._suggested_answer,txt,chatbot)
+          #button.click(self._agent_text, [chatbot,txt3], )
           end_btn.click(self._display_history, [], txt4)
           emptyBtn.click(self.clear_func,[],[])
           emptyBtn.click(lambda: None, None, chatbot, queue=False)