Update app.py

app.py

@@ -6,6 +6,7 @@ import subprocess
 from huggingface_hub import InferenceClient
 from PIL import Image
 import io
+import base64
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -20,9 +21,6 @@ intents.guild_messages = True
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient(token=os.getenv("HF_TOKEN"))
 
-# 특정 채널 ID
-SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
-
 class MyClient(discord.Client):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
@@ -30,7 +28,7 @@ class MyClient(discord.Client):
 
     async def on_ready(self):
         logging.info(f'{self.user} has logged in!')
-        subprocess.Popen(["python", "web.py"])  # Start the web.py server as a separate process
+        subprocess.Popen(["python", "web.py"])
         logging.info("Web.py server has been started.")
 
     async def on_message(self, message):
@@ -50,25 +48,6 @@ class MyClient(discord.Client):
         finally:
             self.is_processing = False
 
-async def generate_image(prompt):
-    try:
-        # 모델 호출
-        response = hf_client(inputs={"inputs": prompt}, options={"wait_for_model": True})  # options는 필요에 따라 조정 가능
-        image_data = response.get('images', [None])[0]  # 응답에서 이미지 데이터 추출
-        if image_data:
-            image_bytes = io.BytesIO(base64.b64decode(image_data))
-            image = Image.open(image_bytes)
-            image_path = "output.png"
-            image.save(image_path)
-            return image_path
-        else:
-            raise ValueError("No image data received.")
-    except Exception as e:
-        logging.error(f"Failed to generate image: {str(e)}")
-        return None
-
-
-
 async def send_image(channel, image_path):
     file = discord.File(image_path)
     await channel.send(file=file)