Update app.py

app.py

@@ -8,8 +8,6 @@ import gradio as gr  # Import Gradio
 import google.generativeai as genai
 from io import BytesIO
 import PIL.Image
-import base64
-from typing import List
 # --- Environment Variables & Setup ---
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
@@ -18,9 +16,6 @@ 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.")
 
@@ -28,16 +23,27 @@ 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"),
+#     ]
+# )
 async def generate_command(
     interaction: discord.Interaction,
     prompt: str,
+    # img_count: int,
+    # aspect_ratio: app_commands.Choice[str],
 ):
     try:
-        await interaction.response.defer()
+        await interaction.response.defer()  # Defer the interaction
         genai.configure(api_key=GEMINI_API_KEY)
         model = genai.GenerativeModel("gemini-2.0-flash-exp")
         response = model.generate_content(prompt, stream=True)
-
         current_message = None  # Initialize outside the loop to keep track of current message
         current_message_content = ""
 
@@ -66,11 +72,10 @@ async def generate_command(
              await current_message.edit(content=current_message_content)
            else:
              await interaction.followup.send(content=current_message_content)
-
     except Exception as e:
         print(e)
 
-
+        
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")
@@ -101,5 +106,3 @@ async def main():
 
 if __name__ == "__main__":
     asyncio.run(main())
-
-