Update app.py

app.py

@@ -1,77 +1,64 @@
-import requests
 import discord
-import logging
-import os
-from transformers import pipeline
+from discord.ext import commands
+from gradio_client import Client
+import multiprocessing
 import subprocess
-import io
-from PIL import Image
-import time
+import sys
 
-# 로깅 설정
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
-
-# 인텐트 설정
+# Set up Discord bot
 intents = discord.Intents.default()
 intents.message_content = True
+bot = commands.Bot(command_prefix='!', intents=intents)
+
+# Gradio client setup
+gradio_client = Client("http://211.233.58.202:7960/")
 
-# 번역 파이프라인 설정
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+# Discord channel ID to monitor
+CHANNEL_ID = 1269529561914413106
 
-# 환경 변수에서 지정된 채널 ID 가져오기
-TARGET_CHANNEL_ID = os.getenv('TARGET_CHANNEL_ID')
+@bot.event
+async def on_ready():
+    print(f'{bot.user} has connected to Discord!')
 
-# 고정된 네거티브 프롬프트
-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"
+@bot.event
+async def on_message(message):
+    if message.author == bot.user:
+        return
 
-# 프롬프트 번역 함수
-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
+    if message.channel.id != CHANNEL_ID and not isinstance(message.channel, discord.Thread):
+        return
 
-def generate_image(prompt, negative_prompt):
-    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"
-    response = requests.post(api_url, headers=headers, json=data)
-    if response.status_code == 200:
-        return response.json()[0]['url']
-    else:
-        logging.error(f"API 요청에 실패했습니다: {response.text}")
-        return None
+    # Generate image using Gradio client
+    try:
+        result = gradio_client.predict(
+            prompt=message.content,
+            seed=123,
+            randomize_seed=False,
+            width=1024,
+            height=576,
+            guidance_scale=5,
+            num_inference_steps=30,
+            api_name="/infer_t2i"
+        )
+        
+        # Send the generated image back to the user
+        await message.channel.send(file=discord.File(result))
+    except Exception as e:
+        await message.channel.send(f"An error occurred: {str(e)}")
 
-class MyClient(discord.Client):
-    async def on_ready(self):
-        logging.info(f'{self.user}로 로그인되었습니다!')
-        subprocess.Popen(["python", "web.py"])  # 별도의 Python 스크립트 실행
-        logging.info("web.py 서버가 시작되었습니다.")
+def run_web():
+    subprocess.run([sys.executable, "web.py"])
 
-    async def on_message(self, message):
-        if message.author == self.user or message.channel.id != TARGET_CHANNEL_ID:
-            return
-        if message.content.startswith('!image '):
-            self.is_processing = True
-            try:
-                prompt = message.content[len('!image '):]
-                prompt_en = translate_prompt(prompt)
-                image_url = generate_image(prompt_en, negative_prompt)
-                if image_url:
-                    await message.channel.send(f"<@{message.author.id}> 님이 요청하신 이미지입니다: {image_url}")
-                else:
-                    await message.channel.send(f"<@{message.author.id}> 이미지 생성에 실패하였습니다.")
-            finally:
-                self.is_processing = False
-        else:
-            await message.channel.send('올바른 명령어를 입력해 주세요. 예) "!image 귀여운 고양이가 잠을 자고있다." 등으로 입력하시면 이미지가 생성됩니다.')
+def run_discord_bot():
+    bot.run('MTIyODQyNTQxNDk0MzQ0MTEwNw.Gfd_ri.rrG_6-Sfp0FYvSIbv-zZ98dpHI-G_Fh9MFCzco')
 
-# 봇 실행
 if __name__ == "__main__":
-    discord_token = os.getenv('DISCORD_TOKEN')
-    discord_client = MyClient(intents=intents)
-    discord_client.run(discord_token)
+    # Start web.py in a separate process
+    web_process = multiprocessing.Process(target=run_web)
+    web_process.start()
 
+    # Run the Discord bot
+    run_discord_bot()
 
+    # Wait for the web process to finish (which it likely won't)
+    web_process.join()