Update app.py

app.py

@@ -55,19 +55,35 @@ def generate_image(prompt, seed, randomize_seed, width, height, guidance_scale,
         logger.error("Error during API request: %s", str(e), exc_info=True)
         return "Failed to generate image due to an error."
 
+def send_to_webhook(prompt, image_url):
+    payload = {
+        "prompt": prompt,
+        "image": image_url
+    }
+    try:
+        response = requests.post(WEBHOOK_URL, json=payload)
+        response.raise_for_status()
+        logger.info(f"Successfully sent data to webhook. Status code: {response.status_code}")
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Failed to send data to webhook: {e}")
+
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     image_data, used_seed, translated_prompt = generate_image(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps)
     
     if isinstance(image_data, bytes):
         # 이미지 데이터를 PIL Image 객체로 변환
         image = Image.open(io.BytesIO(image_data))
+        
+        # 이미지 URL 생성 (실제 환경에 맞게 수정 필요)
+        image_url = f"http://your_server_url/images/{used_seed}.png"
+        
+        # Webhook 호출
+        send_to_webhook(message, image_url)
+        
         return image, f"Used seed: {used_seed}, Translated prompt: {translated_prompt}"
     else:
         return image_data, "Error occurred"
 
-
-
-
 css = """
 footer {
     visibility: hidden;