Update app.py

app.py

@@ -22,7 +22,7 @@ if not DISCORD_BOT_TOKEN or not GEMINI_API_KEY:
 async def hello_command(interaction):
     await interaction.response.send_message("Hello there!")
 
-@tree.command(name="imagen3", description="Generates iamge(s) using imagen3 model")
+@tree.command(name="imagen3", description="Generates image(s) using imagen3 model")
 @app_commands.choices(
     aspect_ratio=[
         app_commands.Choice(name="1:1 (Square)", value="1:1"),
@@ -35,34 +35,46 @@ async def hello_command(interaction):
 async def generate_command(
     interaction: discord.Interaction,
     prompt: str,
-    img_count:int,
+    img_count: int,
     aspect_ratio: app_commands.Choice[str],
 ):
     try:
+        await interaction.response.defer()  # Defer the interaction
         genai.configure(api_key=GEMINI_API_KEY)
         imagen = genai.ImageGenerationModel("imagen-3.0-generate-001")
-    
+
+        # Pass the value of the aspect_ratio
         result = imagen.generate_images(
             prompt=prompt,
             number_of_images=img_count,
             safety_filter_level="block_only_high",
             person_generation="allow_adult",
-            aspect_ratio=aspect_ratio,
+            aspect_ratio=aspect_ratio.value,
         )
-    
-        if result:
-            await interaction.followup.send(content=f"Successful request!")
+
+        # Debugging result structure
+        print(result)
+
+        if result and result.generated_images:
+            files = []
+            for i, image in enumerate(result.generated_images):
+                if isinstance(image, bytes):  # Raw image bytes
+                    image_file = BytesIO(image)
+                    files.append(discord.File(fp=image_file, filename=f"image_{i+1}.png"))
+                elif isinstance(image, str):  # URL
+                    async with aiohttp.ClientSession() as session:
+                        async with session.get(image) as resp:
+                            if resp.status == 200:
+                                image_file = BytesIO(await resp.read())
+                                files.append(discord.File(fp=image_file, filename=f"image_{i+1}.png"))
+            
+            # Send all images
+            await interaction.followup.send(content=f"Here are your {img_count} images for prompt: `{prompt}`", files=files)
+        else:
+            await interaction.followup.send(content="No images were generated.")
     except Exception as e:
-        interaction.followup.send(content=f"exception:{e}")
-    files = []
-    for i, image in enumerate(result.generated_images):
-        # Assuming `image` is a byte array or binary stream of the image
-        image_file = BytesIO(image)  # Convert to BytesIO
-        files.append(discord.File(fp=image_file, filename=f"image_{i+1}.png"))
-
-    # Send images to Discord
-    await interaction.followup.send(content=f"Here are your {img_count} images for prompt: `{prompt}`", files=files)
-    
+        await interaction.followup.send(content=f"Exception: {e}")
+
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")