Update app.py

app.py

@@ -70,22 +70,15 @@ 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=full_content, view=view)
+                await interaction.response.edit_message(content=f"{full_content}\n{image_url_or_error}", view=view)
 
             async def show_less_callback(interaction):
-                await interaction.response.edit_message(content=initial_content, view=view)
+                await interaction.response.edit_message(content=f"{initial_content}\n{image_url_or_error}", view=view)
 
             show_more_button = Button(label="Read More", style=discord.ButtonStyle.primary)
             show_more_button.callback = show_more_callback
@@ -97,14 +90,10 @@ async def generate_command(
                 view.add_item(show_more_button)
             view.add_item(show_less_button)
 
-            # 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
-                )
-
+            await interaction.followup.send(
+                f"{initial_content}\n{image_url_or_error}", # Include image URL initially
+                view=view
+            )
         else:
             await interaction.followup.send(
                 f"Sorry, I couldn't generate an image. {image_url_or_error}"