testing

app.py

@@ -52,12 +52,14 @@ GRADIO_CLIENT = Client("Nevaehni/FLUX.1-schnell", hf_token=HF_TOKEN)
 # Initialize Hugging Face API client
 hf_api = HfApi()
 
+
 @bot.event
 async def on_ready():
     """Event handler triggered when the bot is ready."""
     logger.info(f'Logged in as {bot.user} (ID: {bot.user.id})')
     logger.info('------')
 
+
 def parse_prompt(command: str) -> str:
     """
     Parse the prompt from the command string.
@@ -73,6 +75,7 @@ def parse_prompt(command: str) -> str:
         return match.group(1).strip()
     return ''
 
+
 def create_example_embed() -> Embed:
     """
     Create an embed message with an example !generate command.
@@ -88,6 +91,7 @@ def create_example_embed() -> Embed:
     )
     return embed
 
+
 @bot.command(name='generate')
 async def generate(ctx: commands.Context, *, args: str = None):
     """
@@ -205,6 +209,19 @@ async def generate(ctx: commands.Context, *, args: str = None):
         else:
             # Response is neither dict, str, nor list
             await ctx.send("❌ **Error:** Unexpected response format from the API.")
+
+        # **Delete the user's command message after successfully processing**
+        try:
+            await ctx.message.delete()
+            logger.info(f"Deleted command message from user {ctx.author} (ID: {ctx.author.id}).")
+        except discord.Forbidden:
+            logger.warning("Insufficient permissions to delete the user's message.")
+            await ctx.send("⚠️ **Warning:** I don't have permission to delete your command message.")
+        except discord.NotFound:
+            logger.warning("The message to delete was not found.")
+        except Exception as e:
+            logger.error(f"Unexpected error when trying to delete the message: {e}")
+
     except asyncio.TimeoutError:
         logger.error("API request timed out.")
         await ctx.send("⏰ **Error:** The request to the API timed out. Please try again later.")
@@ -219,7 +236,16 @@ async def generate(ctx: commands.Context, *, args: str = None):
         await ctx.send(f"❌ **Error:** An unexpected error occurred: {str(e)}")
     finally:
         # Delete the processing message
-        await processing_message.delete()
+        try:
+            await processing_message.delete()
+            logger.info("Deleted the processing message.")
+        except discord.Forbidden:
+            logger.warning("Insufficient permissions to delete the processing message.")
+        except discord.NotFound:
+            logger.warning("The processing message to delete was not found.")
+        except Exception as e:
+            logger.error(f"Unexpected error when trying to delete the processing message: {e}")
+
 
 @bot.event
 async def on_command_error(ctx: commands.Context, error):
@@ -238,9 +264,11 @@ async def on_command_error(ctx: commands.Context, error):
         await ctx.send(f"❌ **Error:** {str(error)}")
         logger.error(f"Unhandled command error: {str(error)}")
 
+
 async def handle_root(request):
     return web.Response(text="DarkMuse GOES VROOOOM", status=200)
 
+
 async def start_web_server():
     app = web.Application()
     app.router.add_get('/', handle_root)
@@ -250,9 +278,11 @@ async def start_web_server():
     await site.start()
     logger.info("Web server started on port 7860")
 
+
 async def start_bot():
     await bot.start(DISCORD_BOT_TOKEN)
 
+
 async def schedule_restart():
     """
     Schedule the bot to restart the Hugging Face Space at a specified interval.
@@ -273,6 +303,7 @@ async def schedule_restart():
         # Optional: Add a short delay to prevent rapid retries in case of failure
         await asyncio.sleep(5)  # Wait 5 seconds before the next restart cycle
 
+
 async def main():
     # Start the scheduler as a background task
     asyncio.create_task(schedule_restart())
@@ -282,6 +313,7 @@ async def main():
         start_web_server()
     )
 
+
 # Run the bot and web server concurrently
 if __name__ == '__main__':
     try: