Update app.py

app.py

@@ -3,6 +3,9 @@ from gradio_client import Client
 import os
 import logging
 import requests
+from PIL import Image
+import io
+import base64
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -13,10 +16,17 @@ api_client = Client("http://211.233.58.202:7960/")
 # Zapier 웹훅 URL
 WEBHOOK_URL = "https://hooks.zapier.com/hooks/catch/14523965/264pyhj/"
 
-def send_to_webhook(prompt, image_url):
+def convert_webp_to_png_base64(image_path):
+    with Image.open(image_path) as img:
+        buffered = io.BytesIO()
+        img.convert("RGB").save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode()
+        return f"data:image/png;base64,{img_str}"
+
+def send_to_webhook(prompt, image_data):
     payload = {
         "prompt": prompt,
-        "image_url": image_url
+        "image_data": image_data
     }
     try:
         response = requests.post(WEBHOOK_URL, json=payload)
@@ -47,13 +57,14 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         # 결과 확인 및 처리
         if isinstance(result, tuple) and len(result) >= 1:
             image_path = result[0]
-            # 이미지 URL 생성 (실제 서버 URL로 변경 필요)
-            image_url = f"http://211.233.58.202:7960/file={image_path}"
+            
+            # WebP를 PNG로 변환하고 base64 인코딩
+            image_data = convert_webp_to_png_base64(image_path)
             
             # 웹훅으로 데이터 전송
-            send_to_webhook(message, image_url)
+            send_to_webhook(message, image_data)
             
-            return image_url
+            return image_path  # Gradio 인터페이스를 위해 원본 이미지 경로 반환
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e: