Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
-
+from notion_client import Client as NotionClient
+import requests
+from datetime import datetime
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -10,6 +12,20 @@ logging.basicConfig(level=logging.INFO)
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
+# Notion 클라이언트 설정
+NOTION_TOKEN = "secret_MpVfJphbfo4599fdczYfMYKNOpyzCcvkhhzk3lgTfVk"
+NOTION_DATABASE_ID = "b4ba17da43cd4bedb5482ef10e5ffc8d"
+notion = NotionClient(auth=NOTION_TOKEN)
+
+def add_to_notion(prompt, image_url):
+    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    new_page = {
+        "Prompt": {"title": [{"text": {"content": prompt}}]},
+        "Image": {"files": [{"name": f"Generated Image {now}", "type": "external", "external": {"url": image_url}}]},
+        "Date": {"date": {"start": now}}
+    }
+    notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
+
 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}, "
@@ -31,7 +47,10 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         
         # 결과 확인 및 처리
         if isinstance(result, dict) and 'url' in result:
-            return result['url']
+            image_url = result['url']
+            # Notion에 추가
+            add_to_notion(message, image_url)
+            return image_url
         elif isinstance(result, tuple):
             logging.error("Unexpected tuple response: %s", result)
             return result[0]
@@ -41,6 +60,8 @@ 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."
 
+
+
 css = """
 footer {
     visibility: hidden;