Update app.py

app.py

@@ -158,11 +158,9 @@ with gr.Blocks() as chat_ui:
 
     
     chatbot = gr.Chatbot()
-    with gr.Row():
-        with gr.Column(scale=8):
-            user_input = gr.Textbox(placeholder="Type your message here...", show_label=False)
-        with gr.Column(scale=2):
-            send_btn = gr.Button("Send")
+    with gr.Row(equal_height=True):
+        user_input = gr.Textbox(placeholder="Type your message here...", show_label=False, scale=8)
+        send_btn = gr.Button("Send", scale=2)
     
     upload_btn.click(process_pdf, inputs=[file_upload], outputs=[status])
     
@@ -173,6 +171,7 @@ with gr.Blocks() as chat_ui:
             full_response += word
             yield chat_history[:-1] + [(user_input, full_response)]  # Update last message only
         chat_history.append((user_input, full_response))
+        yield chat_history
     
     send_btn.click(stream_response, inputs=[user_input, chatbot], outputs=[chatbot])