Update app.py

app.py

@@ -5,6 +5,7 @@ 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")
@@ -84,21 +85,29 @@ 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)
-        # 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")
+        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
+        )
+
+    except Exception as e:
+        await interaction.followup.send(f"Error: {str(e)}")
 
 @tree.command(name="hello", description="Says hello!")
 async def hello_command(interaction):