Update app.py

app.py

@@ -5,7 +5,6 @@ import logging
 import requests
 from PIL import Image
 import io
-import tempfile
 import base64
 
 # 로깅 설정
@@ -18,13 +17,11 @@ api_client = Client("http://211.233.58.202:7960/")
 WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY0MDYzMTA0MzE1MjZlNTUzYzUxM2Ei_pc"
 
 # Imgur 클라이언트 ID
-IMGUR_CLIENT_ID = "여기에_당신의_IMGUR_CLIENT_ID를_입력하세요"
+IMGUR_CLIENT_ID = "4990123196cbfcc"
 
-def upload_to_imgur(image_path):
+def upload_to_imgur(image_data):
     try:
-        with open(image_path, "rb") as file:
-            im_bytes = file.read()
-        im_b64 = base64.b64encode(im_bytes).decode("utf-8")
+        im_b64 = base64.b64encode(image_data).decode("utf-8")
         
         url = "https://api.imgur.com/3/image"
         payload = {'image': im_b64, 'type': 'base64'}
@@ -73,13 +70,17 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         if isinstance(result, tuple) and len(result) >= 1:
             local_image_path = result[0]
             
+            # 이미지 파일을 메모리에 로드
+            with open(local_image_path, "rb") as f:
+                image_data = f.read()
+            
             # Imgur에 이미지 업로드
-            imgur_url = upload_to_imgur(local_image_path)
+            imgur_url = upload_to_imgur(image_data)
             
             if imgur_url:
                 # Webhook으로 데이터 전송
                 send_to_webhook(message, imgur_url)
-                return Image.open(local_image_path)
+                return Image.open(io.BytesIO(image_data))
             else:
                 raise ValueError("Failed to upload image to Imgur")
         else:
@@ -90,6 +91,7 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
 
 
 
+
 css = """
 footer {
     visibility: hidden;