Update app.py

app.py

@@ -9,37 +9,54 @@ from PIL import Image
 import io
 
 # 로깅 설정
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(level=logging.DEBUG)  # DEBUG 레벨로 변경
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
 # Notion 클라이언트 설정
 NOTION_TOKEN = os.getenv("NOTION_TOKEN")
-NOTION_DATABASE_ID = "88c9bdad-cb20-4412-9af7-7d5932e1a82a"
+NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
 notion = NotionClient(auth=NOTION_TOKEN)
 
 def add_to_notion(prompt, image_path):
     try:
+        logging.debug(f"Adding to Notion - Prompt: {prompt}, Image Path: {image_path}")
         now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         
         new_page = {
             "Prompt": {"title": [{"text": {"content": prompt}}]},
-            "Image": {"url": 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}")
+        
         response = notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
         logging.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"Prompt: {prompt}")
-        logging.error(f"Image path: {image_path}")
+        logging.error(f"Notion Token: {NOTION_TOKEN[:5]}...{NOTION_TOKEN[-5:]}")  # 토큰의 일부만 로깅
         if hasattr(e, 'response'):
             logging.error(f"Full Notion API response: {e.response.text}")
-        logging.error("Please make sure the Notion integration has access to the database and the database ID is correct.")
-        raise
         
+        # 데이터베이스 정보 확인
+        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)}")
+        
+        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}, "
@@ -72,14 +89,7 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         logging.error("Error during API request: %s", str(e))
         return "Failed to generate image due to an error."
 
-def use_prompt(prompt):
-    return prompt
 
-css = """
-footer {
-    visibility: hidden;
-}
-"""
 
 css = """
 footer {
@@ -148,5 +158,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         outputs=output_image
     )
 
+
+
 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()