Update app.py

app.py

@@ -9,7 +9,9 @@ from PIL import Image
 import io
 
 # 로깅 설정
-logging.basicConfig(level=logging.DEBUG)  # DEBUG 레벨로 변경
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+notion_logger = logging.getLogger('notion')
+notion_logger.setLevel(logging.DEBUG)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
@@ -19,9 +21,26 @@ NOTION_TOKEN = os.getenv("NOTION_TOKEN")
 NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
 notion = NotionClient(auth=NOTION_TOKEN)
 
+def test_notion_connection():
+    try:
+        notion_logger.debug("Testing Notion connection...")
+        user_info = notion.users.me()
+        notion_logger.info(f"Successfully connected to Notion. User info: {user_info}")
+        
+        notion_logger.debug("Retrieving database info...")
+        db_info = notion.databases.retrieve(database_id=NOTION_DATABASE_ID)
+        notion_logger.info(f"Successfully retrieved database info: {db_info}")
+        
+        return True
+    except Exception as e:
+        notion_logger.error(f"Error connecting to Notion: {str(e)}")
+        if hasattr(e, 'response'):
+            notion_logger.error(f"Notion API response: {e.response.text}")
+        return False
+
 def add_to_notion(prompt, image_path):
     try:
-        logging.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
+        notion_logger.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
         now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         
         new_page = {
@@ -29,33 +48,14 @@ def add_to_notion(prompt, image_path):
             "Image": {"url": f"http://211.233.58.202:7960/file={image_path}"},
             "Date": {"date": {"start": now}}
         }
-        logging.debug(f"New page properties: {new_page}")
+        notion_logger.debug(f"New page properties: {new_page}")
         
         response = notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
-        logging.info(f"Successfully added to Notion: {response}")
+        notion_logger.info(f"Successfully added to Notion: {response}")
     except Exception as e:
-        logging.error(f"Error adding to Notion: {str(e)}")
-        logging.error(f"Database ID: {NOTION_DATABASE_ID}")
-        logging.error(f"Notion Token: {NOTION_TOKEN[:5]}...{NOTION_TOKEN[-5:]}")  # 토큰의 일부만 로깅
+        notion_logger.error(f"Error adding to Notion: {str(e)}")
         if hasattr(e, 'response'):
-            logging.error(f"Full Notion API response: {e.response.text}")
-        
-        # 데이터베이스 정보 확인
-        try:
-            logging.debug("Attempting to retrieve database info...")
-            db_info = notion.databases.retrieve(database_id=NOTION_DATABASE_ID)
-            logging.info(f"Database info: {db_info}")
-        except Exception as db_error:
-            logging.error(f"Error retrieving database info: {str(db_error)}")
-        
-        # 통합 정보 확인
-        try:
-            logging.debug("Attempting to retrieve user info...")
-            user_info = notion.users.me()
-            logging.info(f"User info: {user_info}")
-        except Exception as user_error:
-            logging.error(f"Error retrieving user info: {str(user_error)}")
-        
+            notion_logger.error(f"Notion API response: {e.response.text}")
         raise
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
@@ -90,7 +90,6 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         return "Failed to generate image due to an error."
 
 
-
 css = """
 footer {
     visibility: hidden;
@@ -160,7 +159,13 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
 
 
+
 if __name__ == "__main__":
-    logging.info(f"Starting application with Notion Database ID: {NOTION_DATABASE_ID}")
-    logging.info(f"Notion Token (first 5 chars): {NOTION_TOKEN[:5]}")
-    demo.launch()
+    notion_logger.info(f"Starting application with Notion Database ID: {NOTION_DATABASE_ID}")
+    notion_logger.info(f"Notion Token (first 5 chars): {NOTION_TOKEN[:5] if NOTION_TOKEN else 'Not set'}")
+    
+    if test_notion_connection():
+        notion_logger.info("Notion connection test passed. Starting Gradio interface...")
+        demo.launch()
+    else:
+        notion_logger.error("Failed to connect to Notion. Please check your token and database ID.")