Update app.py

app.py

@@ -1,10 +1,10 @@
 import discord
 from discord import app_commands
-from discord.ui import Button, View
 import os
+import requests
 import asyncio
-import aiohttp
-import gradio as gr
+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")
@@ -19,7 +19,7 @@ intents = discord.Intents.default()
 client = discord.Client(intents=intents)
 tree = app_commands.CommandTree(client)
 
-# --- Image Generation Function ---
+
 async def generate_image_async(prompt, aspect_ratio):
     payload = {
         "id": "cm3ugmzv2002gnckiosrwk6xi",
@@ -46,7 +46,7 @@ async def generate_image_async(prompt, aspect_ratio):
         except aiohttp.ClientError as e:
             return f"API request failed: {e}"
 
-# --- Discord Command: /generate ---
+
 @tree.command(name="generate", description="Generates an image based on a text prompt")
 @app_commands.choices(
     aspect_ratio=[
@@ -68,31 +68,9 @@ async def generate_command(
         await interaction.response.defer()
 
         image_url_or_error = await generate_image_async(prompt, aspect_ratio.value)
-
         if image_url_or_error.startswith("http"):
-            # Format the prompt for display with expand/collapse
-            formatted_prompt, initial_content, full_content = format_prompt_for_discord(prompt)
-
-            # Create buttons
-            async def show_more_callback(interaction):
-                await interaction.response.edit_message(content=f"{full_content}\n{image_url_or_error}", view=view)
-
-            async def show_less_callback(interaction):
-                await interaction.response.edit_message(content=f"{initial_content}\n{image_url_or_error}", view=view)
-
-            show_more_button = Button(label="Read More", style=discord.ButtonStyle.primary)
-            show_more_button.callback = show_more_callback
-            show_less_button = Button(label="Hide", style=discord.ButtonStyle.secondary)
-            show_less_button.callback = show_less_callback
-
-            view = View()
-            if len(prompt) > 256:
-                view.add_item(show_more_button)
-            view.add_item(show_less_button)
-
             await interaction.followup.send(
-                f"{initial_content}\n{image_url_or_error}", # Include image URL initially
-                view=view
+                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(
@@ -101,35 +79,25 @@ async def generate_command(
     except Exception as e:
         await interaction.followup.send(f"Error: {str(e)}")
 
-# --- Helper Function: Format Prompt for Discord ---
-def format_prompt_for_discord(prompt):
-    """Formats the prompt for display in Discord with expand/collapse functionality using buttons."""
-    max_length = 256
-    if len(prompt) <= max_length:
-        return prompt, f"**Prompt:** {prompt}", f"**Prompt:** {prompt}"
 
-    truncated_prompt = prompt[:max_length]
-    initial_content = f"**Prompt:** {truncated_prompt}..."
-    full_content = f"**Prompt:** {prompt}"
-
-    return prompt, initial_content, full_content
-
-# --- Discord Command: /hello ---
 @tree.command(name="hello", description="Says hello!")
-async def hello_command(interaction: discord.Interaction):
+async def hello_command(interaction):
     await interaction.response.send_message("Hello there!")
 
-# --- Bot Ready Event ---
+
 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,
@@ -139,6 +107,7 @@ def run_gradio():
         title="Minimal Gradio Interface",
     ).launch(server_name="0.0.0.0", server_port=7860, share=False, show_error=True)
 
+
 # --- Main ---
 async def main():
     bot_task = asyncio.create_task(client.start(DISCORD_BOT_TOKEN))
@@ -146,5 +115,6 @@ async def main():
 
     await asyncio.gather(bot_task, gradio_task)
 
+
 if __name__ == "__main__":
     asyncio.run(main())