Update app.py

app.py

@@ -12,9 +12,10 @@ logging.basicConfig(level=logging.INFO)
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
+# Notion API 설정
 # Notion API 설정
 NOTION_API_KEY = "secret_MpVfJphbfo4599fdczYfMYKNOpyzCcvkhhzk3lgTfVk"
-NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
+NOTION_DATABASE_ID = "b396824926394365a7ba7558b14edace"  # 새로운 데이터베이스 ID
 NOTION_API_URL = "https://api.notion.com/v1"
 
 headers = {
@@ -61,15 +62,16 @@ def upload_to_notion(prompt, image_url):
             }
         ]
     }
-
-    response = requests.post(f"{NOTION_API_URL}/pages", headers=headers, data=json.dumps(data))
     
-    if response.status_code == 200:
+    try:
+        response = requests.post(f"{NOTION_API_URL}/pages", headers=headers, data=json.dumps(data))
+        response.raise_for_status()
         logging.info("Successfully added to Notion")
         return response.json()["url"]
-    else:
-        logging.error(f"Failed to add to Notion: {response.status_code}, {response.text}")
-        raise Exception("Failed to add to Notion")
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Failed to add to Notion: {e}")
+        logging.error(f"Response content: {response.text}")
+        raise Exception(f"Failed to add to Notion: {e}")
 
 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}, "