Update app.py

app.py

@@ -1,26 +1,30 @@
+import discord
+from discord import app_commands
 import os
+from dotenv import load_dotenv
 import requests
-import discord
-from discord.ext import commands
+
+load_dotenv()
+
+# Set up intents
+intents = discord.Intents.default()
+client = discord.Client(intents=intents)
+tree = app_commands.CommandTree(client)
 
 # GLIF Simple API endpoint
 GLIF_API_URL = "https://simple-api.glif.app"
 
+
 # Function to call the GLIF API
-def generate_image(prompt):
-    # Hardcoded aspect ratio (9:16)
-    aspect_ratio = "9:16"
-    
+def generate_image(prompt, aspect_ratio):
     # Prepare the payload for the GLIF API
     payload = {
         "id": "cm3ugmzv2002gnckiosrwk6xi",  # Glif ID
-        "inputs": [prompt, aspect_ratio]  # Passing the prompt and the fixed aspect ratio
+        "inputs": [prompt, aspect_ratio]
     }
 
     # Set up headers with the API token
-    headers = {
-        "Authorization": f"Bearer {os.getenv('GLIF_API_KEY')}"
-    }
+    headers = {"Authorization": f"Bearer {os.getenv('GLIF_API_TOKEN')}"}
 
     # Make the request to the GLIF API
     response = requests.post(GLIF_API_URL, json=payload, headers=headers)
@@ -37,28 +41,55 @@ def generate_image(prompt):
     else:
         return f"API request failed with status code: {response.status_code}"
 
-# Initialize the Discord bot
-intents = discord.Intents.default()
-intents.message_content = True
-bot = commands.Bot(command_prefix="!", intents=intents)
 
-@bot.event
+# Define a slash command for image generation
+@tree.command(name="generate",
+              description="Generates an image based on a text prompt")
+@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="4:5 (Portrait)", value="4:5"),
+    app_commands.Choice(name="5:4 (Landscape)", value="5:4"),
+    app_commands.Choice(name="3:4 (Portrait)", value="3:4"),
+    app_commands.Choice(name="4:3 (Landscape)", value="4:3")
+])
+async def generate_command(interaction: discord.Interaction, prompt: str,
+                           aspect_ratio: app_commands.Choice[str]):
+    """Generates an image based on the user's prompt and aspect ratio."""
+    await interaction.response.defer()  # Acknowledge the interaction
+
+    try:
+        image_url = generate_image(prompt, aspect_ratio.value)
+
+        if image_url.startswith("http"):
+            await interaction.followup.send(
+                f"Here's your generated image based on the prompt '{prompt}' with aspect ratio {aspect_ratio.name}:\n{image_url}"
+            )
+        else:
+            await interaction.followup.send(
+                f"Sorry, I couldn't generate an image. {image_url}")
+    except Exception as e:
+        await interaction.followup.send(f"An error occurred: {e}")
+
+
+# Define the hello slash command
+@tree.command(name="hello", description="Says hello!")
+async def hello_command(interaction):
+    await interaction.response.send_message("Hello there!")
+
+
+@client.event
 async def on_ready():
-    print(f'Logged in as {bot.user.name}')
-
-@bot.command(name="generate")
-async def generate(ctx, *, prompt: str):
-    """Generate an image based on the provided prompt."""
-    await ctx.send(f"Generating image for prompt: {prompt}...")
-    
-    # Call the GLIF API to generate the image
-    image_url = generate_image(prompt)
-    
-    if image_url.startswith("http"):
-        await ctx.send(image_url)
-    else:
-        await ctx.send(f"Failed to generate image: {image_url}")
+    await tree.sync()  # Sync the slash commands with discord
+    print("Bot is ready!")
+
+
+# Get the token from an environment variable
+token = os.getenv("DISCORD_BOT_TOKEN")
 
-# Run the bot
-print(os.getenv('DISCORD_BOT_TOKEN'))
-bot.run(os.getenv('DISCORD_BOT_TOKEN'))
+# check if the token is a string
+if isinstance(token, str):
+    client.run(token)
+else:
+    print("Invalid token format")