Update app.py

app.py

@@ -56,6 +56,7 @@ def add_page_to_notion_database(database_id, name, prompt, image_url):
     )
     return new_page.url
 
+
 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}, "
                  f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
@@ -81,10 +82,10 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
             image_url = f"http://211.233.58.202:7960/file={image_path}"
             
             # Notion 데이터베이스 생성 (이미 존재한다면 기존 ID 사용)
-            database_id = os.getenv("NOTION_DATABASE_ID")
+            database_id = NOTION_DATABASE_ID  # 이 변수를 미리 정의해야 합니다
             if not database_id:
                 database_id = create_notion_database()
-                os.environ["NOTION_DATABASE_ID"] = database_id
+                NOTION_DATABASE_ID = database_id  # 전역 변수로 저장
             
             # Notion에 페이지 추가
             page_url = add_page_to_notion_database(
@@ -103,14 +104,6 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         return "Failed to generate image due to an error.", None
 
 
-
-
-
-
-
-
-
-
 def use_prompt(prompt):
     return prompt