Update app.py

app.py

@@ -17,6 +17,9 @@ intents.message_content = True
 
 bot = commands.Bot(command_prefix='!', intents=intents)
 
+MAX_WORKERS = 5
+
+worker_semaphore = asyncio.Semaphore(MAX_WORKERS)
 
 async def jojo(ctx):
     start_time = time.time()  
@@ -28,19 +31,40 @@ async def jojo(ctx):
     await ctx.send(f"{style} image generated in {generation_time:.2f} seconds.")
     await ctx.send(file=discord.File(im))
 
+async def process_requests(queue):
+    while True:
+        ctx = await queue.get()
+        async with worker_semaphore:
+            await jojo(ctx)
+        queue.task_done()
 
 @bot.command()
 async def command(ctx, num_requests: int):
-    tasks = []
+    start_time = time.time() 
+    queue = asyncio.Queue()
+
+    # Start the worker tasks
+    worker_tasks = []
+    for _ in range(MAX_WORKERS):
+        task = asyncio.create_task(process_requests(queue))
+        worker_tasks.append(task)
 
+    # Add the requests to the queue
     for _ in range(num_requests):
-        task = asyncio.create_task(jojo(ctx))
-        tasks.append(task)
+        await queue.put(ctx)
 
-    await asyncio.gather(*tasks)
-    await ctx.send("Command executed.")
+    # Wait for all requests to be processed
+    await queue.join()
 
+    # Cancel the worker tasks
+    for task in worker_tasks:
+        task.cancel()
 
+    await ctx.send("Command executed.")
+    end_time = time.time()
+    generation_time = end_time - start_time    
+    await ctx.send(f"{style} everything done in {generation_time:.2f} seconds.")    
+    
 def run_bot():
     bot.run(DISCORD_TOKEN)