Update app.py

app.py

@@ -140,6 +140,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     image_state = gr.State()
     client_id = gr.State()
 
+    # Hidden toggles as backend state
+    show_left = gr.Checkbox(value=True, visible=False)
+    show_right = gr.Checkbox(value=True, visible=False)
+
     with gr.Row():
         toggle_left_btn = gr.Button("📄 Toggle Document Panel")
         toggle_right_btn = gr.Button("🎙️ Toggle Voice Panel")
@@ -167,23 +171,29 @@ 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")
 
-    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 update_layout(show_left_val, show_right_val):
+        center_scale = 2
+        if not show_left_val and not show_right_val:
+            center_scale = 4
+        elif not show_left_val or not show_right_val:
+            center_scale = 3
+        return (
+            gr.update(visible=show_left_val),
+            gr.update(scale=center_scale),
+            gr.update(visible=show_right_val)
+        )
+
+    def toggle_left_fn(current):
+        return not current
 
-    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)
+    def toggle_right_fn(current):
+        return not current
 
-    toggle_left_btn.click(fn=toggle_left_panel,
-                          inputs=[left_col, right_col],
-                          outputs=[left_col, center_col])
+    toggle_left_btn.click(fn=toggle_left_fn, inputs=show_left, outputs=show_left)
+    toggle_right_btn.click(fn=toggle_right_fn, inputs=show_right, outputs=show_right)
 
-    toggle_right_btn.click(fn=toggle_right_panel,
-                           inputs=[left_col, right_col],
-                           outputs=[right_col, center_col])
+    show_left.change(fn=update_layout, inputs=[show_left, show_right], outputs=[left_col, center_col, right_col])
+    show_right.change(fn=update_layout, inputs=[show_left, show_right], outputs=[left_col, center_col, right_col])
 
     send_btn.click(fn=handle_chat, inputs=[user_prompt, chat_state, thread_state, image_state],
                    outputs=[user_prompt, chat, thread_state, image_state])
@@ -196,4 +206,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()