Update app.py

app.py

@@ -28,7 +28,7 @@ model_path = snapshot_download(
 
 # 모델 로드 함수
 def load_pipeline(pipeline_type):
-    logging.debug(f'Loading pipeline: {pipeline_type}')
+    logging.debug(f'파이프라인 로드 중: {pipeline_type}')
     if pipeline_type == "text2img":
         return StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_fast=True)
     elif pipeline_type == "img2img":
@@ -40,6 +40,9 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 # 번역 파이프라인 설정
 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"
+
 # 디스코드 봇 클래스
 class MyClient(discord.Client):
     def __init__(self, *args, **kwargs):
@@ -61,25 +64,31 @@ class MyClient(discord.Client):
             try:
                 prompt = message.content[len('!image '):]
                 prompt_en = translate_prompt(prompt)
-                logging.debug(f'Translated prompt: {prompt_en}')
-                image_path = await self.generate_image(prompt_en)
-                await message.channel.send(file=discord.File(image_path, 'generated_image.png'))
+                logging.debug(f'번역된 프롬프트: {prompt_en}')
+                logging.debug(f'고정된 네거티브 프롬프트: {negative_prompt}')
+
+                image_path = await self.generate_image(prompt_en, negative_prompt)
+                user_id = message.author.id  # 사용자의 ID 캡처
+                await message.channel.send(
+                    f"<@{user_id}> 님이 요청하신 이미지입니다.",
+                    file=discord.File(image_path, 'generated_image.png')
+                )
             finally:
                 self.is_processing = False
 
-    async def generate_image(self, prompt):
+    async def generate_image(self, prompt, negative_prompt):
         generator = torch.Generator(device=device).manual_seed(torch.seed())
-        images = self.text2img_pipeline(prompt, num_inference_steps=50, generator=generator)["images"]
+        images = self.text2img_pipeline(prompt, negative_prompt=negative_prompt, num_inference_steps=50, generator=generator)["images"]
         image_path = f'/tmp/{uuid.uuid4()}.png'
         images[0].save(image_path)
         return image_path
 
 # 프롬프트 번역 함수
 def translate_prompt(prompt):
-    logging.debug(f'Translating prompt: {prompt}')
+    logging.debug(f'프롬프트 번역 중: {prompt}')
     translation = translator(prompt, max_length=512)
     translated_text = translation[0]['translation_text']
-    logging.debug(f'Translated text: {translated_text}')
+    logging.debug(f'번역된 텍스트: {translated_text}')
     return translated_text
 
 # 디스코드 토큰 및 봇 실행