Update app.py

app.py

@@ -86,7 +86,8 @@ def respond(message, history, name, country): # Generate a response using the mo
     ]
 
     if history:
-        messages.extend(history)
+        messages.extend([{"role": "user", "content": m[0]} for m in history])
+        messages.extend([{"role": "assistant", "content": m[1]} for m in history])
 
     messages.append({"role": "user", "content": message})
 
@@ -96,7 +97,9 @@ def respond(message, history, name, country): # Generate a response using the mo
         temperature=0.2
     )
 
-    return response["choices"][0]["message"]["content"].strip()
+    bot_reply = response["choices"][0]["message"]["content"].strip()
+
+    return [message, bot_reply], history + [[message, bot_reply]]
 
 
 # SURVEY
@@ -135,7 +138,7 @@ with gr.Blocks() as demo:
     send_btn.click(
         fn=respond,
         inputs=[msg, chat_state, name_state, country_state],
-        outputs=chatbot
+        outputs=[chatbot, chat_state]
     )
 
 demo.launch()