Update app.py

app.py

@@ -1,17 +1,8 @@
 import os
-
-# 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
+from datetime import datetime
+import pytz
+import gradio as gr
+from notion_client import Client
 
 # Install and import Groq
 try:
@@ -20,107 +11,141 @@ except ImportError:
     os.system('pip install groq')
     from groq import Groq
 
-import gradio as gr
-import logging
-from datetime import datetime
-from typing import Optional
-import traceback
-
-# Set up logging with more detail
-logging.basicConfig(
-    level=logging.DEBUG,  # Changed to DEBUG for more detailed logs
-    format='%(asctime)s - %(levelname)s - %(message)s'
-)
-logger = logging.getLogger(__name__)
-
-# Debug: Print environment variables (will be masked in logs)
-logger.debug(f"NOTION_API_KEY exists: {'NOTION_API_KEY' in os.environ}")
-logger.debug(f"NOTION_DB_ID exists: {'NOTION_DB_ID' in os.environ}")
-logger.debug(f"NOTION_DB_ID value: {os.getenv('NOTION_DB_ID')}")
-
-# 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
+# Initialize Groq client with API key
+client = Groq(api_key=os.getenv('groq_key'))
 
-try:
-    notion_api_key = os.getenv('NOTION_API_KEY')
-    if not notion_api_key:
-        raise ValueError("NOTION_API_KEY is not set in environment variables")
-        
-    notion = Client(auth=notion_api_key)
-    NOTION_DB_ID = os.getenv('NOTION_DB_ID')
-    
-    if not NOTION_DB_ID:
-        raise ValueError("NOTION_DB_ID is not set in environment variables")
-        
-    # Test Notion connection
-    try:
-        notion.databases.retrieve(NOTION_DB_ID)
-        logger.info("Successfully connected to Notion database")
-    except Exception as e:
-        logger.error(f"Failed to access Notion database: {str(e)}")
-        raise
-        
-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
-    """
+# 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_message, bot_response):
+    """Log conversation to Notion database"""
     try:
-        logger.debug(f"Attempting to log to Notion for user: {name}")
-        logger.debug(f"Using database ID: {NOTION_DB_ID}")
-        
-        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)
-
-        # Debug log the data being sent
-        logger.debug(f"Preparing to send data to Notion:")
-        logger.debug(f"Name: {name_truncated}")
-        logger.debug(f"Timestamp: {current_time}")
-        logger.debug(f"User Input Length: {len(user_input_truncated)}")
-        logger.debug(f"Bot Response Length: {len(bot_response_truncated)}")
-
-        page = notion.pages.create(
+        notion.pages.create(
             parent={"database_id": NOTION_DB_ID},
             properties={
                 "Name": {
-                    "title": [{"text": {"content": name_truncated}}]
+                    "title": [
+                        {
+                            "text": {
+                                "content": name
+                            }
+                        }
+                    ]
                 },
                 "Timestamp": {
-                    "rich_text": [{"text": {"content": current_time}}]
+                    "date": {
+                        "start": datetime.now(pytz.UTC).isoformat()
+                    }
                 },
                 "User Input": {
-                    "rich_text": [{"text": {"content": user_input_truncated}}]
+                    "rich_text": [
+                        {
+                            "text": {
+                                "content": user_message
+                            }
+                        }
+                    ]
                 },
                 "Bot Response": {
-                    "rich_text": [{"text": {"content": bot_response_truncated}}]
+                    "rich_text": [
+                        {
+                            "text": {
+                                "content": bot_response
+                            }
+                        }
+                    ]
                 }
             }
         )
-        
-        logger.info(f"Successfully logged to Notion. Page ID: {page.get('id', 'unknown')}")
-        return page
-        
     except Exception as e:
-        logger.error(f"Failed to log to Notion: {str(e)}")
-        logger.error(f"Full traceback: {traceback.format_exc()}")
-        return None
+        print(f"Error logging to Notion: {e}")
+
+def process_message(message, history):
+    # Prepare the messages including history and new message
+    messages = [
+        {
+            "role": "system",
+            "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})
+
+    # Get response from Groq
+    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,
+    )
+
+    # 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 response_text
+
+# Create Gradio interface
+with gr.Blocks() as demo:
+    name_input = gr.Textbox(
+        placeholder="請輸入您的名字...",
+        label="名字",
+    )
+    
+    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 bot(user_message, history, name):
+        history[-1][1] = ""
+        for response in process_message(history[-1][0], history[:-1]):
+            history[-1][1] = response
+            yield history
+        
+        # Log the complete conversation to Notion after the response is complete
+        if history[-1][1]:  # Only log if there's a response
+            log_to_notion(name, user_message, history[-1][1])
+
+    msg.submit(
+        user,
+        [msg, chatbot],
+        [msg, chatbot],
+        queue=False
+    ).then(
+        bot,
+        [msg, chatbot, name_input],
+        chatbot
+    )
+
+    clear.click(lambda: None, None, chatbot, queue=False)
 
-[rest of the code remains the same...]
+# Launch the app
+if __name__ == "__main__":
+    demo.queue()
+    demo.launch()