Update app.py

app.py

@@ -33,14 +33,16 @@ with gr.Blocks() as demo:
     msg = gr.Textbox(placeholder="Type a message...")
 
     def respond(user_input, chat_history):
-        # stream output
         bot_reply = ""
+        # Start by adding the user input
+        chat_history.append((user_input, ""))  # empty bot reply for now
         for partial in chat_fn_stream(user_input):
             bot_reply = partial
-            # append the latest partial to chat
+            # Update the last bot reply
             chat_history[-1] = (user_input, bot_reply)
             yield chat_history, chat_history
 
+
     state = gr.State([])
     msg.submit(respond, [msg, state], [chatbot_ui, state])