Update app.py

app.py

@@ -1,29 +1,60 @@
 import os
+from datetime import datetime
+import gradio as gr
+from notion_client import Client
+
 try:
     from groq import Groq
 except ImportError:
     os.system('pip install groq')
     from groq import Groq
-import gradio as gr
 
 # 設定 API key
 groq_key = os.getenv("groq_key")
 if not groq_key:
     raise ValueError("請設定環境變數 'groq_key'，包含您的 API 金鑰")
 
+notion_api_key = os.getenv("NOTION_API_KEY")
+notion_db_id = os.getenv("NOTION_DB_ID")
+
+if not notion_api_key or not notion_db_id:
+    raise ValueError("請設定環境變數 'NOTION_API_KEY' 和 'NOTION_DB_ID'，以連結 Notion 資料庫")
+
 client = Groq(api_key=groq_key)
+notion = Client(auth=notion_api_key)
+
+# 定義函數，將對話記錄寫入 Notion
+def log_to_notion(name, timestamp, user_input, bot_response):
+    """
+    將聊天記錄寫入 Notion 資料庫
+    :param name: 用戶名稱
+    :param timestamp: 時間戳
+    :param user_input: 用戶輸入
+    :param bot_response: 機器人回應
+    """
+    try:
+        notion.pages.create(
+            parent={"database_id": notion_db_id},
+            properties={
+                "Name": {"title": [{"text": {"content": name}}]},
+                "Timestamp": {"date": {"start": timestamp}},
+                "User Input": {"rich_text": [{"text": {"content": user_input}}]},
+                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]},
+            },
+        )
+    except Exception as e:
+        print(f"Failed to log to Notion: {str(e)}")
 
 # 定義 Chatbot 的回應邏輯
-def chat_with_groq(history):
+def chat_with_groq(name, history):
     """
-    與 Groq 聊天機器人互動，根據 Gradio 的 Chatbot 結構處理歷史記錄。
+    與 Groq 聊天機器人互動，根據 Gradio 的 Chatbot 結構處理歷史記錄，並記錄到 Notion
+    :param name: 用戶名稱
     :param history: List of tuples (user_input, bot_response)
     :return: Updated history with new bot response
     """
-    # 提取最新的用戶輸入
     user_message = history[-1][0] if history else "你好！"
 
-    # 初始化對話，帶有 role: system 的設定
     messages = [
         {
             "role": "system",
@@ -31,15 +62,12 @@ def chat_with_groq(history):
         }
     ]
 
-    # 添加歷史記錄作為上下文
-    for user_msg, bot_msg in history[:-1]:  # 忽略最後一條，因為它是新的輸入
+    for user_msg, bot_msg in history[:-1]:
         messages.append({"role": "user", "content": user_msg})
         messages.append({"role": "assistant", "content": bot_msg})
-    
-    # 添加最新的用戶訊息
+
     messages.append({"role": "user", "content": user_message})
 
-    # 呼叫 Groq API 生成回應
     try:
         completion = client.chat.completions.create(
             model="llama3-8b-8192",
@@ -47,21 +75,25 @@ def chat_with_groq(history):
             temperature=1,
             max_tokens=1024,
             top_p=1,
-            stream=False,  # Gradio 不支援流式輸出，設為 False
+            stream=False,
             stop=None,
         )
-        # 提取 AI 回應
         response = completion.choices[0].message.content.strip()
     except Exception as e:
         response = f"抱歉，發生了一個錯誤：{str(e)}"
-    
-    # 將新回應加入歷史記錄
+
+    # 記錄到 Notion 資料庫
+    timestamp = datetime.now().isoformat()
+    log_to_notion(name, timestamp, user_message, response)
+
     history[-1][1] = response
     return history
 
 # 建立 Gradio 介面
 with gr.Blocks() as demo:
     gr.Markdown("## 🎨 AI 美術老師 Chatbot")
+    with gr.Row():
+        name_input = gr.Textbox(label="Name", placeholder="請輸入您的名字...")
     chatbot = gr.Chatbot()
     with gr.Row():
         msg = gr.Textbox(
@@ -77,7 +109,7 @@ with gr.Blocks() as demo:
         return "", chat_history
 
     msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False).then(
-        chat_with_groq, chatbot, chatbot
+        lambda history: chat_with_groq(name_input.value, history), chatbot, chatbot
     )
     clear.click(lambda: None, None, chatbot, queue=False)