Update app.py

app.py

@@ -6,6 +6,8 @@ import requests
 import pyimgur
 from PIL import Image
 import io
+import tempfile
+import shutil
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -21,10 +23,10 @@ IMGUR_CLIENT_ID = "여기에_당신의_IMGUR_CLIENT_ID를_입력하세요"
 imgur_client = pyimgur.Imgur(IMGUR_CLIENT_ID)
 
 def convert_to_png(image_path):
-    with Image.open(image_path) as img:
-        png_path = image_path.rsplit('.', 1)[0] + '.png'
-        img.save(png_path, 'PNG')
-    return png_path
+    with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp_file:
+        with Image.open(image_path) as img:
+            img.save(temp_file.name, 'PNG')
+    return temp_file.name
 
 def upload_to_imgur(image_path):
     try:
@@ -70,8 +72,15 @@ 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 tempfile.NamedTemporaryFile(suffix='.webp', delete=False) as temp_file:
+                shutil.copy2(local_image_path, temp_file.name)
+                temp_image_path = temp_file.name
+            
             # Imgur에 이미지 업로드
-            imgur_url = upload_to_imgur(local_image_path)
+            imgur_url = upload_to_imgur(temp_image_path)
+            os.remove(temp_image_path)  # 임시 파일 삭제
+            
             if imgur_url:
                 # Webhook으로 데이터 전송
                 send_to_webhook(message, imgur_url)
@@ -85,7 +94,6 @@ 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;