Update app.py

app.py

@@ -20,8 +20,8 @@ def mock_sources():
     # 假資料模擬來源列表
     return ["來源一：時間的四則問題", "來源二：新文章"]
 
-def toggle_visibility(visible_state):
-    return not visible_state
+def toggle_visibility(current_visibility):
+    return gr.update(visible=not current_visibility)
 
 with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
@@ -54,13 +54,10 @@ with gr.Blocks() as demo:
         toggle_features = gr.Button("顯示/隱藏 功能卡片")
 
     history = gr.State([])
-    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.visible)
-    toggle_chat.click(toggle_visibility, inputs=chat_visible, outputs=chat_column.visible)
-    toggle_features.click(toggle_visibility, inputs=feature_visible, outputs=feature_column.visible)
+    toggle_sources.click(toggle_visibility, inputs=source_column.visible, outputs=source_column)
+    toggle_chat.click(toggle_visibility, inputs=chat_column.visible, outputs=chat_column)
+    toggle_features.click(toggle_visibility, inputs=feature_column.visible, outputs=feature_column)
 
     ask_button.click(mock_question_answer, inputs=[question, history], outputs=[chatbot, chatbot])
     summary_button.click(mock_summary, inputs=[], outputs=[summary])