Update app.py

app.py

@@ -45,7 +45,7 @@ async def on_ready():
 
 @bot.command(name='generate')
 @commands.cooldown(rate=1, per=10, type=commands.BucketType.user)
-async def generate(ctx, *description):
+async def generate(ctx, *, description: str):
     """
     Command to generate an image based on a description using the Hugging Face model.
     """
@@ -53,7 +53,6 @@ async def generate(ctx, *description):
         await ctx.send("You need to provide a description for the image.")
         return
 
-    description = " ".join(description)
     try:
         payload = {"inputs": description}
         image_bytes = query(payload)
@@ -82,7 +81,19 @@ async def generate_error(ctx, error):
 @commands.has_permissions(administrator=True)
 async def set_prefix(ctx, prefix: str):
     bot.command_prefix = commands.when_mentioned_or(prefix)
-    await ctx.sen
+    await ctx.send(f"Command prefix changed to: {prefix}")
 
+@bot.command(name='help_custom')
+async def help_custom(ctx):
+    help_text = (
+        "Here are the commands you can use:\n"
+        "!generate <description> - Generate an image based on the description.\n"
+        "!setprefix <prefix> - Change the command prefix (admin only).\n"
+        "!help_custom - Display this help message."
+    )
+    await ctx.send(help_text)
+
+# Run the bot
+bot.run(DISCORD_TOKEN)