Update app.py

app.py

@@ -1,7 +1,7 @@
 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
@@ -19,6 +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",
@@ -45,6 +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=[
@@ -66,12 +68,31 @@ 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 = format_prompt_for_discord(prompt)
+            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=full_content, view=view)
+
+            async def show_less_callback(interaction):
+                await interaction.response.edit_message(content=initial_content, 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"Here's your generated image based on the prompt with aspect ratio {aspect_ratio.name}:\n{formatted_prompt}\n{image_url_or_error}"
+                f"{initial_content}\n{image_url_or_error}",
+                view=view
             )
         else:
             await interaction.followup.send(
@@ -80,27 +101,27 @@ 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 spoiler tags."""
-    max_length = 256  # Adjust as needed
+    """Formats the prompt for display in Discord with expand/collapse functionality using buttons."""
+    max_length = 256
     if len(prompt) <= max_length:
-        return f"**Prompt:** {prompt}"  # No need to truncate
+        return prompt, f"**Prompt:** {prompt}", f"**Prompt:** {prompt}"
 
     truncated_prompt = prompt[:max_length]
     hidden_part = prompt[max_length:]
 
-    formatted_message = (
-        f"**Prompt:** {truncated_prompt}...\n"
-        f"<details><summary>Read More</summary>{hidden_part}</details>\n"
-        f"<details><summary>Hide</summary></details>"
-    )
+    initial_content = f"**Prompt:** {truncated_prompt}..."
+    full_content = f"**Prompt:** {prompt}"
 
-    return formatted_message
+    return prompt, initial_content, full_content
 
+# --- Discord Command: /hello ---
 @tree.command(name="hello", description="Says hello!")
-async def hello_command(interaction):
+async def hello_command(interaction: discord.Interaction):
     await interaction.response.send_message("Hello there!")
 
+# --- Bot Ready Event ---
 async def on_ready():
     await tree.sync()
     print("Bot is ready!")