Update app.py

app.py

@@ -172,10 +172,12 @@ div.progress-text.svelte-z7cif2.meta-text {display: none;}
 
 # Use Gradio Blocks to wrap components
 with gr.Blocks(css=css) as demo:
-    chat = gr.ChatInterface(chat_interface, clear_btn=None, undo_btn=None, retry_btn=None)
-    
-    # Button to retrieve history and redirect
-    redirect_button = gr.Button("Close Chat")
+    with gr.Row():  # Align horizontally in the same row
+        with gr.Column(scale=8):  # 80% for chat interface
+            chat = gr.ChatInterface(chat_interface, clear_btn=None, undo_btn=None, retry_btn=None)
+
+        with gr.Column(scale=2):  # 20% for custom button
+            redirect_button = gr.Button("Close Chat")
     
     # Connect the button with the function, and handle the redirection
     redirect_button.click(fn=retrieve_history_and_redirect, inputs=[gr.State()])