Update app.py

app.py

@@ -1,95 +1,105 @@
 import discord
 from discord import app_commands
 import os
-from dotenv import load_dotenv
 import requests
+import asyncio
+from threading import Thread
 
-load_dotenv()
+# NOTE: This example requires the "intents" extra to be installed.
+#       By default, discord.py does not install it.
+#       To install it, run the following command in your terminal:
+#       python -m pip install -U discord.py[intents]
 
-# Set up intents
+# --- Environment Variables & Setup ---
+DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
+GLIF_API_TOKEN = os.getenv("GLIF_API_TOKEN")
+GLIF_API_URL = "https://simple-api.glif.app"
+
+# Check if tokens are set
+if not DISCORD_BOT_TOKEN or not GLIF_API_TOKEN:
+    raise ValueError(
+        "Both DISCORD_BOT_TOKEN and GLIF_API_TOKEN must be set as environment variables."
+    )
+
+# --- Discord Bot Setup ---
 intents = discord.Intents.default()
+# intents.message_content = True  # If you need message content intent in the future
 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
+# --- GLIF API Interaction ---
 def generate_image(prompt, aspect_ratio):
-    # Prepare the payload for the GLIF API
     payload = {
-        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Glif ID
-        "inputs": [prompt, aspect_ratio]
+        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Your GLIF model ID
+        "inputs": [prompt, aspect_ratio],
     }
+    headers = {"Authorization": f"Bearer {GLIF_API_TOKEN}"}
 
-    # 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]):
+# --- Discord Slash Commands ---
+@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
+    await interaction.response.defer()
 
     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}")
+                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!")
 
+# --- Bot Initialization and Event Loop ---
+async def start_bot():
+    await client.start(DISCORD_BOT_TOKEN)
 
-@client.event
 async def on_ready():
-    await tree.sync()  # Sync the slash commands with discord
+    await tree.sync()
     print("Bot is ready!")
 
+client.event(on_ready)
 
-# Get the token from an environment variable
-token = os.getenv("DISCORD_BOT_TOKEN")
+def run_bot():
+    asyncio.run(start_bot())
 
-# check if the token is a string
-if isinstance(token, str):
-    client.run(token)
-else:
-    print("Invalid token format")
+if __name__ == "__main__":
+    # Run the bot in a separate thread
+    bot_thread = Thread(target=run_bot)
+    bot_thread.start()