Update app.py

app.py

@@ -17,13 +17,17 @@ translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 # 고정된 네거티브 프롬프트
 negative_prompt = "blur, low quality, bad composition, ugly, disfigured, weird colors, low quality, jpeg artifacts, lowres, grainy, deformed structures, blurry, opaque, low contrast, distorted details, details are low"
 
-# 인퍼런스 API를 사용하기 위한 함수
+# 이미지 생성 및 응답 처리 함수
 def generate_image(prompt, negative_prompt):
     headers = {
         "Authorization": f"Bearer {os.getenv('HF_TOKEN')}"
     }
     data = {
-        "inputs": prompt + " " + negative_prompt  # 프롬프트와 네거티브 프롬프트를 하나의 문자열로 결합
+        "inputs": {
+            "prompt": prompt,
+            "negative_prompt": negative_prompt,
+            "num_inference_steps": 50
+        }
     }
     api_url = "https://api-inference.huggingface.co/models/fluently/Fluently-XL-Final"
     response = requests.post(api_url, headers=headers, json=data)
@@ -34,19 +38,8 @@ def generate_image(prompt, negative_prompt):
         logging.error(f"API 요청에 실패했습니다: {response.text}")
         return None
 
-# 프롬프트 번역 함수
-def translate_prompt(prompt):
-    logging.debug(f'프롬프트 번역 중: {prompt}')
-    translation = translator(prompt, max_length=512)
-    translated_text = translation[0]['translation_text']
-    logging.debug(f'번역된 텍스트: {translated_text}')
-    return translated_text
-
-# 디스코드 봇 클래스
+# 디스코드 메시지 이벤트 처리
 class MyClient(discord.Client):
-    async def on_ready(self):
-        logging.info(f'{self.user}로 로그인되었습니다!')
-
     async def on_message(self, message):
         if message.author == self.user:
             return
@@ -72,7 +65,7 @@ class MyClient(discord.Client):
             # "!image" 명령어가 아닐 경우 안내 메시지 전송
             await message.channel.send('올바른 명령어를 입력해 주세요. 예) "!image 귀여운 고양이가 잠을 자고있다." 등으로 입력하시면 이미지가 생성됩니다.')
 
-# 디스코드 토큰 및 봇 실행
+# 봇 실행
 if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')
     discord_client = MyClient(intents=intents)