Update app.py

app.py

@@ -1,29 +1,48 @@
 import os
+from datetime import datetime
 try:
     from groq import Groq
+    from notion_client import Client
 except ImportError:
-    os.system('pip install groq')
+    os.system('pip install groq notion-client')
     from groq import Groq
+    from notion_client import Client
 import gradio as gr
 
-# 設定 API key
+# 設定 API keys
 groq_key = os.getenv("groq_key")
+notion_key = os.getenv("NOTION_API_KEY")
+notion_db_id = os.getenv("NOTION_DB_ID")
+
 if not groq_key:
-    raise ValueError("請設定環境變數 'groq_key'，包含您的 API 金鑰")
+    raise ValueError("請設定環境變數 'groq_key'")
+if not notion_key or not notion_db_id:
+    raise ValueError("請設定 NOTION_API_KEY 和 NOTION_DB_ID")
+
+groq_client = Groq(api_key=groq_key)
+notion_client = Client(auth=notion_key)
 
-client = Groq(api_key=groq_key)
+def log_to_notion(name, user_input, bot_response):
+    """記錄對話到 Notion database"""
+    try:
+        notion_client.pages.create(
+            parent={"database_id": notion_db_id},
+            properties={
+                "Name": {"title": [{"text": {"content": name}}]},
+                "Timestamp": {"date": {"start": datetime.utcnow().isoformat()}},
+                "User Input": {"rich_text": [{"text": {"content": user_input}}]},
+                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]}
+            }
+        )
+    except Exception as e:
+        print(f"Notion logging error: {str(e)}")
 
-# 定義 Chatbot 的回應邏輯
-def chat_with_groq(history):
+def chat_with_groq(history, name):
     """
-    與 Groq 聊天機器人互動，根據 Gradio 的 Chatbot 結構處理歷史記錄。
-    :param history: List of tuples (user_input, bot_response)
-    :return: Updated history with new bot response
+    與 Groq 聊天機器人互動，並記錄到 Notion
     """
-    # 提取最新的用戶輸入
     user_message = history[-1][0] if history else "你好！"
 
-    # 初始化對話，帶有 role: system 的設定
     messages = [
         {
             "role": "system",
@@ -31,37 +50,44 @@ 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(
+        completion = groq_client.chat.completions.create(
             model="llama3-8b-8192",
             messages=messages,
             temperature=1,
             max_tokens=1024,
             top_p=1,
-            stream=False,  # Gradio 不支援流式輸出，設為 False
+            stream=False,
             stop=None,
         )
-        # 提取 AI 回應
         response = completion.choices[0].message.content.strip()
+        
+        # 記錄到 Notion
+        log_to_notion(name, user_message, response)
+        
     except Exception as e:
         response = f"抱歉，發生了一個錯誤：{str(e)}"
     
-    # 將新回應加入歷史記錄
     history[-1][1] = response
     return history
 
 # 建立 Gradio 介面
 with gr.Blocks() as demo:
     gr.Markdown("## 🎨 AI 美術老師 Chatbot")
+    
+    # 新增使用者名稱輸入
+    name_input = gr.Textbox(
+        label="您的名字",
+        placeholder="請輸入您的名字...",
+        value=""
+    )
+    
     chatbot = gr.Chatbot()
     with gr.Row():
         msg = gr.Textbox(
@@ -70,17 +96,32 @@ with gr.Blocks() as demo:
         )
         clear = gr.Button("清除對話")
     
-    # 定義互動邏輯
-    def user_input(user_message, chat_history):
+    def user_input(user_message, chat_history, name):
+        if not name.strip():
+            return user_message, chat_history, "請先輸入您的名字"
         chat_history = chat_history or []
-        chat_history.append((user_message, None))  # 暫時加入用戶輸入，等待回應
-        return "", chat_history
+        chat_history.append((user_message, None))
+        return "", chat_history, ""
 
-    msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False).then(
-        chat_with_groq, chatbot, chatbot
+    # 新增錯誤訊息顯示
+    error_message = gr.Textbox(label="提示訊息", interactive=False)
+    
+    msg.submit(
+        user_input, 
+        [msg, chatbot, name_input], 
+        [msg, chatbot, error_message], 
+        queue=False
+    ).then(
+        chat_with_groq,
+        [chatbot, name_input],
+        chatbot
     )
-    clear.click(lambda: None, None, chatbot, queue=False)
+    
+    def clear_chat():
+        return None, ""
+    
+    clear.click(clear_chat, None, [chatbot, error_message], queue=False)
 
 # 啟動應用程式
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()