Update app.py

app.py

@@ -6,6 +6,8 @@ from transformers import pipeline
 import subprocess
 import io
 from PIL import Image
+import time
+
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -78,6 +80,26 @@ class MyClient(discord.Client):
             # "!image" 명령어가 아닐 경우 안내 메시지 전송
             await message.channel.send('올바른 명령어를 입력해 주세요. 예) "!image 귀여운 고양이가 잠을 자고있다." 등으로 입력하시면 이미지가 생성됩니다.')
 
+
+def generate_image_with_retry(prompt, negative_prompt, max_retries=3):
+    headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+    combined_prompt = f"{prompt}. {negative_prompt}"
+    data = {"inputs": combined_prompt}
+    api_url = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
+
+    for attempt in range(max_retries):
+        response = requests.post(api_url, headers=headers, json=data)
+        if response.status_code == 200:
+            return response.json()[0]['url']
+        elif response.status_code == 503 and 'estimated_time' in response.json():
+            wait_time = response.json()['estimated_time']
+            logging.info(f"모델 로딩 중, {wait_time}초 후 재시도합니다.")
+            time.sleep(wait_time)  # 대기 후 재시도
+        else:
+            logging.error(f"API 요청에 실패했습니다: {response.text}")
+            return None
+    logging.error("최대 재시도 횟수 초과")
+    return None
 # 봇 실행
 if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')