update send_btn.click

app.py

@@ -136,9 +136,13 @@ with gr.Blocks() as demo:
     send_btn = gr.Button("Send")
 
     send_btn.click(
-        fn=respond,
-        inputs=[msg, chat_state, name_state, country_state],
-        outputs=[chatbot, chat_state]
+    fn=respond,
+    inputs=[msg, chat_state, name_state, country_state],
+    outputs=[chat_state, chat_state]
+    ).then(
+        fn=lambda history: history,
+        inputs=[chat_state],
+        outputs=chatbot
     )
 
 demo.launch()