Update app.py

app.py

@@ -1,38 +1,95 @@
 import discord
 from discord import app_commands
 import os
-import asyncio
-import time
+from dotenv import load_dotenv
+import requests
 
-DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
+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, aspect_ratio):
+    # Prepare the payload for the GLIF API
+    payload = {
+        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Glif ID
+        "inputs": [prompt, aspect_ratio]
+    }
+
+    # Set up headers with the API token
+    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)
+
+    if response.status_code == 200:
+        # Parse the response
+        response_data = response.json()
+        if "output" in response_data:
+            # Return the image URL
+            return response_data["output"]
+        elif "error" in response_data:
+            # Handle errors (even though the status code is 200)
+            return f"Error: {response_data['error']}"
+    else:
+        return f"API request failed with status code: {response.status_code}"
+
+
+# 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):
-    print("Hello command triggered")
-    await interaction.response.send_message("Hello there! I am functional")
+    await interaction.response.send_message("Hello there!")
+
 
+@client.event
 async def on_ready():
-    print("on_ready() called")
-    await tree.sync()
+    await tree.sync()  # Sync the slash commands with discord
     print("Bot is ready!")
 
-client.event(on_ready)
-
-async def main():
-    print("Inside main()")
-    async with client:
-        print("Starting client...")
-        await client.start(DISCORD_BOT_TOKEN)
-        print("Bot has started.")
 
-        # Keep the main function alive:
-        while True:
-            await asyncio.sleep(60)  # Sleep for 60 seconds
+# Get the token from an environment variable
+token = os.getenv("DISCORD_BOT_TOKEN")
 
-if __name__ == "__main__":
-    print("Running main()")
-    asyncio.run(main())
+# check if the token is a string
+if isinstance(token, str):
+    client.run(token)
+else:
+    print("Invalid token format")