Update app.py

app.py

@@ -11,8 +11,21 @@ import sys
 
 # 로깅 설정
 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)
+logger = logging.getLogger(__name__)
+
+try:
+    import gradio as gr
+    from gradio_client import Client
+    import os
+    from notion_client import Client as NotionClient
+    import requests
+    from datetime import datetime
+    from PIL import Image
+    import io
+    logger.info("All modules imported successfully")
+except ImportError as e:
+    logger.error(f"Error importing modules: {e}")
+    sys.exit(1)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
@@ -22,43 +35,35 @@ NOTION_TOKEN = os.getenv("NOTION_TOKEN")
 NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
 
 def test_notion_connection():
+    logger.info("Starting Notion connection test")
     if not NOTION_TOKEN:
-        notion_logger.error("NOTION_TOKEN is not set. Please set the environment variable.")
+        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...")
+        logger.debug("Testing Notion connection...")
         user_info = notion.users.me()
-        notion_logger.info(f"Successfully connected to Notion. User info: {user_info}")
+        logger.info(f"Successfully connected to Notion. User info: {user_info}")
         
-        notion_logger.debug("Retrieving database info...")
+        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}")
+        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)}")
+        logger.error(f"Error connecting to Notion: {str(e)}")
         if hasattr(e, 'response'):
-            notion_logger.error(f"Notion API response: {e.response.text}")
+            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.")
+        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}")
+        logger.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
         now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         
         new_page = {
@@ -66,17 +71,17 @@ def add_to_notion(prompt, image_path):
             "Image": {"url": f"http://211.233.58.202:7960/file={image_path}"},
             "Date": {"date": {"start": now}}
         }
-        notion_logger.debug(f"New page properties: {new_page}")
+        logger.debug(f"New page properties: {new_page}")
         
         response = notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
-        notion_logger.info(f"Successfully added to Notion: {response}")
+        logger.info(f"Successfully added to Notion: {response}")
     except Exception as e:
-        notion_logger.error(f"Error adding to Notion: {str(e)}")
+        logger.error(f"Error adding to Notion: {str(e)}")
         if hasattr(e, 'response'):
-            notion_logger.error(f"Notion API response: {e.response.text}")
+            logger.error(f"Notion API response: {e.response.text}")
 
 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}, "
+    logger.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
                  f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
                  f"num_inference_steps: {num_inference_steps}")
     
@@ -92,7 +97,7 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
             num_inference_steps=num_inference_steps,
             api_name="/infer_t2i"
         )
-        logging.info("API response received: %s", result)
+        logger.info("API response received: %s", result)
         
         # 결과 확인 및 처리
         if isinstance(result, tuple) and len(result) >= 1:
@@ -103,11 +108,12 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
-        logging.error("Error during API request: %s", str(e))
+        logger.error("Error during API request: %s", str(e))
         return "Failed to generate image due to an error."
 
 
 
+
 css = """
 footer {
     visibility: hidden;
@@ -177,11 +183,20 @@ 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'}")
+    logger.info("Starting application")
+    logger.info(f"Notion Database ID: {NOTION_DATABASE_ID}")
+    logger.info(f"Notion Token (first 5 chars): {NOTION_TOKEN[:5] if NOTION_TOKEN else 'Not set'}")
+    
+    notion_connection_successful = test_notion_connection()
+    
+    if notion_connection_successful:
+        logger.info("Notion connection test passed. Starting Gradio interface...")
+        notion = NotionClient(auth=NOTION_TOKEN)
+    else:
+        logger.error("Failed to connect to Notion. The application will continue without Notion integration.")
+        notion = None
     
     demo.launch()