Update app.py

app.py

@@ -7,9 +7,10 @@ import requests
 from datetime import datetime
 from PIL import Image
 import io
+import sys
 
 # 로깅 설정
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', stream=sys.stdout)
 notion_logger = logging.getLogger('notion')
 notion_logger.setLevel(logging.DEBUG)
 
@@ -19,9 +20,13 @@ api_client = Client("http://211.233.58.202:7960/")
 # Notion 클라이언트 설정
 NOTION_TOKEN = os.getenv("NOTION_TOKEN")
 NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
-notion = NotionClient(auth=NOTION_TOKEN)
 
 def test_notion_connection():
+    if not NOTION_TOKEN:
+        notion_logger.error("NOTION_TOKEN is not set. Please set the environment variable.")
+        return False
+
+    notion = NotionClient(auth=NOTION_TOKEN)
     try:
         notion_logger.debug("Testing Notion connection...")
         user_info = notion.users.me()
@@ -38,7 +43,20 @@ def test_notion_connection():
             notion_logger.error(f"Notion API response: {e.response.text}")
         return False
 
+# Notion 연결 테스트 실행
+notion_connection_successful = test_notion_connection()
+
+if notion_connection_successful:
+    notion = NotionClient(auth=NOTION_TOKEN)
+else:
+    notion_logger.error("Failed to connect to Notion. The application will continue without Notion integration.")
+    notion = None
+
 def add_to_notion(prompt, image_path):
+    if not notion:
+        notion_logger.warning("Notion integration is not available. Skipping add_to_notion.")
+        return
+
     try:
         notion_logger.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
         now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -56,7 +74,6 @@ def add_to_notion(prompt, image_path):
         notion_logger.error(f"Error adding to Notion: {str(e)}")
         if hasattr(e, 'response'):
             notion_logger.error(f"Notion API response: {e.response.text}")
-        raise
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     logging.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
@@ -90,6 +107,7 @@ 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,12 +178,10 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
 
 
 
+# ... (나머지 코드는 그대로 유지)
+
 if __name__ == "__main__":
     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.")
+    demo.launch()