Update app.py

app.py

@@ -1,10 +1,9 @@
-
 import gradio as gr
 from gradio_client import Client
 import os
 import logging
 import requests
-
+import base64
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
 
@@ -14,10 +13,18 @@ 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 download_image(image_url):
+    response = requests.get(image_url)
+    if response.status_code == 200:
+        return response.content
+    else:
+        raise Exception(f"Failed to download image: {response.status_code}")
+
+def send_to_webhook(prompt, image_content):
     payload = {
         "prompt": prompt,
-        "image_url": image_url
+        "image": base64.b64encode(image_content).decode('utf-8')
     }
     try:
         response = requests.post(WEBHOOK_URL, json=payload)
@@ -51,8 +58,11 @@ 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_url)
+            send_to_webhook(message, image_content)
             
             return image_url
         else:
@@ -61,8 +71,6 @@ 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;