Update app.py

app.py

@@ -5,6 +5,8 @@ import logging
 from notion_client import Client as NotionClient
 import requests
 from datetime import datetime
+from PIL import Image
+import io
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -13,15 +15,20 @@ logging.basicConfig(level=logging.INFO)
 api_client = Client("http://211.233.58.202:7960/")
 
 # Notion 클라이언트 설정
-NOTION_TOKEN = "secret_MpVfJphbfo4599fdczYfMYKNOpyzCcvkhhzk3lgTfVk"
+NOTION_TOKEN = "your_notion_integration_token"
 NOTION_DATABASE_ID = "b4ba17da43cd4bedb5482ef10e5ffc8d"
 notion = NotionClient(auth=NOTION_TOKEN)
 
-def add_to_notion(prompt, image_url):
+def add_to_notion(prompt, image_path):
     now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    # 이미지를 Notion에 업로드
+    with open(image_path, "rb") as f:
+        response = notion.files.upload(file=f)
+    
     new_page = {
         "Prompt": {"title": [{"text": {"content": prompt}}]},
-        "Image": {"files": [{"name": f"Generated Image {now}", "type": "external", "external": {"url": image_url}}]},
+        "Image": {"files": [{"name": f"Generated Image {now}", "type": "file", "file": {"url": response.url}}]},
         "Date": {"date": {"start": now}}
     }
     notion.pages.create(parent={"database_id": NOTION_DATABASE_ID}, properties=new_page)
@@ -46,20 +53,19 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         logging.info("API response received: %s", result)
         
         # 결과 확인 및 처리
-        if isinstance(result, dict) and 'url' in result:
-            image_url = result['url']
+        if isinstance(result, tuple) and len(result) >= 1:
+            image_path = result[0]
             # Notion에 추가
-            add_to_notion(message, image_url)
-            return image_url
-        elif isinstance(result, tuple):
-            logging.error("Unexpected tuple response: %s", result)
-            return result[0]
+            add_to_notion(message, image_path)
+            return Image.open(image_path)
         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."
 
+def use_prompt(prompt):
+    return prompt
 
 
 css = """
@@ -68,8 +74,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"],