Update app.py

app.py

@@ -18,12 +18,12 @@ install_and_import("groq")
 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")
+# 從環境變數中提取 Notion API Key 和資料庫 ID
+NOTION_API_KEY = os.getenv("NOTION_API_KEY")
+NOTION_DB_ID = os.getenv("NOTION_DB_ID")
 
-if NOTION_API_KEY == "your_notion_api_key" or NOTION_DB_ID == "your_database_id":
-    raise ValueError("請設置 Notion API Key 和資料庫 ID！")
+if not NOTION_API_KEY or not NOTION_DB_ID:
+    raise ValueError("請設置 NOTION_API_KEY 和 NOTION_DB_ID 環境變數！")
 
 # 初始化 Notion 客戶端
 notion = Client(auth=NOTION_API_KEY)
@@ -51,7 +51,7 @@ class SimpleChatBot:
 
         # 使用 groq 的 chat.completions API 生成回應
         completion = client.chat.completions.create(
-            model="llama-3.1-70b-versatile",
+            model="llama3-8b-8192",
             messages=messages,
             temperature=1,
             max_tokens=1024,
@@ -59,7 +59,8 @@ class SimpleChatBot:
             stream=False,
         )
 
-        response_content = completion["choices"][0]["message"]["content"]
+        # 修正：正確提取 completion 的內容
+        response_content = completion.choices[0].message.content
         return response_content
 
 chatbot = SimpleChatBot()