Update app.py

app.py

@@ -16,18 +16,26 @@ api_client = Client("http://211.233.58.202:7960/")
 
 # Notion 클라이언트 설정
 NOTION_TOKEN = os.getenv("NOTION_TOKEN")
-NOTION_DATABASE_ID = "b4ba17da43cd4bedb5482ef10e5ffc8d"
+NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
 notion = NotionClient(auth=NOTION_TOKEN)
 
 def add_to_notion(prompt, image_path):
-    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    
-    new_page = {
-        "Prompt": {"title": [{"text": {"content": prompt}}]},
-        "Image": {"url": image_path},
-        "Date": {"date": {"start": now}}
-    }
-    notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
+    try:
+        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        
+        new_page = {
+            "Prompt": {"title": [{"text": {"content": prompt}}]},
+            "Image": {"url": image_path},
+            "Date": {"date": {"start": now}}
+        }
+        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}")
+        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}, "
@@ -63,14 +71,13 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
 def use_prompt(prompt):
     return prompt
 
-
-
 css = """
 footer {
     visibility: hidden;
 }
 """
 
+
 # 이미지 생성을 위한 예제 프롬프트
 examples = [
     ["A glamorous young woman with long, wavy blonde hair and smokey eye makeup, posing in a luxury hotel room. She’s wearing a sparkly gold cocktail dress and holding up a white card with 'openfree.ai' written on it in elegant calligraphy. Soft, warm lighting creates a luxurious atmosphere. ", "q1.webp"],
@@ -95,12 +102,8 @@ examples = [
     ["A fantasy map of a fictional world, with detailed terrain and cities.", "q19.webp"]
 ]
 
-def use_prompt(prompt):
-    return prompt
 
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
-
-    
     with gr.Row():
         input_text = gr.Textbox(label="Enter your prompt for image generation")
         output_image = gr.Image(label="Generated Image")