Update app.py

app.py

@@ -1,36 +1,22 @@
 import discord
 import os
-import threading
 import gradio as gr
-import requests
-import json
-import random
-import time
-import re
-from discord import Embed, Color
+import asyncio
 from discord.ext import commands
 from gradio_client import Client
 from PIL import Image
-#from ratelimiter import RateLimiter
-
-import asyncio
-import httpx
-
-from tasks import make_request_async
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
 DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
 
 jojogan = Client("akhaliq/JoJoGAN", DFIF_TOKEN)
 
-
 intents = discord.Intents.default()
 intents.message_content = True
 
 bot = commands.Bot(command_prefix='!', intents=intents)
 
 
-#---------------------------------------------------------------------------------------------------------------------------
 async def jojo(ctx):
     start_time = time.time()  
     style = 'JoJo'
@@ -40,30 +26,36 @@ async def jojo(ctx):
     generation_time = end_time - start_time    
     await ctx.send(f"{style} image generated in {generation_time:.2f} seconds.")
     await ctx.send(file=discord.File(im))
-#----------------------------------------------------------------------------------------------------------------------------
+
 
 @bot.command()
-async def command(ctx):
-    threads = []
-    
-    for i in range(5):
-        thread = threading.Thread(
-            target=jojo,
-            args=[ctx]
-        )
-        thread.start()
-        threads.append(thread)
-    
-    for thread in threads:
-        thread.join()
+async def command(ctx, num_requests: int):
+    tasks = []
+
+    for _ in range(num_requests):
+        task = asyncio.create_task(jojo(ctx))
+        tasks.append(task)
+
+    await asyncio.gather(*tasks)
+    await ctx.send("Command executed.")
+
 
 def run_bot():
     bot.run(DISCORD_TOKEN)
 
-threading.Thread(target=run_bot).start()
 
-def greet(name):
-    return "Hello " + name + "!"
+async def run_gradio_interface():
+    def greet(name):
+        return "Hello " + name + "!"
+
+    demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+    await demo.launch()
+
+
+async def main():
+    bot_task = asyncio.create_task(run_bot())
+    gradio_task = asyncio.create_task(run_gradio_interface())
+    await asyncio.gather(bot_task, gradio_task)
+
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+asyncio.run(main())