Update app.py

app.py

@@ -3,35 +3,73 @@ from gradio_client import Client
 import os
 import logging
 import requests
-import base64
+import json
+from datetime import datetime
+
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
 
-# Zapier 웹훅 URL
-WEBHOOK_URL = "https://hooks.zapier.com/hooks/catch/14523965/264pyhj/"
+# Notion API 설정
+NOTION_API_KEY = "secret_MpVfJphbfo4599fdczYfMYKNOpyzCcvkhhzk3lgTfVk"
+NOTION_DATABASE_ID = "88c9bdadcb2044129af77d5932e1a82a"
+NOTION_API_URL = "https://api.notion.com/v1"
+
+headers = {
+    "Authorization": f"Bearer {NOTION_API_KEY}",
+    "Content-Type": "application/json",
+    "Notion-Version": "2022-06-28"
+}
 
+def upload_to_notion(prompt, image_url):
+    current_time = datetime.now().isoformat()
+    
+    data = {
+        "parent": {"database_id": NOTION_DATABASE_ID},
+        "properties": {
+            "Name": {
+                "title": [
+                    {
+                        "text": {
+                            "content": f"Generated Image - {current_time}"
+                        }
+                    }
+                ]
+            },
+            "Prompt": {
+                "rich_text": [
+                    {
+                        "text": {
+                            "content": prompt
+                        }
+                    }
+                ]
+            }
+        },
+        "children": [
+            {
+                "object": "block",
+                "type": "image",
+                "image": {
+                    "type": "external",
+                    "external": {
+                        "url": image_url
+                    }
+                }
+            }
+        ]
+    }
 
-def download_image(image_url):
-    response = requests.get(image_url)
+    response = requests.post(f"{NOTION_API_URL}/pages", headers=headers, data=json.dumps(data))
+    
     if response.status_code == 200:
-        return response.content
+        logging.info("Successfully added to Notion")
+        return response.json()["url"]
     else:
-        raise Exception(f"Failed to download image: {response.status_code}")
-
-def send_to_webhook(prompt, image_content):
-    payload = {
-        "prompt": prompt,
-        "image": base64.b64encode(image_content).decode('utf-8')
-    }
-    try:
-        response = requests.post(WEBHOOK_URL, json=payload)
-        response.raise_for_status()
-        logging.info(f"Successfully sent data to webhook. Status code: {response.status_code}")
-    except requests.exceptions.RequestException as e:
-        logging.error(f"Failed to send data to webhook: {e}")
+        logging.error(f"Failed to add to Notion: {response.status_code}, {response.text}")
+        raise Exception("Failed to add to Notion")
 
 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}, "
@@ -58,18 +96,18 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
             # 이미지 URL 생성 (실제 서버 URL로 변경 필요)
             image_url = f"http://211.233.58.202:7960/file={image_path}"
             
-            # 이미지 다운로드
-            image_content = download_image(image_url)
-            
-            # 웹훅으로 데이터 전송
-            send_to_webhook(message, image_content)
+            # Notion에 업로드
+            notion_url = upload_to_notion(message, image_url)
             
-            return image_url
+            return image_url, notion_url
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
         logging.error("Error during API request: %s", str(e))
-        return "Failed to generate image due to an error."
+        return "Failed to generate image due to an error.", None
+
+def use_prompt(prompt):
+    return prompt
 
 css = """
 footer {
@@ -77,8 +115,6 @@ footer {
 }
 """
 
-
-
 # 이미지 생성을 위한 예제 프롬프트
 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"],
@@ -103,15 +139,11 @@ 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")
+        notion_link = gr.Textbox(label="Notion Page Link")
     
     with gr.Row():
         seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="Seed", value=123)
@@ -143,8 +175,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
     input_text.submit(
         fn=respond, 
         inputs=[input_text, seed, randomize_seed, width, height, guidance_scale, num_inference_steps], 
-        outputs=output_image
+        outputs=[output_image, notion_link]
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)