Update app.py

app.py

@@ -8,7 +8,6 @@ import torch
 from diffusers import DiffusionPipeline
 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()])
@@ -20,8 +19,8 @@ intents.message_content = True
 # 번역 파이프라인 설정
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
-# 환경 변수에서 지정된 채널 ID 가져오기
-TARGET_CHANNEL_ID = int(os.getenv('TARGET_CHANNEL_ID'))  # 채널 ID는 정수형이어야 함
+# 환경 변수에서 지정된 채널 ID 가져오기, 기본값 사용
+TARGET_CHANNEL_ID = int(os.getenv('TARGET_CHANNEL_ID', '123456789012345678'))  # 기본값 사용, 실제 채널 ID로 대체 필요
 
 # 고정된 네거티브 프롬프트
 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"
@@ -45,7 +44,6 @@ def generate_image(prompt, negative_prompt):
     combined_prompt = f"{prompt}. {negative_prompt}"
     result = pipeline(combined_prompt)
     image = result.images[0]  # 첫 번째 이미지 선택
-    # 이미지 URL 대신 이미지를 반환하도록 수정
     return image
 
 class MyClient(discord.Client):
@@ -85,3 +83,4 @@ if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')
     discord_client = MyClient(intents=intents)
     discord_client.run(discord_token)
+