Update app.py

app.py

@@ -1,11 +1,17 @@
 import os
-import gradio as gr
-from datetime import datetime
-from notion_client import Client
 
-# Initialize Notion client with API key
-notion = Client(auth=os.getenv('NOTION_API_KEY'))
-NOTION_DB_ID = os.getenv('NOTION_DB_ID')
+# Install required packages
+try:
+    from notion_client import Client
+except ImportError:
+    os.system('pip install notion-client')
+    from notion_client import Client
+
+try:
+    import pytz
+except ImportError:
+    os.system('pip install pytz')
+    import pytz
 
 # Install and import Groq
 try:
@@ -14,87 +20,195 @@ except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
-# Initialize Groq client with API key
-client = Groq(api_key=os.getenv('groq_key'))
+import gradio as gr
+import logging
+from datetime import datetime
+from typing import Optional
+import traceback
 
-def log_to_notion(name, user_input, bot_response):
-    """Log conversation data to Notion."""
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize clients with error handling
+try:
+    groq_client = Groq(api_key=os.getenv('groq_key'))
+    logger.info("Groq client initialized successfully")
+except Exception as e:
+    logger.error(f"Failed to initialize Groq client: {str(e)}")
+    raise
+
+try:
+    notion = Client(auth=os.getenv('NOTION_API_KEY'))
+    NOTION_DB_ID = os.getenv('NOTION_DB_ID')
+    if not NOTION_DB_ID:
+        raise ValueError("NOTION_DB_ID environment variable is not set")
+    logger.info("Notion client initialized successfully")
+except Exception as e:
+    logger.error(f"Failed to initialize Notion client: {str(e)}")
+    raise
+
+def safe_text_truncate(text: str, max_length: int = 2000) -> str:
+    """Safely truncate text to avoid Notion API limits"""
+    return text[:max_length] if text else ""
+
+def log_to_notion(name: str, user_input: str, bot_response: str) -> Optional[dict]:
+    """
+    Log chat interaction to Notion database with error handling
+    Returns the created page or None if failed
+    """
     try:
-        notion.pages.create(
+        taipei_tz = pytz.timezone('Asia/Taipei')
+        current_time = datetime.now(taipei_tz).strftime("%Y-%m-%d %H:%M:%S")
+        
+        # Truncate long texts to avoid Notion API limits
+        user_input_truncated = safe_text_truncate(user_input)
+        bot_response_truncated = safe_text_truncate(bot_response)
+        name_truncated = safe_text_truncate(name, 100)
+
+        page = notion.pages.create(
             parent={"database_id": NOTION_DB_ID},
             properties={
-                "Name": {"title": [{"text": {"content": name}}]},
-                "Timestamp": {"date": {"start": datetime.utcnow().isoformat() + "Z"}},
-                "User Input": {"rich_text": [{"text": {"content": user_input}}]},
-                "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]},
-            },
+                "Name": {
+                    "title": [{"text": {"content": name_truncated}}]
+                },
+                "Timestamp": {
+                    "rich_text": [{"text": {"content": current_time}}]
+                },
+                "User Input": {
+                    "rich_text": [{"text": {"content": user_input_truncated}}]
+                },
+                "Bot Response": {
+                    "rich_text": [{"text": {"content": bot_response_truncated}}]
+                }
+            }
         )
+        logger.info(f"Successfully logged to Notion for user: {name}")
+        return page
     except Exception as e:
-        print(f"Failed to log to Notion: {e}")
-
-def process_message(message, history):
-    """Process user message with Groq."""
-    messages = [
-        {
-            "role": "system",
-            "content": (
-                "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，"
-                "你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n"
-                "說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍"
-            ),
-        }
-    ]
-
-    for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": bot_msg})
-
-    messages.append({"role": "user", "content": message})
-
-    completion = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
-        messages=messages,
-        temperature=1,
-        max_tokens=1024,
-        top_p=1,
-        stream=True,
-        stop=None,
-    )
+        logger.error(f"Failed to log to Notion: {str(e)}\n{traceback.format_exc()}")
+        return None
 
-    response_text = ""
-    for chunk in completion:
-        delta_content = chunk.choices[0].delta.content
-        if delta_content is not None:
-            response_text += delta_content
-            yield response_text
+def process_message(message: str, history: list):
+    """Process message with error handling"""
+    try:
+        messages = [
+            {
+                "role": "system",
+                "content": "你是一個高中數學老師，使用的語言是英文。學生用中文問妳任何字彙，你都可以告訴他那個中文對應的英文和例句，以及在數學上的可能用法以及數學例題和解法。\n說明數學上的可能用法時，先用中文講一遍再用B1程度的英文複述一遍\n"
+            }
+        ]
+
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            if bot_msg:  # Only add bot message if it exists
+                messages.append({"role": "assistant", "content": bot_msg})
+
+        messages.append({"role": "user", "content": message})
+
+        completion = groq_client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=messages,
+            temperature=1,
+            max_tokens=1024,
+            top_p=1,
+            stream=True,
+            stop=None,
+        )
+
+        response_text = ""
+        for chunk in completion:
+            if chunk.choices and chunk.choices[0].delta:
+                delta_content = chunk.choices[0].delta.content
+                if delta_content:
+                    response_text += delta_content
+                    yield response_text
+
+    except Exception as e:
+        error_msg = f"Error processing message: {str(e)}"
+        logger.error(f"{error_msg}\n{traceback.format_exc()}")
+        yield error_msg
 
 # Create Gradio interface
 with gr.Blocks() as demo:
-    name_input = gr.Textbox(placeholder="請輸入您的姓名", label="Name")
-    chatbot = gr.Chatbot(height=600, show_label=False, container=True)
-    msg = gr.Textbox(placeholder="輸入您的問題...", show_label=False)
+    # 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):
-        return "", history + [[user_message, None]]
+    def user(user_message: str, history: list, name: str):
+        """Handle user input with validation"""
+        try:
+            if not name.strip():
+                gr.Warning("請先輸入姓名")
+                return user_message, history
+            return "", history + [[user_message, None]]
+        except Exception as e:
+            logger.error(f"Error in user function: {str(e)}\n{traceback.format_exc()}")
+            return "", history
 
-    def bot(history, name):
-        history[-1][1] = ""
-        user_input = history[-1][0]
-        for response in process_message(user_input, history[:-1]):
-            history[-1][1] = response
-            yield history
+    def bot(history: list, name: str):
+        """Handle bot response with error handling"""
+        try:
+            if not history:
+                return history
 
-        # Log to Notion after full response
-        bot_response = history[-1][1]
-        log_to_notion(name, user_input, bot_response)
+            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 complete response
+            if full_response:
+                notion_response = log_to_notion(
+                    name=name.strip(),
+                    user_input=history[-1][0],
+                    bot_response=full_response
+                )
+                if not notion_response:
+                    logger.warning(f"Failed to log to Notion for user: {name}")
+                    
+        except Exception as e:
+            error_msg = f"Error in bot response: {str(e)}"
+            logger.error(f"{error_msg}\n{traceback.format_exc()}")
+            history[-1][1] = error_msg
+            yield history
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, name_input], 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
 if __name__ == "__main__":
-    demo.queue()
-    demo.launch()
+    try:
+        demo.queue()
+        demo.launch()
+    except Exception as e:
+        logger.error(f"Failed to launch app: {str(e)}\n{traceback.format_exc()}")
+        raise