Update app.py

app.py

@@ -61,7 +61,8 @@ def chat_with_zephyr(message, history):
     # Extract Zephyr's response
     zephyr_response = response.split("Zephyr:")[-1].strip()
     
-    return zephyr_response
+    # Return the new message pair
+    return "", zephyr_response
 
 css = """
 body {
@@ -79,7 +80,18 @@ with gr.Blocks(css=css) as iface:
     msg = gr.Textbox(placeholder="Tell Zephyr what's on your mind...", label="Your message")
     clear = gr.Button("Clear Chat")
 
-    msg.submit(chat_with_zephyr, [msg, chatbot], [chatbot])
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        user_message = history[-1][0]
+        bot_response = chat_with_zephyr(user_message, history[:-1])
+        history[-1][1] = bot_response
+        return history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
     clear.click(lambda: None, None, chatbot, queue=False)
 
     gr.Markdown("""