Update app.py

app.py

@@ -3,6 +3,9 @@ from gradio_client import Client
 import os
 import logging
 import requests
+import pyimgur
+from PIL import Image
+import io
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -13,10 +16,22 @@ api_client = Client("http://211.233.58.202:7960/")
 # Webhook URL
 WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY0MDYzMTA0MzE1MjZlNTUzYzUxM2Ei_pc"
 
+# Imgur 클라이언트 ID
+IMGUR_CLIENT_ID = "여기에_당신의_IMGUR_CLIENT_ID를_입력하세요"
+imgur_client = pyimgur.Imgur(IMGUR_CLIENT_ID)
+
+def upload_to_imgur(image_path):
+    try:
+        uploaded_image = imgur_client.upload_image(image_path, title="Generated Image")
+        return uploaded_image.link
+    except Exception as e:
+        logging.error(f"Failed to upload image to Imgur: {e}")
+        return None
+
 def send_to_webhook(prompt, image_url):
     payload = {
         "prompt": prompt,
-        "image_url": image_url
+        "image": image_url
     }
     try:
         response = requests.post(WEBHOOK_URL, json=payload)
@@ -45,23 +60,23 @@ 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']
-            # Webhook으로 데이터 전송
-            send_to_webhook(message, image_url)
-            return image_url
-        elif isinstance(result, tuple):
-            logging.error("Unexpected tuple response: %s", result)
-            image_url = result[0]
-            # Webhook으로 데이터 전송
-            send_to_webhook(message, image_url)
-            return image_url
+        if isinstance(result, tuple) and len(result) >= 1:
+            local_image_path = result[0]
+            # Imgur에 이미지 업로드
+            imgur_url = upload_to_imgur(local_image_path)
+            if imgur_url:
+                # Webhook으로 데이터 전송
+                send_to_webhook(message, imgur_url)
+                return Image.open(local_image_path)
+            else:
+                raise ValueError("Failed to upload image to Imgur")
         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."
 
+
 css = """
 footer {
     visibility: hidden;
@@ -137,5 +152,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         outputs=output_image
     )
 
+# ... (나머지 코드는 그대로 유지)
+
 if __name__ == "__main__":
     demo.launch()