Update app.py

app.py

@@ -135,10 +135,6 @@ def update_image_display(image_url):
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("# 📄 Document AI Assistant")
 
-    with gr.Row():
-        toggle_left = gr.Button("📄 Toggle Document Panel", elem_id="toggle-left")
-        toggle_right = gr.Button("🎙️ Toggle Voice Panel", elem_id="toggle-right")
-
     chat_state = gr.State([])
     thread_state = gr.State()
     image_state = gr.State()
@@ -166,31 +162,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             voice_send_btn = gr.Button("🟢 Send Voice to Assistant")
             clear_transcript_btn = gr.Button("🧹 Clear Transcript")
 
-    layout = gr.Row([left_col, center_col, right_col])
-
-    def toggle_columns(left_visible, right_visible):
-        center_scale = 2
-        if not left_visible and not right_visible:
-            center_scale = 4
-        elif not left_visible or not right_visible:
-            center_scale = 3
-        return (
-            gr.update(visible=not left_visible),
-            gr.update(scale=center_scale),
-            gr.update(visible=not right_visible),
-        )
-
-    toggle_left.click(
-        fn=lambda: toggle_columns(left_col.visible, right_col.visible),
-        inputs=[],
-        outputs=[left_col, center_col, right_col]
-    )
-
-    toggle_right.click(
-        fn=lambda: toggle_columns(left_col.visible, right_col.visible),
-        inputs=[],
-        outputs=[left_col, center_col, right_col]
-    )
+    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])
 
     send_btn.click(fn=handle_chat, inputs=[user_prompt, chat_state, thread_state, image_state],
                    outputs=[user_prompt, chat, thread_state, image_state])
@@ -203,4 +190,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     image_state.change(fn=update_image_display, inputs=image_state, outputs=image_display)
     app.load(fn=create_ws, outputs=[client_id])
 
-app.launch()
+app.launch()