Update app.py

app.py

@@ -26,10 +26,18 @@ from datetime import datetime
 from typing import Optional
 import traceback
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
+# 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'))
@@ -39,11 +47,24 @@ except Exception as e:
     raise
 
 try:
-    notion = Client(auth=os.getenv('NOTION_API_KEY'))
+    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 environment variable is not set")
-    logger.info("Notion client initialized successfully")
+        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
@@ -58,6 +79,9 @@ def log_to_notion(name: str, user_input: str, bot_response: str) -> Optional[dic
     Returns the created page or None if failed
     """
     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")
         
@@ -66,6 +90,13 @@ def log_to_notion(name: str, user_input: str, bot_response: str) -> Optional[dic
         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(
             parent={"database_id": NOTION_DB_ID},
             properties={
@@ -83,132 +114,13 @@ def log_to_notion(name: str, user_input: str, bot_response: str) -> Optional[dic
                 }
             }
         )
-        logger.info(f"Successfully logged to Notion for user: {name}")
+        
+        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)}\n{traceback.format_exc()}")
+        logger.error(f"Failed to log to Notion: {str(e)}")
+        logger.error(f"Full traceback: {traceback.format_exc()}")
         return None
 
-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:
-    # 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: 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: list, name: str):
-        """Handle bot response with error handling"""
-        try:
-            if not history:
-                return history
-
-            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,
-        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)
-
-if __name__ == "__main__":
-    try:
-        demo.queue()
-        demo.launch()
-    except Exception as e:
-        logger.error(f"Failed to launch app: {str(e)}\n{traceback.format_exc()}")
-        raise
+[rest of the code remains the same...]