Update app.py

app.py

@@ -18,6 +18,9 @@ intents = discord.Intents.default()
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
 
+genai.configure(api_key=GEMINI_API_KEY)
+model = genai.GenerativeModel("gemini-2.0-flash-exp")
+
 if not DISCORD_BOT_TOKEN or not GEMINI_API_KEY:
     raise ValueError("Both DISCORD_BOT_TOKEN and GEMINI_API_KEY must be set.")
 
@@ -32,8 +35,6 @@ async def generate_command(
 ):
     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
 
         # Handle uploaded attachments
@@ -89,7 +90,59 @@ 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):
+    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
+        ref_message = await interaction.channel.fetch_message(interaction.message.reference.message_id)
         
+        # Check if the message contains attachments
+        if not ref_message.attachments:
+            await interaction.response.send_message("The replied-to message does not contain any images.", ephemeral=True)
+            return
+
+        # Ensure the first attachment is 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)
+            return
+
+        # Defer the interaction
+        await interaction.response.defer()
+
+        # Download the image
+        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")
+                else:
+                    await interaction.followup.send(
+                        f"Failed to download the image: {attachment.filename}, status code: {resp.status}"
+                    )
+                    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}```")
+
+    except Exception as e:
+        print(e)
+        await interaction.followup.send(f"An error occurred: {e}")
+
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")