Update app.py

app.py

@@ -3,6 +3,7 @@ from discord import app_commands
 import os
 import requests
 import asyncio
+import aiohttp  # Use aiohttp for asynchronous HTTP requests
 
 # --- Environment Variables & Setup ---
 DISCORD_BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
@@ -17,14 +18,14 @@ if not DISCORD_BOT_TOKEN or not GLIF_API_TOKEN:
 
 # --- Discord Bot Setup ---
 intents = discord.Intents.default()
-# intents.message_content = True  # Uncomment if you need message content intent
+# intents.message_content = True  # If you need message content intent
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
 
-# --- GLIF API Interaction ---
-def generate_image(prompt, aspect_ratio):
+# --- Asynchronous GLIF API Interaction with aiohttp ---
+async def generate_image_async(prompt, aspect_ratio):
     """
-    Generates an image using the GLIF API based on the given prompt and aspect ratio.
+    Generates an image using the GLIF API asynchronously.
 
     Args:
         prompt: The text prompt for image generation.
@@ -39,20 +40,25 @@ def generate_image(prompt, aspect_ratio):
     }
     headers = {"Authorization": f"Bearer {GLIF_API_TOKEN}"}
 
-    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)
-
-        response_data = response.json()
-        if "output" in response_data:
-            return response_data["output"]
-        elif "error" in response_data:
-            return f"Error: {response_data['error']}"
-        else:
-            return "Error: Unexpected response from GLIF API."
-
-    except requests.exceptions.RequestException as e:
-        return f"API request failed: {e}"
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.post(
+                GLIF_API_URL, json=payload, headers=headers, timeout=15
+            ) as response:  # Increased timeout to 15 seconds
+                response.raise_for_status()
+
+                response_data = await response.json()
+                if "output" in response_data:
+                    return response_data["output"]
+                elif "error" in response_data:
+                    return f"Error: {response_data['error']}"
+                else:
+                    return "Error: Unexpected response from GLIF API."
+
+        except asyncio.TimeoutError:
+            return "Error: GLIF API request timed out."
+        except aiohttp.ClientError as e:
+            return f"API request failed: {e}"
 
 # --- Discord Slash Commands ---
 @tree.command(
@@ -75,24 +81,25 @@ async def generate_command(
     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}")
+    await interaction.response.defer()  # Acknowledge within 3 seconds
+
+    # Call the asynchronous image generation function
+    image_url_or_error = await generate_image_async(
+        prompt, aspect_ratio.value
+    )
+
+    if image_url_or_error.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_or_error}"
+        )
+    else:
+        await interaction.followup.send(
+            f"Sorry, I couldn't generate an image. {image_url_or_error}"
+        )
 
 @tree.command(name="hello", description="Says hello!")
 async def hello_command(interaction):
-    await interaction.response.send_message("Hello there!I think I got this!")
+    await interaction.response.send_message("Hello there!")
 
 # --- Bot Initialization and Event Loop ---
 async def on_ready():