Update app.py

app.py

@@ -25,45 +25,43 @@ 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="gemini", description="chat with gemini")
-# @app_commands.choices(
-#     aspect_ratio=[
-#         app_commands.Choice(name="1:1 (Square)", value="1:1"),
-#         app_commands.Choice(name="9:16 (Vertical)", value="9:16"),
-#         app_commands.Choice(name="16:9 (Horizontal)", value="16:9"),
-#         app_commands.Choice(name="3:4", value="3:4"),
-#         app_commands.Choice(name="4:3", value="4:3"),
-#     ]
-# )
+@tree.command(name="gemini", description="Chat with Gemini")
 async def generate_command(
     interaction: discord.Interaction,
     prompt: str,
-    images: List[discord.Attachment] = None,
 ):
     try:
         await interaction.response.defer()  # Defer the interaction
         genai.configure(api_key=GEMINI_API_KEY)
         model = genai.GenerativeModel("gemini-2.0-flash-exp")
         content = []  # List to store the content parts
-        if images:
-            for image in images:
+
+        # Handle uploaded attachments
+        if interaction.attachments:
+            for attachment in interaction.attachments:
+                # Download each attachment
                 async with aiohttp.ClientSession() as session:
-                    async with session.get(image.url) as resp:
-                         if resp.status == 200:
+                    async with session.get(attachment.url) as resp:
+                        if resp.status == 200:
                             image_bytes = await resp.read()
-                            base64_image = base64.b64encode(image_bytes).decode('utf-8')
-                            content.append({'mime_type':'image/jpeg', 'data': base64_image})
-
-                         else:
-                            await interaction.followup.send(f"Failed to download image: {image.filename}, status code: {resp.status}")
+                            base64_image = base64.b64encode(image_bytes).decode("utf-8")
+                            content.append({"mime_type": "image/jpeg", "data": base64_image})
+                        else:
+                            await interaction.followup.send(
+                                f"Failed to download image: {attachment.filename}, status code: {resp.status}"
+                            )
                             return
+
+        # Add the text prompt to the content
         content.append(prompt)
 
+        # Call the Generative AI model
         response = model.generate_content(content, stream=True)
 
         current_message = None
         current_message_content = ""
 
+        # Process streaming responses
         for part in response:
             text_chunk = part.text
 
@@ -75,17 +73,18 @@ async def generate_command(
                     current_message = await interaction.followup.send(content=current_message_content)
             else:
                 if current_message:
-                  await current_message.edit(content=current_message_content)
+                    await current_message.edit(content=current_message_content)
                 else:
-                  await interaction.followup.send(content=current_message_content)
+                    await interaction.followup.send(content=current_message_content)
                 current_message_content = text_chunk
                 current_message = await interaction.followup.send(content=current_message_content)
 
         if current_message_content:
-           if current_message:
-             await current_message.edit(content=current_message_content)
-           else:
-            await interaction.followup.send(content=current_message_content)
+            if current_message:
+                await current_message.edit(content=current_message_content)
+            else:
+                await interaction.followup.send(content=current_message_content)
+
     except Exception as e:
         print(e)
         await interaction.followup.send(f"An error occurred: {e}")