Update app.py

app.py

@@ -28,7 +28,7 @@ async def on_ready():
     logging.info(f'{bot.user.name} has connected to Discord!')
     try:
         # Gradio 연결 테스트
-        gradio_client.predict("test", api_name="/infer_t2i")
+        response = gradio_client.predict("test", api_name="/infer_t2i")
         logging.info("Gradio API connection successful")
     except Exception as e:
         logging.error(f"Gradio API connection failed: {str(e)}")
@@ -58,10 +58,11 @@ async def on_message(message):
         )
         
         logging.debug(f"Predict result: {result}")
-        
-        if isinstance(result, tuple):
+        if isinstance(result, tuple) and len(result) > 0:
             file_path = result[0]
-        await message.channel.send(file=discord.File(file_path))
+            await message.channel.send(file=discord.File(file_path))
+        else:
+            logging.error("Unexpected result format from Gradio API")
     except asyncio.TimeoutError:
         await message.channel.send("Image generation timed out. Please try again.")
     except Exception as e:
@@ -81,20 +82,17 @@ def run_web():
     subprocess.run([sys.executable, "web.py"])
 
 def run_discord_bot():
-    # 환경 변수에서 디스코드 토큰 읽기
     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.")
+        sys.exit(1)  # Exit if token is not set
 
 if __name__ == "__main__":
-    # 웹 서버를 별도의 프로세스로 시작
     web_process = subprocess.Popen([sys.executable, "web.py"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     logging.info("Web server process started.")
 
-    # Discord 봇 실행
     run_discord_bot()
 
-    # 웹 프로세스 종료 대기
-    web_process.wait()
+    web_process.wait()  # Ensure the web process completes