Update app.py

app.py

@@ -1,13 +1,19 @@
 import os
 import gradio as gr
 from datetime import datetime
-from notion_client import Client
+import pytz
 try:
     from groq import Groq
 except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
+try:
+    from notion_client import Client
+except ImportError:
+    os.system('pip install notion-client')
+    from notion_client import Client
+
 # Initialize Groq client
 client = Groq(api_key=os.getenv('groq_key'))
 
@@ -17,87 +23,62 @@ NOTION_DB_ID = os.getenv('NOTION_DB_ID')
 
 def log_to_notion(name, user_input, bot_response):
     """Log conversation to Notion database"""
+    taipei_tz = pytz.timezone('Asia/Taipei')
+    current_time = datetime.now(taipei_tz).strftime("%Y-%m-%d %H:%M:%S")
+    
     try:
         notion.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}}]}
+                "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[:2000] if bot_response else ""  # Notion has a 2000 character limit
+                            }
+                        }
+                    ]
+                }
             }
         )
+        return True
     except Exception as e:
         print(f"Error logging to Notion: {e}")
+        return False
 
-def translate_to_english(word):
-    """Generate math-related English translation with bilingual explanation"""
-    # [Previous implementation remains the same]
-    messages = [
-        {
-            "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):
+def chat_response(message, chat_history, name):
     """Generate response for chatbot"""
-    # [Previous implementation remains the same]
     messages = [
         {
             "role": "system",
-            "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在���學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍\n"
+            "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n 說明數學上的可能用法時，先用中文講一遍再用 B1 程度的英文複述一遍\n"
         }
     ]
     
@@ -116,84 +97,36 @@ def chat_response(message, chat_history):
         stream=False
     )
     
-    return completion.choices[0].message.content
-
-def respond(message, name, history):
-    """Process chatbot response and update history"""
-    response = chat_response(message, history)
+    response = completion.choices[0].message.content
     
     # Log to Notion
-    if name.strip():  # Only log if name is provided
-        log_to_notion(name, message, response)
+    log_to_notion(name, message, response)
+    
+    return response
+
+def respond(message, history, name):
+    """Process chatbot response and update history"""
+    if not name.strip():
+        return "請先輸入您的名字 | Please enter your name first", history
     
+    response = chat_response(message, history, name)
     history.append((message, response))
     return "", history
 
-# [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)),
-        url("data:image/svg+xml,%3Csvg width='60' height='60' viewBox='0 0 60 60' xmlns='http://www.w3.org/2000/svg'%3E%3Cg fill='none' fill-rule='evenodd'%3E%3Cg fill='%23000000' fill-opacity='0.05'%3E%3Cpath d='M36 34v-4h-2v4h-4v2h4v4h2v-4h4v-2h-4zm0-30V0h-2v4h-4v2h4v4h2V6h4V4h-4zM6 34v-4H4v4H0v2h4v4h2v-4h4v-2H6zM6 4V0H4v4H0v2h4v4h2V6h4V4H6z'/%3E%3C/g%3E%3C/g%3E%3C/svg%3E");
-    min-height: 100vh;
-    padding: 20px;
-}
-
-.contain {
-    max-width: 1200px !important;
-    margin: auto !important;
-}
-
-.title {
-    text-align: center;
-    color: #1a237e;
-    font-size: 2.5em;
-    font-weight: bold;
-    margin-bottom: 1em;
-}
-
-.input-box {
-    border-radius: 8px !important;
-    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05) !important;
-}
-
-.output-box {
-    border-radius: 8px !important;
-    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05) !important;
-}
-"""
+# Custom CSS remains the same...
 
 # 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)
+        # Your existing translation system code...
+        pass
     
     with gr.Tab("💬 數學對話系統"):
         name_input = gr.Textbox(
-            label="姓名 | Name",
-            placeholder="請輸入您的姓名... | Please enter your name...",
-            lines=1,
+            label="您的名字 | Your Name",
+            placeholder="請輸入您的名字... | Please enter your name...",
             elem_classes=["input-box"]
         )
         
@@ -211,9 +144,9 @@ with gr.Blocks(css=custom_css) as demo:
             elem_classes=["input-box"]
         )
         
-        clear = gr.ClearButton([msg, chatbot, name_input], value="🗑️ 清除對話 | Clear Chat")
+        clear = gr.ClearButton([msg, chatbot], value="🗑️ 清除對話 | Clear Chat")
         
-        msg.submit(respond, [msg, name_input, chatbot], [msg, chatbot])
+        msg.submit(respond, [msg, chatbot, name_input], [msg, chatbot])
 
 if __name__ == "__main__":
     demo.launch()