Update app.py

app.py

@@ -7,6 +7,7 @@ import aiohttp  # Use aiohttp for asynchronous HTTP requests
 import gradio as gr  # Import Gradio
 import google.generativeai as genai
 from io import BytesIO
+import PIL.Image
 # --- Environment Variables & Setup ---
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
@@ -22,59 +23,30 @@ 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 image(s) using imagen3 model")
-@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")
+# @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"),
+#     ]
+# )
 async def generate_command(
     interaction: discord.Interaction,
     prompt: str,
-    img_count: int,
-    aspect_ratio: app_commands.Choice[str],
+    # 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.value,
-        )
-
-        # 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:
-        await interaction.followup.send(content=f"Exception: {e}")
+        model = genai.GenerativeModel("gemini-2.0-flash-exp")
+        response = model.generate_content(prompt)
+        await interaction.followup.send(content=response.text)
 
+        
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")