Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 from PIL import Image
 import io
 import base64
+import traceback
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -18,7 +19,7 @@ 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_data):
     try:
@@ -48,65 +49,12 @@ def upload_to_imgur(image_data):
         
         return response.json()['data']['link']
     except Exception as e:
-        logger.error(f"Failed to upload image to Imgur: {e}")
+        logger.error(f"Failed to upload image to Imgur: {str(e)}")
+        logger.error(f"Traceback: {traceback.format_exc()}")
         if hasattr(e, 'response'):
             logger.error(f"Response content: {e.response.text}")
         return None
 
-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):
-    logger.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
-                 f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
-                 f"num_inference_steps: {num_inference_steps}")
-    
-    try:
-        # 이미지 생성 요청
-        result = api_client.predict(
-            prompt=message,
-            seed=seed,
-            randomize_seed=randomize_seed,
-            width=width,
-            height=height,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            api_name="/infer_t2i"
-        )
-        logger.info("API response received: %s", result)
-        
-        # 결과 확인 및 처리
-        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(image_data)
-            
-            if imgur_url:
-                # Webhook으로 데이터 전송
-                send_to_webhook(message, imgur_url)
-                return Image.open(io.BytesIO(image_data))
-            else:
-                raise ValueError("Failed to upload image to Imgur")
-        else:
-            raise ValueError("Unexpected API response format")
-    except Exception as e:
-        logger.error("Error during API request: %s", str(e))
-        return "Failed to generate image due to an error."
-
 # ... (나머지 코드는 그대로 유지)