Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 import subprocess
+from datetime import datetime
+import gradio as gr
 
 # 自動檢查並安裝必要的套件
 def install_and_import(package):
@@ -9,33 +11,31 @@ def install_and_import(package):
         subprocess.check_call(["pip", "install", package])
 
 # 安裝依賴
+install_and_import("notion-client")
 install_and_import("gradio")
-install_and_import("gspread")
 install_and_import("groq")
 
-import gradio as gr
-import gspread
-from datetime import datetime
+from notion_client import Client
 from groq import Groq
 
+# 設定 Notion API 和資料庫 ID
+NOTION_API_KEY = os.getenv("NOTION_API_KEY", "your_notion_api_key")
+NOTION_DB_ID = os.getenv("NOTION_DB_ID", "your_database_id")
+
+if NOTION_API_KEY == "your_notion_api_key" or NOTION_DB_ID == "your_database_id":
+    raise ValueError("請設置 Notion API Key 和資料庫 ID！")
+
+# 初始化 Notion 客戶端
+notion = Client(auth=NOTION_API_KEY)
+
 # 設定 Groq 的 API Key
 groq_key = os.getenv("groq_key")
 if not groq_key:
-    raise ValueError("groq_key environment variable is not set")
+    raise ValueError("groq_key 環境變數未設置！")
 
 client = Groq(api_key=groq_key)
 
-# 使用共享連結的方式操作 Google Sheet
-sheet_url = "https://docs.google.com/spreadsheets/d/1BQ23fiiGu5wN4RmRC4XmrKS6qCWHuymlbsoLEEmEN_4/edit?usp=sharing"
-
-# 使用 gspread 連接 Google Sheet（不需要憑證，依賴共享權限）
-try:
-    gc = gspread.oauth()  # 使用匿名方式連接
-    sheet = gc.open_by_url(sheet_url).sheet1  # 獲取第一個工作表
-except Exception as e:
-    raise ValueError(f"無法連接 Google Sheet，請確認連結或權限設置：{e}")
-
-# 定義簡單的 Chatbot
+# 定義 Chatbot 類
 class SimpleChatBot:
     def __init__(self):
         self.initial_prompt = [
@@ -64,14 +64,21 @@ class SimpleChatBot:
 
 chatbot = SimpleChatBot()
 
-# 記錄聊天內容到 Google Sheet
-def log_to_sheet(user_message, bot_response):
+# 記錄聊天內容到 Notion 資料庫
+def log_to_notion(user_message, bot_response):
     try:
         timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        sheet.append_row([timestamp, user_message, bot_response])
-        print("Logged to Google Sheet successfully.")
+        notion.pages.create(
+            parent={"database_id": NOTION_DB_ID},
+            properties={
+                "Timestamp": {"date": {"start": timestamp}},
+                "User Input": {"rich_text": [{"text": {"content": user_message}}]},
+                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]},
+            },
+        )
+        print("Logged to Notion successfully.")
     except Exception as e:
-        print(f"Error logging to Google Sheet: {e}")
+        print(f"Error logging to Notion: {e}")
 
 # 定義回應邏輯
 def respond(message, chat_history):
@@ -80,8 +87,8 @@ def respond(message, chat_history):
     chat_history.append({"role": "user", "content": message})
     chat_history.append({"role": "assistant", "content": response})
 
-    # 記錄到 Google Sheet
-    log_to_sheet(message, response)
+    # 記錄到 Notion
+    log_to_notion(message, response)
 
     return chat_history, ""