Update app.py

app.py

@@ -2,11 +2,6 @@ import os
 import gradio as gr
 from datetime import datetime
 from notion_client import Client
-
-# Initialize Notion Client
-notion = Client(auth=os.getenv('NOTION_API_KEY'))
-NOTION_DB_ID = os.getenv('NOTION_DB_ID')
-
 try:
     from groq import Groq
 except ImportError:
@@ -16,34 +11,103 @@ except ImportError:
 # Initialize Groq client
 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 conversation to Notion database."""
+    """Log conversation to Notion database"""
     try:
         notion.pages.create(
             parent={"database_id": NOTION_DB_ID},
             properties={
                 "Name": {"title": [{"text": {"content": name}}]},
-                "Timestamp": {"date": {"start": datetime.now().isoformat() }},
+                "Timestamp": {"date": {"start": datetime.utcnow().isoformat()}},
                 "User Input": {"rich_text": [{"text": {"content": user_input}}]},
-                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]},
+                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]}
             }
         )
     except Exception as e:
-        print(f"Failed to log to Notion: {e}")
+        print(f"Error logging to Notion: {e}")
 
-def chat_response(message, chat_history):
-    """Generate response for chatbot."""
+def translate_to_english(word):
+    """Generate math-related English translation with bilingual explanation"""
+    # [Previous implementation remains the same]
     messages = [
-        {"role": "system", "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用 B1程度的英文複述一遍\n"}
+        {
+            "role": "system",
+            "content": """你是一個數學翻譯專家。請按照以下格式提供翻譯：
+
+1. 英文翻譯：[列出數學相關的英文翻譯]
+2. 中文解釋：[用中文詳細解釋這個詞在數學上的意義]
+3. English Explanation：[用英文重述上述中文解釋的內容]
+4. 數學使用場景：[說明在哪些數學情境會使用到這個詞]
+5. Mathematical Context：[用英文重述使用場景]
+
+如果該詞彙沒有數學相關的意思，請說明。"""
+        },
+        {
+            "role": "user",
+            "content": f"請提供中文詞彙 '{word}' 在數學上的英文翻譯和詳細解釋。"
+        }
     ]
+    
+    completion = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=messages,
+        temperature=0.7,
+        max_tokens=400,
+        stream=False
+    )
+    
+    return completion.choices[0].message.content
 
+def generate_example(word):
+    """Generate math-related example sentence for Chinese word"""
+    # [Previous implementation remains the same]
+    messages = [
+        {
+            "role": "system",
+            "content": """你是一個數學教師。請按照以下格式提供例句：
+
+1. 英文例句：[寫出一個數學相關的英文例句]
+2. 中文翻譯：[該例句的中文翻譯]
+3. 句子解釋：[用中文解釋這個例句中的數學概念]
+4. Sentence Explanation：[用英文重述上述解釋]"""
+        },
+        {
+            "role": "user",
+            "content": f"請用中文詞彙 '{word}' 的英文翻譯造一個數學相關的例句。"
+        }
+    ]
+    
+    completion = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=messages,
+        temperature=0.7,
+        max_tokens=400,
+        stream=False
+    )
+    
+    return completion.choices[0].message.content
+
+def chat_response(message, chat_history):
+    """Generate response for chatbot"""
+    # [Previous implementation remains the same]
+    messages = [
+        {
+            "role": "system",
+            "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍\n"
+        }
+    ]
+    
     for msg in chat_history:
         messages.append({"role": "user", "content": msg[0]})
         if msg[1]:
             messages.append({"role": "assistant", "content": msg[1]})
-
+    
     messages.append({"role": "user", "content": message})
-
+    
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=messages,
@@ -51,20 +115,21 @@ def chat_response(message, chat_history):
         max_tokens=1024,
         stream=False
     )
-
+    
     return completion.choices[0].message.content
 
-def respond(message, history, name):
-    """Process chatbot response, update history, and log to Notion."""
+def respond(message, name, history):
+    """Process chatbot response and update history"""
     response = chat_response(message, history)
-    history.append((message, response))
-
+    
     # Log to Notion
-    log_to_notion(name, message, response)
-
+    if name.strip():  # Only log if name is provided
+        log_to_notion(name, message, response)
+    
+    history.append((message, response))
     return "", history
 
-# Custom CSS for mathematical theme
+# [Previous CSS implementation remains the same]
 custom_css = """
 #component-0 {
     background-image: linear-gradient(rgba(255, 255, 255, 0.9), rgba(255, 255, 255, 0.9)),
@@ -100,24 +165,55 @@ custom_css = """
 # Create Gradio interface
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown("# 雙語數學詞彙學習系統 | Bilingual Mathematics Learning System", elem_classes=["title"])
-
+    
+    with gr.Tab("📚 單字英譯系統"):
+        with gr.Row():
+            word_input = gr.Textbox(
+                label="請輸入中文詞彙 | Enter Chinese Term",
+                placeholder="輸入中文詞彙（例如：向量、函數、極限）...",
+                lines=1,
+                elem_classes=["input-box"]
+            )
+            
+        with gr.Row():
+            translate_btn = gr.Button("🔄 英譯 | Translate", variant="primary")
+            example_btn = gr.Button("📝 例句 | Example")
+            
+        output_text = gr.Textbox(
+            label="翻譯結果 | Translation Result",
+            lines=10,
+            placeholder="翻譯結果將在這裡顯示... | Translation results will be displayed here...",
+            elem_classes=["output-box"]
+        )
+        
+        translate_btn.click(translate_to_english, inputs=word_input, outputs=output_text)
+        example_btn.click(generate_example, inputs=word_input, outputs=output_text)
+    
     with gr.Tab("💬 數學對話系統"):
         name_input = gr.Textbox(
-            label="Your Name | 請輸入您的名字",
-            placeholder="Enter your name here...",
+            label="姓名 | Name",
+            placeholder="請輸入您的姓名... | Please enter your name...",
             lines=1,
             elem_classes=["input-box"]
         )
-        chatbot = gr.Chatbot([], elem_id="chatbot", bubble_full_width=False, avatar_images=("👨‍🎓", "👨‍🏫"))
+        
+        chatbot = gr.Chatbot(
+            [],
+            elem_id="chatbot",
+            bubble_full_width=False,
+            avatar_images=("👨‍🎓", "👨‍🏫")
+        )
+        
         msg = gr.Textbox(
             label="發送訊息 | Send Message",
             placeholder="請輸入您的問題... | Enter your question...",
             show_label=False,
             elem_classes=["input-box"]
         )
-        clear = gr.ClearButton([msg, chatbot], value="🗑️ 清除對話 | Clear Chat")
-
-        msg.submit(respond, [msg, chatbot, name_input], [msg, chatbot])
+        
+        clear = gr.ClearButton([msg, chatbot, name_input], value="🗑️ 清除對話 | Clear Chat")
+        
+        msg.submit(respond, [msg, name_input, chatbot], [msg, chatbot])
 
 if __name__ == "__main__":
     demo.launch()