Update app.py

app.py

@@ -50,14 +50,23 @@ async def on_message(message):
             ),
             timeout=60  # 60 seconds timeout
         )
+        
+        # 로깅을 추가하여 반환값 확인
+        print(f"Predict result: {result}")
+        
+        # 결과가 튜플이라고 가정하고 첫 번째 요소만 사용
+        if isinstance(result, tuple):
+            result = result[0]
+
         await message.channel.send(file=discord.File(result))
     except asyncio.TimeoutError:
-        await message.channel.send(file=discord.File(result[0]))  # result가 튜플인 경우
+        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}")
 
+
 @bot.event
 async def on_disconnect():
     print("Bot disconnected. Attempting to reconnect...")