Update app.py

app.py

@@ -1,95 +1,117 @@
 import discord
 from discord import app_commands
 import os
-from dotenv import load_dotenv
 import requests
+import asyncio
 
-load_dotenv()
+# --- 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 in the Hugging Face Space's settings."
+    )
 
-# Set up intents
+# --- Discord Bot Setup ---
 intents = discord.Intents.default()
+# intents.message_content = True  # Uncomment if you need message content intent
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
 
-# GLIF Simple API endpoint
-GLIF_API_URL = "https://simple-api.glif.app"
+# --- GLIF API Interaction ---
+def generate_image(prompt, aspect_ratio):
+    """
+    Generates an image using the GLIF API based on the given prompt and aspect ratio.
 
+    Args:
+        prompt: The text prompt for image generation.
+        aspect_ratio: The desired aspect ratio (e.g., "1:1", "16:9").
 
-# Function to call the GLIF API
-def generate_image(prompt, aspect_ratio):
-    # Prepare the payload for the GLIF API
+    Returns:
+        The URL of the generated image, or an error message if the API request failed.
+    """
     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)
+    try:
+        response = requests.post(GLIF_API_URL, json=payload, headers=headers)
+        response.raise_for_status()  # Raise an exception for bad status codes (4xx or 5xx)
 
-    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]):
+        else:
+            return "Error: Unexpected response from GLIF API."
+
+    except requests.exceptions.RequestException as e:
+        return f"API request failed: {e}"
+
+# --- 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
 
     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!")
 
-
-@client.event
+# --- Bot Initialization and Event Loop ---
 async def on_ready():
-    await tree.sync()  # Sync the slash commands with discord
+    await tree.sync()
     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.")
 
-# Get the token from an environment variable
-token = os.getenv("DISCORD_BOT_TOKEN")
+        # Keep the main function alive:
+        while True:
+            await asyncio.sleep(60)  # Sleep for 60 seconds
 
-# check if the token is a string
-if isinstance(token, str):
-    client.run(token)
-else:
-    print("Invalid token format")
+if __name__ == "__main__":
+    print("Running main()")
+    asyncio.run(main())