Update app.py

app.py

@@ -3,11 +3,10 @@ from gradio_client import Client
 import os
 import logging
 import requests
-import pyimgur
 from PIL import Image
 import io
 import tempfile
-import shutil
+import base64
 
 # 로깅 설정
 logging.basicConfig(level=logging.INFO)
@@ -20,20 +19,21 @@ WEBHOOK_URL = "https://connect.pabbly.com/workflow/sendwebhookdata/IjU3NjUwNTY0M
 
 # Imgur 클라이언트 ID
 IMGUR_CLIENT_ID = "여기에_당신의_IMGUR_CLIENT_ID를_입력하세요"
-imgur_client = pyimgur.Imgur(IMGUR_CLIENT_ID)
-
-def convert_to_png(image_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:
-        png_path = convert_to_png(image_path)
-        uploaded_image = imgur_client.upload_image(png_path, title="Generated Image")
-        os.remove(png_path)  # 임시 PNG 파일 삭제
-        return uploaded_image.link
+        with open(image_path, "rb") as file:
+            im_bytes = file.read()
+        im_b64 = base64.b64encode(im_bytes).decode("utf-8")
+        
+        url = "https://api.imgur.com/3/image"
+        payload = {'image': im_b64, 'type': 'base64'}
+        headers = {'Authorization': f'Client-ID {IMGUR_CLIENT_ID}'}
+        
+        response = requests.post(url, headers=headers, data=payload)
+        response.raise_for_status()
+        
+        return response.json()['data']['link']
     except Exception as e:
         logging.error(f"Failed to upload image to Imgur: {e}")
         return None
@@ -72,14 +72,9 @@ 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(temp_image_path)
-            os.remove(temp_image_path)  # 임시 파일 삭제
+            imgur_url = upload_to_imgur(local_image_path)
             
             if imgur_url:
                 # Webhook으로 데이터 전송
@@ -94,6 +89,7 @@ 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;