Update app.py

app.py

@@ -66,13 +66,14 @@ class SimpleChatBot:
 chatbot = SimpleChatBot()
 
 # 記錄聊天內容到 Notion 資料庫
-def log_to_notion(user_message, bot_response):
+def log_to_notion(user_name, user_message, bot_response):
     try:
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         notion.pages.create(
             parent={"database_id": NOTION_DB_ID},
             properties={
                 "Timestamp": {"date": {"start": timestamp}},
+                "Name": {"title": [{"text": {"content": user_name or "Anonymous"}}]},
                 "User Input": {"rich_text": [{"text": {"content": user_message}}]},
                 "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]},
             },
@@ -82,16 +83,16 @@ def log_to_notion(user_message, bot_response):
         print(f"Error logging to Notion: {e}")
 
 # 定義回應邏輯
-def respond(message, chat_history):
+def respond(user_name, message, chat_history):
     chat_history = [{"role": entry["role"], "content": entry["content"]} for entry in chat_history]
     response = chatbot.get_response(message, chat_history)
     chat_history.append({"role": "user", "content": message})
     chat_history.append({"role": "assistant", "content": response})
 
     # 記錄到 Notion
-    log_to_notion(message, response)
+    log_to_notion(user_name, message, response)
 
-    return chat_history, ""
+    return chat_history, "", ""
 
 # 建立 Gradio 界面
 with gr.Blocks(title="簡單的Gradio聊天機器人") as demo:
@@ -100,9 +101,14 @@ with gr.Blocks(title="簡單的Gradio聊天機器人") as demo:
     chatbot_interface = gr.Chatbot(type="messages")
     
     with gr.Row():
+        user_name = gr.Textbox(placeholder="輸入你的名字（可空白）", label="你的名字")
         user_input = gr.Textbox(placeholder="輸入訊息...", label="你的訊息")
         send_button = gr.Button("發送")
     
-    send_button.click(respond, inputs=[user_input, chatbot_interface], outputs=[chatbot_interface, user_input])
+    send_button.click(
+        respond, 
+        inputs=[user_name, user_input, chatbot_interface], 
+        outputs=[chatbot_interface, user_name, user_input]
+    )
     
     demo.launch(share=True)