Update app.py

app.py

@@ -20,8 +20,8 @@ def mock_sources():
     # 假資料模擬來源列表
     return ["來源一：時間的四則問題", "來源二：新文章"]
 
-def toggle_visibility(current_visibility):
-    return gr.update(visible=not current_visibility)
+def toggle_visibility(current_state):
+    return gr.update(visible=not current_state)
 
 with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
@@ -53,12 +53,15 @@ with gr.Blocks() as demo:
         toggle_chat = gr.Button("顯示/隱藏 對話區域")
         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_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)
+    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])
 
+    history = gr.State([])
     ask_button.click(mock_question_answer, inputs=[question, history], outputs=[chatbot, chatbot])
     summary_button.click(mock_summary, inputs=[], outputs=[summary])