Update app.py

app.py

@@ -33,16 +33,16 @@ def process_selected_files(selected_files, file_list):
     # 假資料模擬處理 RAG
     return f"已處理的文件: {', '.join(selected_paths)}"
 
-def toggle_visibility(current_state):
-    return gr.update(visible=not current_state), not current_state
+def toggle_switch(state):
+    return gr.update(value=not state), not state
 
 with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
 
     with gr.Row():
-        toggle_sources = gr.Button("顯示/隱藏 來源選單")
-        toggle_chat = gr.Button("顯示/隱藏 對話區域")
-        toggle_features = gr.Button("顯示/隱藏 功能卡片")
+        source_toggle = gr.Checkbox(label="顯示來源選單", value=True)
+        chat_toggle = gr.Checkbox(label="顯示對話區域", value=True)
+        feature_toggle = gr.Checkbox(label="顯示功能卡片", value=True)
 
     with gr.Row():
         with gr.Column(visible=True) as source_column:
@@ -75,13 +75,9 @@ with gr.Blocks() as demo:
             with gr.Tab("其他功能"):
                 gr.Markdown("此處可以添加更多功能卡片")
 
-    source_visible = gr.State(True)
-    chat_visible = gr.State(True)
-    feature_visible = gr.State(True)
-
-    toggle_sources.click(toggle_visibility, inputs=source_visible, outputs=[source_column, source_visible])
-    toggle_chat.click(toggle_visibility, inputs=chat_visible, outputs=[chat_column, chat_visible])
-    toggle_features.click(toggle_visibility, inputs=feature_visible, outputs=[feature_column, feature_visible])
+    source_toggle.change(toggle_switch, inputs=source_toggle, outputs=[source_column, source_toggle])
+    chat_toggle.change(toggle_switch, inputs=chat_toggle, outputs=[chat_column, chat_toggle])
+    feature_toggle.change(toggle_switch, inputs=feature_toggle, outputs=[feature_column, feature_toggle])
 
     history = gr.State([])
     ask_button.click(mock_question_answer, inputs=[question, history], outputs=[chatbot, chatbot])