Update app.py

app.py

@@ -70,15 +70,22 @@ async def generate_command(
         image_url_or_error = await generate_image_async(prompt, aspect_ratio.value)
 
         if image_url_or_error.startswith("http"):
+            # Download the image
+            async with aiohttp.ClientSession() as session:
+                async with session.get(image_url_or_error) as resp:
+                    if resp.status != 200:
+                        return await interaction.followup.send('Could not download file...')
+                    image_data = await resp.read()
+
             # Format the prompt for display with expand/collapse
             formatted_prompt, initial_content, full_content = format_prompt_for_discord(prompt)
 
             # Create buttons
             async def show_more_callback(interaction):
-                await interaction.response.edit_message(content=f"{full_content}\n{image_url_or_error}", view=view)
+                await interaction.response.edit_message(content=full_content, view=view)
 
             async def show_less_callback(interaction):
-                await interaction.response.edit_message(content=f"{initial_content}\n{image_url_or_error}", view=view)
+                await interaction.response.edit_message(content=initial_content, view=view)
 
             show_more_button = Button(label="Read More", style=discord.ButtonStyle.primary)
             show_more_button.callback = show_more_callback
@@ -90,10 +97,14 @@ async def generate_command(
                 view.add_item(show_more_button)
             view.add_item(show_less_button)
 
-            await interaction.followup.send(
-                f"{initial_content}\n{image_url_or_error}", # Include image URL initially
-                view=view
-            )
+            # Send the image as an attachment
+            with io.BytesIO(image_data) as image_file:
+                await interaction.followup.send(
+                    content=initial_content,
+                    file=discord.File(image_file, 'generated_image.png'),
+                    view=view
+                )
+
         else:
             await interaction.followup.send(
                 f"Sorry, I couldn't generate an image. {image_url_or_error}"