Update app.py

app.py

@@ -8,7 +8,8 @@ import io
 import base64
 
 # 로깅 설정
-logging.basicConfig(level=logging.DEBUG)  # DEBUG 레벨로 변경
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
 
 # API 클라이언트 설정
 api_client = Client("http://211.233.58.202:7960/")
@@ -21,23 +22,27 @@ IMGUR_CLIENT_ID = "여기에_당신의_IMGUR_CLIENT_ID를_입력하세요"
 
 def upload_to_imgur(image_data):
     try:
+        logger.debug(f"Image data type: {type(image_data)}")
+        logger.debug(f"Image data length: {len(image_data)}")
+        
         im_b64 = base64.b64encode(image_data).decode("utf-8")
         
         url = "https://api.imgur.com/3/image"
-        payload = {'image': im_b64}
+        payload = {'image': im_b64, 'type': 'base64'}
         headers = {'Authorization': f'Client-ID {IMGUR_CLIENT_ID}'}
         
+        logger.debug("Sending request to Imgur API")
         response = requests.post(url, headers=headers, data=payload)
-        logging.debug(f"Imgur API response status: {response.status_code}")
-        logging.debug(f"Imgur API response content: {response.text}")
+        logger.debug(f"Imgur API response status: {response.status_code}")
+        logger.debug(f"Imgur API response content: {response.text}")
         
         response.raise_for_status()
         
         return response.json()['data']['link']
-    except requests.exceptions.RequestException as e:
-        logging.error(f"Failed to upload image to Imgur: {e}")
+    except Exception as e:
+        logger.error(f"Failed to upload image to Imgur: {e}")
         if hasattr(e, 'response'):
-            logging.error(f"Response content: {e.response.text}")
+            logger.error(f"Response content: {e.response.text}")
         return None
 
 def send_to_webhook(prompt, image_url):
@@ -48,12 +53,12 @@ def send_to_webhook(prompt, image_url):
     try:
         response = requests.post(WEBHOOK_URL, json=payload)
         response.raise_for_status()
-        logging.info(f"Successfully sent data to webhook. Status code: {response.status_code}")
+        logger.info(f"Successfully sent data to webhook. Status code: {response.status_code}")
     except requests.exceptions.RequestException as e:
-        logging.error(f"Failed to send data to webhook: {e}")
+        logger.error(f"Failed to send data to webhook: {e}")
 
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
-    logging.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
+    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}")
     
@@ -69,7 +74,7 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
             num_inference_steps=num_inference_steps,
             api_name="/infer_t2i"
         )
-        logging.info("API response received: %s", result)
+        logger.info("API response received: %s", result)
         
         # 결과 확인 및 처리
         if isinstance(result, tuple) and len(result) >= 1:
@@ -91,9 +96,12 @@ def respond(message, seed, randomize_seed, width, height, guidance_scale, num_in
         else:
             raise ValueError("Unexpected API response format")
     except Exception as e:
-        logging.error("Error during API request: %s", str(e))
+        logger.error("Error during API request: %s", str(e))
         return "Failed to generate image due to an error."
 
+# ... (나머지 코드는 그대로 유지)
+
+
 
 css = """
 footer {
@@ -170,6 +178,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         outputs=output_image
     )
 
+
 if __name__ == "__main__":
-    logging.info(f"Starting application with Imgur Client ID: {IMGUR_CLIENT_ID[:5]}...")  # 보안을 위해 전체 ID를 로깅하지 않습니다
+    logger.info(f"Starting application with Imgur Client ID: {IMGUR_CLIENT_ID[:5]}...")
     demo.launch()