Update app.py

app.py

@@ -20,6 +20,12 @@ CHANNEL_ID = 1269529561914413106
 @bot.event
 async def on_ready():
     print(f'{bot.user} has connected to Discord!')
+    try:
+        # Simple API call to test connection
+        gradio_client.predict("test", api_name="/infer_t2i")
+        print("Gradio API connection successful")
+    except Exception as e:
+        print(f"Gradio API connection failed: {str(e)}")
 
 @bot.event
 async def on_message(message):
@@ -48,8 +54,9 @@ async def on_message(message):
     except asyncio.TimeoutError:
         await message.channel.send("Image generation timed out. Please try again.")
     except Exception as e:
-        await message.channel.send(f"An error occurred: {str(e)}")
-        print(f"Error in on_message: {str(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():