Update app.py

app.py

@@ -23,20 +23,20 @@ with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
 
     with gr.Row():
-        with gr.Column():
+        with gr.Column() as source_column:
             gr.Markdown("### 來源選單")
             sources = gr.CheckboxGroup(
                 choices=mock_sources(), label="選取所有來源", interactive=True
             )
             upload_file = gr.File(label="從電腦添加文件", file_types=[".txt", ".pdf", ".docx"])
 
-        with gr.Column():
+        with gr.Column() as chat_column:
             gr.Markdown("### 對話區域")
             chatbot = gr.Chatbot(label="聊天記錄")
             question = gr.Textbox(label="輸入問題，例如：文件的核心觀點是什麼？")
             ask_button = gr.Button("提問")
 
-        with gr.Column():
+        with gr.Column() as feature_column:
             gr.Markdown("### 功能卡片")
             with gr.Tab("摘要生成"):
                 summary_button = gr.Button("生成摘要")
@@ -44,7 +44,20 @@ with gr.Blocks() as demo:
             with gr.Tab("其他功能"):
                 gr.Markdown("此處可以添加更多功能卡片")
 
+    with gr.Row():
+        toggle_sources = gr.Button("顯示/隱藏 來源選單")
+        toggle_chat = gr.Button("顯示/隱藏 對話區域")
+        toggle_features = gr.Button("顯示/隱藏 功能卡片")
+
     history = gr.State([])
+
+    def toggle_visibility(toggle_state, component):
+        return gr.update(visible=not toggle_state)
+
+    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])