Update app.py

app.py

@@ -2,30 +2,36 @@ import discord
 from discord.ext import commands
 from gradio_client import Client
 import asyncio
-import multiprocessing
+import os
+import logging
 import subprocess
 import sys
 
-# Set up Discord bot
+# 로깅 설정
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
+
+# 인텐트 설정
 intents = discord.Intents.default()
 intents.message_content = True
+
+# Discord Bot 설정
 bot = commands.Bot(command_prefix='!', intents=intents)
 
-# Gradio client setup
+# Gradio API 클라이언트 설정
 gradio_client = Client("http://211.233.58.202:7960/")
 
-# Discord channel ID to monitor
+# Discord 채널 ID
 CHANNEL_ID = 1269529561914413106
 
 @bot.event
 async def on_ready():
-    print(f'{bot.user} has connected to Discord!')
+    logging.info(f'{bot.user.name} has connected to Discord!')
     try:
-        # Simple API call to test connection
+        # Gradio 연결 테스트
         gradio_client.predict("test", api_name="/infer_t2i")
-        print("Gradio API connection successful")
+        logging.info("Gradio API connection successful")
     except Exception as e:
-        print(f"Gradio API connection failed: {str(e)}")
+        logging.error(f"Gradio API connection failed: {str(e)}")
 
 @bot.event
 async def on_message(message):
@@ -51,40 +57,44 @@ async def on_message(message):
             timeout=60  # 60 seconds timeout
         )
         
-        print(f"Predict result: {result}")
+        logging.debug(f"Predict result: {result}")
         
-        # 파일 경로만 추출하여 사용
-        file_path = result[0]  # 튜플의 첫 번째 요소가 파일 경로
+        if isinstance(result, tuple):
+            file_path = result[0]
         await message.channel.send(file=discord.File(file_path))
     except asyncio.TimeoutError:
         await message.channel.send("Image generation timed out. Please try again.")
     except Exception as e:
         error_message = f"An error occurred: {str(e)}"
         await message.channel.send(error_message)
-        print(f"Error in on_message: {error_message}")
-
+        logging.error(f"Error in on_message: {error_message}")
 
 @bot.event
 async def on_disconnect():
-    print("Bot disconnected. Attempting to reconnect...")
+    logging.info("Bot disconnected. Attempting to reconnect...")
 
 @bot.event
 async def on_resume():
-    print("Bot connection resumed.")
+    logging.info("Bot connection resumed.")
 
 def run_web():
     subprocess.run([sys.executable, "web.py"])
 
 def run_discord_bot():
-    bot.run('MTIyODQyNTQxNDk0MzQ0MTEwNw.Gfd_ri.rrG_6-Sfp0FYvSIbv-zZ98dpHI-G_Fh9MFCzco')
+    # 환경 변수에서 디스코드 토큰 읽기
+    discord_token = os.getenv('DISCORD_TOKEN')
+    if discord_token:
+        bot.run(discord_token)
+    else:
+        logging.error("Discord token is not set. Please check your environment variables.")
 
 if __name__ == "__main__":
-    # Start web.py in a separate process
-    web_process = multiprocessing.Process(target=run_web)
-    web_process.start()
+    # 웹 서버를 별도의 프로세스로 시작
+    web_process = subprocess.Popen([sys.executable, "web.py"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    logging.info("Web server process started.")
 
-    # Run the Discord bot
+    # Discord 봇 실행
     run_discord_bot()
 
-    # Wait for the web process to finish (which it likely won't)
-    web_process.join()
+    # 웹 프로세스 종료 대기
+    web_process.wait()