Update app.py

app.py

@@ -8,8 +8,9 @@ def mock_question_answer(question, history):
         "文件的摘要是什麼？": "這份文件討論了如何利用人工智慧工具，提升企業的運營效率和決策速度。"
     }
     response = answers.get(question, "抱歉，我無法回答這個問題。請嘗試其他問題！")
-    history.append((question, response))
-    return history, response
+    history.append({"role": "user", "content": question})
+    history.append({"role": "assistant", "content": response})
+    return history, ""
 
 def mock_summary():
     # 假資料模擬摘要
@@ -19,24 +20,27 @@ def mock_sources():
     # 假資料模擬來源列表
     return ["來源一：時間的四則問題", "來源二：新文章"]
 
+def toggle_visibility(visible_state):
+    return not visible_state
+
 with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
 
     with gr.Row():
-        with gr.Column() as source_column:
+        with gr.Column(visible=True) 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() as chat_column:
+        with gr.Column(visible=True) as chat_column:
             gr.Markdown("### 對話區域")
-            chatbot = gr.Chatbot(label="聊天記錄")
+            chatbot = gr.Chatbot(label="聊天記錄", type="messages")
             question = gr.Textbox(label="輸入問題，例如：文件的核心觀點是什麼？")
             ask_button = gr.Button("提問")
 
-        with gr.Column() as feature_column:
+        with gr.Column(visible=True) as feature_column:
             gr.Markdown("### 功能卡片")
             with gr.Tab("摘要生成"):
                 summary_button = gr.Button("生成摘要")
@@ -50,13 +54,13 @@ 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)
 
-    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])
+    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)
 
     ask_button.click(mock_question_answer, inputs=[question, history], outputs=[chatbot, chatbot])
     summary_button.click(mock_summary, inputs=[], outputs=[summary])