Update app.py

app.py

@@ -5,7 +5,6 @@ import requests
 import asyncio
 import aiohttp  # Use aiohttp for asynchronous HTTP requests
 import gradio as gr  # Import Gradio
-from io import BytesIO
 
 # --- Environment Variables & Setup ---
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
@@ -85,29 +84,21 @@ async def generate_command(
     try:
         await interaction.response.defer()
 
-        # Fetch the URL of the image
         image_url_or_error = await generate_image_async(prompt, aspect_ratio.value)
-        if not image_url_or_error.startswith("http"):
-            await interaction.followup.send(f"Sorry, I couldn't generate an image: {image_url_or_error}")
-            return
-
-        # Download the image
-        async with aiohttp.ClientSession() as session:
-            async with session.get(image_url_or_error) as image_response:
-                image_response.raise_for_status()
-                image_data = await image_response.read()
-
-        # Prepare the file-like object
-        file = discord.File(BytesIO(image_data), filename="generated_image.png")
-
-        # Send the image as an attachment
-        await interaction.followup.send(
-            f"Here's your generated image based on the prompt '{prompt}' with aspect ratio {aspect_ratio.name}:",
-            file=file
-        )
-
+        # await interaction.response.send_message(f"Error:{image_url_or_error}")
+        if image_url_or_error:
+        # if image_url_or_error.startswith("http"):
+            await interaction.followup.send(
+                # f"Here's your generated image based on the prompt '{prompt}' with aspect ratio {aspect_ratio.name}:\n{image_url_or_error}"
+                f"{image_url_or_error}"
+            )
+        else:
+            await interaction.followup.send(
+                f"Sorry, I couldn't generate an image"
+            )
     except Exception as e:
-        await interaction.followup.send(f"Error: {str(e)}")
+        await interaction.followup.send(f"Error")
+
 
 @tree.command(name="hello", description="Says hello!")
 async def hello_command(interaction):