Update app.py

app.py

@@ -4,36 +4,25 @@ import os
 import requests
 import asyncio
 import aiohttp  # Use aiohttp for asynchronous HTTP requests
+import gradio as gr  # Import Gradio
 
 # --- 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."
-    )
+    raise ValueError("Both DISCORD_BOT_TOKEN and GLIF_API_TOKEN must be set.")
 
 # --- Discord Bot Setup ---
 intents = discord.Intents.default()
-# intents.message_content = True  # If you need message content intent
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
 
-# --- Asynchronous GLIF API Interaction with aiohttp ---
+
 async def generate_image_async(prompt, aspect_ratio):
-    """
-    Generates an image using the GLIF API asynchronously.
-    Args:
-        prompt: The text prompt for image generation.
-        aspect_ratio: The desired aspect ratio (e.g., "1:1", "16:9").
-    Returns:
-        The URL of the generated image, or an error message if the API request failed.
-    """
     payload = {
-        "id": "cm3ugmzv2002gnckiosrwk6xi",  # Your GLIF model ID
+        "id": "cm3ugmzv2002gnckiosrwk6xi",
         "inputs": [prompt, aspect_ratio],
     }
     headers = {"Authorization": f"Bearer {GLIF_API_TOKEN}"}
@@ -42,9 +31,8 @@ async def generate_image_async(prompt, aspect_ratio):
         try:
             async with session.post(
                 GLIF_API_URL, json=payload, headers=headers, timeout=15
-            ) as response:  # Increased timeout to 15 seconds
+            ) as response:
                 response.raise_for_status()
-
                 response_data = await response.json()
                 if "output" in response_data:
                     return response_data["output"]
@@ -58,19 +46,13 @@ async def generate_image_async(prompt, aspect_ratio):
         except aiohttp.ClientError as e:
             return f"API request failed: {e}"
 
-# --- Discord Slash Commands ---
-@tree.command(
-    name="generate", description="Generates an image based on a text prompt"
-)
+
+@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(
@@ -79,50 +61,56 @@ async def generate_command(
     aspect_ratio: app_commands.Choice[str],
 ):
     try:
-        await interaction.response.defer()  # MUST be the first await
-
-        # ... rest of your code ... 
+        await interaction.response.defer()
+
+        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}"
+            )
     except Exception as e:
-        print(e)
-    """Generates an image based on the user's prompt and aspect ratio."""
-    # 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}"
-        )
+        await interaction.followup.send(f"Error: {str(e)}")
+
 
 @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 on_ready():
     await tree.sync()
     print("Bot is ready!")
 
+
 client.event(on_ready)
 
+
+# --- Gradio Interface ---
+def echo_text(text):
+    return text
+
+
+def run_gradio():
+    gr.Interface(
+        fn=echo_text,
+        inputs="text",
+        outputs="text",
+        live=False,
+        title="Minimal Gradio Interface",
+    ).launch(server_name="0.0.0.0", server_port=7860, share=False, show_error=True)
+
+
+# --- Main ---
 async def main():
-    print("Inside main()")
-    async with client:
-        print("Starting client...")
-        await client.start(DISCORD_BOT_TOKEN)
-        print("Bot has started.")
+    bot_task = asyncio.create_task(client.start(DISCORD_BOT_TOKEN))
+    gradio_task = asyncio.to_thread(run_gradio)
+
+    await asyncio.gather(bot_task, gradio_task)
 
-        # Keep the main function alive:
-        while True:
-            await asyncio.sleep(60)  # Sleep for 60 seconds
 
 if __name__ == "__main__":
-    print("Running main()")
     asyncio.run(main())