activethreadcount = active_count()

app.py

@@ -1,6 +1,7 @@
 import discord
 import os
 import threading
+from threading import active_count
 import gradio as gr 
 import requests
 import json
@@ -37,6 +38,12 @@ async def lunar(ctx):
 
 @bot.command()
 async def jojo(ctx):
+
+    # get the number of active threads
+    activethreadcount = active_count()
+    # report the number of active threads
+    print(activethreadcount)
+    
     attachment = await discord_before(ctx)
     if attachment:
         asyncio.create_task(generation(ctx, attachment))
@@ -49,6 +56,12 @@ async def discord_before(ctx):
             ctx.message.add_reaction('⌛')  
         )
         attachment = ctx.message.attachments[0]
+
+        # get the number of active threads
+        activethreadcount = active_count()
+        # report the number of active threads
+        print(activethreadcount)
+        
         return attachment
     else:
         await asyncio.gather(
@@ -63,6 +76,12 @@ async def generation(ctx, attachment):
         im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, style)
         #im = await asyncio.to_thread(jojogan.predict, attachment.url, style)
         #im = jojogan.predict(attachment.url, style)
+
+        # get the number of active threads
+        activethreadcount = active_count()
+        # report the number of active threads
+        print(activethreadcount)
+        
         await asyncio.gather(
             ctx.send(f'{ctx.author.mention} Here is the jojo version of it'),
             ctx.send(file=discord.File(im)),
@@ -70,13 +89,7 @@ async def generation(ctx, attachment):
             ctx.message.add_reaction('✅')
         )   
 
-async def discord_after(ctx, im):
-    await asyncio.gather(
-        ctx.send(f'{ctx.author.mention} Here is the jojo version of it'),
-        ctx.send(file=discord.File(im)),
-        ctx.message.remove_reaction('⌛', bot.user),
-        ctx.message.add_reaction('✅')
-    )  
+
 
  
 
@@ -86,8 +99,12 @@ def get_total_threads():
     return multiprocessing.cpu_count()
 
 total_threads = get_total_threads()
-print("Total Threads:", total_threads)
+print("Total cpus:", total_threads)
 
+# get the number of active threads
+activethreadcount = active_count()
+# report the number of active threads
+print(activethreadcount)
     
 def run_bot():
     bot.run(DISCORD_TOKEN)
@@ -104,7 +121,13 @@ demo.launch()
 
 
 '''
- 
+ async def discord_after(ctx, im):
+    await asyncio.gather(
+        ctx.send(f'{ctx.author.mention} Here is the jojo version of it'),
+        ctx.send(file=discord.File(im)),
+        ctx.message.remove_reaction('⌛', bot.user),
+        ctx.message.add_reaction('✅')
+    )  
 
 '''