Update app.py

app.py

@@ -1,5 +1,8 @@
 import os
+from datetime import datetime
+from notion_client import Client
 import gradio as gr
+import pytz
 
 # Install and import Groq
 try:
@@ -11,6 +14,25 @@ except ImportError:
 # Initialize Groq client with API key
 client = Groq(api_key=os.getenv('groq_key'))
 
+# Initialize Notion client
+notion = Client(auth=os.getenv('NOTION_API_KEY'))
+NOTION_DB_ID = os.getenv('NOTION_DB_ID')
+
+def log_to_notion(name, user_input, bot_response):
+    """Log chat interaction to Notion database"""
+    taipei_tz = pytz.timezone('Asia/Taipei')
+    current_time = datetime.now(taipei_tz).strftime("%Y-%m-%d %H:%M:%S")
+    
+    notion.pages.create(
+        parent={"database_id": NOTION_DB_ID},
+        properties={
+            "Name": {"title": [{"text": {"content": name}}]},
+            "Timestamp": {"rich_text": [{"text": {"content": current_time}}]},
+            "User Input": {"rich_text": [{"text": {"content": user_input}}]},
+            "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]}
+        }
+    )
+
 def process_message(message, history):
     # Prepare the messages including history and new message
     messages = [
@@ -19,12 +41,12 @@ def process_message(message, history):
             "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍\n"
         }
     ]
-    
+
     # Add chat history
     for user_msg, bot_msg in history:
         messages.append({"role": "user", "content": user_msg})
         messages.append({"role": "assistant", "content": bot_msg})
-    
+
     # Add the new message
     messages.append({"role": "user", "content": message})
 
@@ -41,40 +63,68 @@ def process_message(message, history):
 
     # Initialize response text
     response_text = ""
-    
+
     # Stream the response
     for chunk in completion:
         delta_content = chunk.choices[0].delta.content
         if delta_content is not None:
             response_text += delta_content
-            # Yield partial response for real-time updates
             yield response_text
 
 # Create Gradio interface
 with gr.Blocks() as demo:
+    # Add name input field at the top
+    name_input = gr.Textbox(
+        placeholder="請輸入您的姓名...",
+        label="姓名",
+        show_label=True,
+    )
+    
     chatbot = gr.Chatbot(
         height=600,
         show_label=False,
         container=True,
     )
+    
     msg = gr.Textbox(
         placeholder="輸入您的問題...",
         show_label=False,
     )
+    
     clear = gr.Button("清除對話")
 
-    def user(user_message, history):
+    def user(user_message, history, name):
+        # Validate name input
+        if not name.strip():
+            return "請先輸入姓名", history
         return "", history + [[user_message, None]]
 
-    def bot(history):
+    def bot(history, name):
         history[-1][1] = ""
+        full_response = ""
         for response in process_message(history[-1][0], history[:-1]):
             history[-1][1] = response
+            full_response = response
             yield history
+        
+        # Log to Notion after the complete response is generated
+        log_to_notion(
+            name=name.strip(),
+            user_input=history[-1][0],
+            bot_response=full_response
+        )
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot, chatbot
+    msg.submit(
+        user,
+        inputs=[msg, chatbot, name_input],
+        outputs=[msg, chatbot],
+        queue=False
+    ).then(
+        bot,
+        inputs=[chatbot, name_input],
+        outputs=[chatbot]
     )
+
     clear.click(lambda: None, None, chatbot, queue=False)
 
 # Launch the app