Update app.py

app.py

@@ -34,24 +34,35 @@ async def generate_command(
     prompt: str,
 ):
     try:
-        await interaction.response.defer()  # Defer the interaction
+        # Defer the interaction to allow time for processing
+        await interaction.response.defer()
+
+        genai.configure(api_key=GEMINI_API_KEY)
+        model = genai.GenerativeModel("gemini-2.0-flash-exp")
         content = []  # List to store the content parts
 
-        # Handle uploaded attachments
-        if interaction.attachments:
-            for attachment in interaction.attachments:
-                # Download each attachment
-                async with aiohttp.ClientSession() as session:
-                    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: {attachment.filename}, status code: {resp.status}"
-                            )
-                            return
+        # Handle attachments if the command is used in reply to a message
+        if interaction.message.reference:
+            ref_message = await interaction.channel.fetch_message(interaction.message.reference.message_id)
+
+            if ref_message.attachments:  # Check if the replied-to message has attachments
+                for attachment in ref_message.attachments:
+                    if not attachment.content_type or not attachment.content_type.startswith("image/"):
+                        await interaction.followup.send(f"The attachment {attachment.filename} is not an image.", ephemeral=True)
+                        continue
+
+                    # Download each attachment
+                    async with aiohttp.ClientSession() as session:
+                        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: {attachment.filename}, status code: {resp.status}"
+                                )
+                                return
 
         # Add the text prompt to the content
         content.append(prompt)
@@ -90,59 +101,50 @@ async def generate_command(
         print(e)
         await interaction.followup.send(f"An error occurred: {e}")
 
-@tree.command(name="prompt", description="Generate a prompt based on an image")
-async def prompt_from_image(interaction: discord.Interaction):
+@tree.command(name="discribe", description="Process an image from a message")
+async def process_image(interaction: discord.Interaction):
     try:
         # Ensure the command is replying to a message
         if not interaction.message.reference:
             await interaction.response.send_message("Please reply to a message with an image to use this command.", ephemeral=True)
             return
 
-        # Get the original message
+        # Fetch the original message being replied to
         ref_message = await interaction.channel.fetch_message(interaction.message.reference.message_id)
-        
-        # Check if the message contains attachments
+
+        # Check if the original message has attachments
         if not ref_message.attachments:
-            await interaction.response.send_message("The replied-to message does not contain any images.", ephemeral=True)
+            await interaction.response.send_message("The replied-to message does not contain any attachments.", ephemeral=True)
             return
 
-        # Ensure the first attachment is an image
+        # Process the first attachment (assuming it's an image)
         attachment = ref_message.attachments[0]
-        if not attachment.content_type or not attachment.content_type.startswith("image/"):
-            await interaction.response.send_message("The replied-to message must contain an image.", ephemeral=True)
+        if not attachment.content_type.startswith("image/"):
+            await interaction.response.send_message("The attachment is not an image.", ephemeral=True)
             return
 
-        # Defer the interaction
+        # Defer the response to allow time for processing
         await interaction.response.defer()
 
-        # Download the image
+        # Download and process the image as needed
         async with aiohttp.ClientSession() as session:
             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")
+                    # Process the image bytes as needed
+                    # For example, pass them to your AI model
                 else:
-                    await interaction.followup.send(
-                        f"Failed to download the image: {attachment.filename}, status code: {resp.status}"
-                    )
+                    await interaction.followup.send(f"Failed to download the image: {attachment.filename}")
                     return
 
-        # Call the Generative AI model to generate a description
-        content = [{"mime_type": "image/jpeg", "data": base64_image}]
-        
-        # Ask Gemini to generate a description (prompt)
-        response = model.generate_content(content, stream=False)
-
-        # Collect the response text
-        prompt = "".join([part.text for part in response])
-
-        # Send the generated prompt back to the user
-        await interaction.followup.send(f"Here is the suggested prompt for the image:\n```{prompt}```")
+        # After processing, send a response
+        await interaction.followup.send("Image processed successfully.")
 
     except Exception as e:
-        print(e)
+        print(f"An error occurred: {e}")
         await interaction.followup.send(f"An error occurred: {e}")
 
+
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")