Update app.py

app.py

@@ -1,11 +1,9 @@
+
 import gradio as gr
 from gradio_client import Client
 import os
 import logging
 import requests
-from PIL import Image
-import io
-import base64
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -16,17 +14,10 @@ api_client = Client("http://211.233.58.202:7960/")
 # Zapier 웹훅 URL
 WEBHOOK_URL = "https://hooks.zapier.com/hooks/catch/14523965/264pyhj/"
 
-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):
+def send_to_webhook(prompt, image_url):
     payload = {
         "prompt": prompt,
-        "image_data": image_data
+        "image_url": image_url
     }
     try:
         response = requests.post(WEBHOOK_URL, json=payload)
@@ -57,14 +48,13 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         # 결과 확인 및 처리
         if isinstance(result, tuple) and len(result) >= 1:
             image_path = result[0]
-            
-            # WebP를 PNG로 변환하고 base64 인코딩
-            image_data = convert_webp_to_png_base64(image_path)
+            # 이미지 URL 생성 (실제 서버 URL로 변경 필요)
+            image_url = f"http://211.233.58.202:7960/file={image_path}"
             
             # 웹훅으로 데이터 전송
-            send_to_webhook(message, image_data)
+            send_to_webhook(message, image_url)
             
-            return image_path  # Gradio 인터페이스를 위해 원본 이미지 경로 반환
+            return image_url
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
@@ -72,6 +62,7 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         return "Failed to generate image due to an error."
 
 
+
 css = """
 footer {
     visibility: hidden;