Update app.py

app.py

@@ -140,44 +140,50 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     image_state = gr.State()
     client_id = gr.State()
 
-    left_col = gr.Column(visible=True, scale=1)
-    with left_col:
-        image_display = gr.Image(label="🖼️ Document", type="filepath", show_download_button=False, height=600)
-
-    center_col = gr.Column(scale=2)
-    with center_col:
-        chat = gr.Chatbot(label="💬 Chat", height=600)
-        with gr.Row():
-            user_prompt = gr.Textbox(placeholder="Ask your question...", show_label=False, scale=8)
-            send_btn = gr.Button("Send", variant="primary", scale=2)
-        with gr.Row():
-            clear_chat_btn = gr.Button("🗑️ Clear Chat")
-
-    right_col = gr.Column(visible=True, scale=1)
-    with right_col:
-        gr.Markdown("### 🎙️ Voice Input")
-        voice_input = gr.Audio(label="Tap to Record", streaming=True, type="numpy", show_label=True)
-        voice_transcript = gr.Textbox(label="Transcript", lines=2, interactive=False)
-        with gr.Row():
-            voice_send_btn = gr.Button("🟢 Send Voice to Assistant")
-            clear_transcript_btn = gr.Button("🧹 Clear Transcript")
-
-    def toggle_left_panel():
-        new_vis = not left_col.visible
-        new_center_scale = 2 if new_vis and right_col.visible else (3 if new_vis or right_col.visible else 4)
-        return gr.update(visible=new_vis), gr.update(scale=new_center_scale)
-
-    def toggle_right_panel():
-        new_vis = not right_col.visible
-        new_center_scale = 2 if new_vis and left_col.visible else (3 if new_vis or left_col.visible else 4)
-        return gr.update(visible=new_vis), gr.update(scale=new_center_scale)
-
     with gr.Row():
         toggle_left_btn = gr.Button("📄 Toggle Document Panel")
         toggle_right_btn = gr.Button("🎙️ Toggle Voice Panel")
 
-    toggle_left_btn.click(fn=toggle_left_panel, outputs=[left_col, center_col])
-    toggle_right_btn.click(fn=toggle_right_panel, outputs=[right_col, center_col])
+    with gr.Row(equal_height=True) as layout:
+        left_col = gr.Column(visible=True, scale=1)
+        with left_col:
+            image_display = gr.Image(label="🖼️ Document", type="filepath", show_download_button=False, height=600)
+
+        center_col = gr.Column(scale=2)
+        with center_col:
+            chat = gr.Chatbot(label="💬 Chat", height=600)
+            with gr.Row():
+                user_prompt = gr.Textbox(placeholder="Ask your question...", show_label=False, scale=8)
+                send_btn = gr.Button("Send", variant="primary", scale=2)
+            with gr.Row():
+                clear_chat_btn = gr.Button("🗑️ Clear Chat")
+
+        right_col = gr.Column(visible=True, scale=1)
+        with right_col:
+            gr.Markdown("### 🎙️ Voice Input")
+            voice_input = gr.Audio(label="Tap to Record", streaming=True, type="numpy", show_label=True)
+            voice_transcript = gr.Textbox(label="Transcript", lines=2, interactive=False)
+            with gr.Row():
+                voice_send_btn = gr.Button("🟢 Send Voice to Assistant")
+                clear_transcript_btn = gr.Button("🧹 Clear Transcript")
+
+    def toggle_left_panel(current_visibility, right_visible):
+        new_vis = not current_visibility
+        center_scale = 2 if new_vis and right_visible else (3 if new_vis or right_visible else 4)
+        return gr.update(visible=new_vis), gr.update(scale=center_scale)
+
+    def toggle_right_panel(left_visible, current_visibility):
+        new_vis = not current_visibility
+        center_scale = 2 if new_vis and left_visible else (3 if new_vis or left_visible else 4)
+        return gr.update(visible=new_vis), gr.update(scale=center_scale)
+
+    toggle_left_btn.click(fn=toggle_left_panel,
+                          inputs=[left_col, right_col],
+                          outputs=[left_col, center_col])
+
+    toggle_right_btn.click(fn=toggle_right_panel,
+                           inputs=[left_col, right_col],
+                           outputs=[right_col, center_col])
 
     send_btn.click(fn=handle_chat, inputs=[user_prompt, chat_state, thread_state, image_state],
                    outputs=[user_prompt, chat, thread_state, image_state])