Update app.py

app.py

@@ -163,6 +163,7 @@ async def jojo(ctx):
     try:  
         if await safetychecks(ctx): #✅
             await ctx.message.add_reaction('🤖') 
+            await ctx.message.add_reaction('⏳') 
             thread = await ctx.message.create_thread(name=f'{ctx.author} Jojo Thread')         
             if ctx.message.attachments:
                 await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...yare yare, daze ...')  
@@ -175,11 +176,13 @@ async def jojo(ctx):
                 await ctx.message.add_reaction('✅') # img + face    
             else: # no image
                 await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
-                await ctx.message.add_reaction('❌')             
+                await ctx.message.add_reaction('❌') 
+                await ctx.message.remove_reaction('⏳', bot.user)
     except Exception as e: # no generation / img + no face
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention}Error: {e}")
-        await ctx.message.add_reaction('❌')            
+        await ctx.message.add_reaction('❌')  
+        await ctx.message.remove_reaction('⏳', bot.user)
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # Disney ❌
@@ -187,7 +190,8 @@ async def jojo(ctx):
 async def disney(ctx):
     try:
         if await safetychecks(ctx): #✅
-            await ctx.message.add_reaction('🤖') 
+            await ctx.message.add_reaction('🤖')
+            await ctx.message.add_reaction('⏳') 
             thread = await ctx.message.create_thread(name=f'{ctx.author} disney Thread')         
             if ctx.message.attachments:
                 await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')  
@@ -198,11 +202,13 @@ async def disney(ctx):
                 await ctx.message.add_reaction('✅') # img + face    
             else: # no image
                 await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
-                await ctx.message.add_reaction('❌')             
+                await ctx.message.add_reaction('❌') 
+                await ctx.message.remove_reaction('⏳', bot.user)
     except Exception as e: # no generation / img + no face
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention}Error: {e}")
         await ctx.message.add_reaction('❌') 
+        await ctx.message.remove_reaction('⏳', bot.user)
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # Spider-Verse ✅
@@ -211,6 +217,7 @@ async def spidey(ctx):
     try:
         if await safetychecks(ctx): #✅
             await ctx.message.add_reaction('🤖') 
+            await ctx.message.add_reaction('⏳') 
             thread = await ctx.message.create_thread(name=f'{ctx.author} spider-verse Thread')         
             if ctx.message.attachments:
                 await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')  
@@ -221,11 +228,13 @@ async def spidey(ctx):
                 await ctx.message.add_reaction('✅') # img + face    
             else: # no image
                 await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
-                await ctx.message.add_reaction('❌')             
+                await ctx.message.add_reaction('❌')  
+                await ctx.message.remove_reaction('⏳', bot.user)
     except Exception as e: # no generation / img + no face
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention}Error: {e}")
         await ctx.message.add_reaction('❌') 
+        await ctx.message.remove_reaction('⏳', bot.user)
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # sketch ✅
@@ -234,6 +243,7 @@ async def sketch(ctx):
     try:    
         if await safetychecks(ctx): #✅
             await ctx.message.add_reaction('🤖') 
+            await ctx.message.add_reaction('⏳') 
             thread = await ctx.message.create_thread(name=f'{ctx.author} sketch Thread')         
             if ctx.message.attachments:
                 await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')  
@@ -244,11 +254,13 @@ async def sketch(ctx):
                 await ctx.message.add_reaction('✅') # img + face    
             else: # no image
                 await thread.send(f"{ctx.author.mention}No attachments to be found...Can't animify dat! Try sending me an image 😉")
-                await ctx.message.add_reaction('❌')             
+                await ctx.message.add_reaction('❌') 
+                await ctx.message.remove_reaction('⏳', bot.user)
     except Exception as e: # no generation / img + no face
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention}Error: {e}")
         await ctx.message.add_reaction('❌') 
+        await ctx.message.remove_reaction('⏳', bot.user)
 #----------------------------------------------------------------------------------------------------------------------------------------------
 # deepfloydif stage 1 generation ✅
 def inference(prompt):
@@ -314,7 +326,7 @@ async def deepfloydif(ctx, *, prompt: str):
             print(f"Error: {e}")
             thread = await ctx.message.create_thread(name=f'DFIF1 Error')
             await thread.send(f"{ctx.author.mention} Error during stage 1 pre-generation.")
-            await ctx.message.remove_reaction('⏳')
+            await ctx.message.remove_reaction('⏳', bot.user)
             await ctx.message.add_reaction(failure_emoji)
             
         #generation✅-------------------------------------------------------
@@ -336,7 +348,7 @@ async def deepfloydif(ctx, *, prompt: str):
             print(f"Error: {e}")
             thread = await ctx.message.create_thread(name=f'Generation Error')
             await thread.send(f"{ctx.author.mention} Error during stage 1 generation.")
-            await ctx.message.remove_reaction('⏳')
+            await ctx.message.remove_reaction('⏳', bot.user)
             await ctx.message.add_reaction(failure_emoji)            
         #posting images✅----------------------------------------------------------------    
         try:
@@ -394,7 +406,7 @@ async def deepfloydif(ctx, *, prompt: str):
             print(f"Error: {e}")
             thread = await ctx.message.create_thread(name=f'Posting Error')
             await thread.send(f"{ctx.author.mention} Encountered error while posting combined image in thread.")
-            await ctx.message.remove_reaction('⏳')
+            await ctx.message.remove_reaction('⏳', bot.user)
             await ctx.message.add_reaction(failure_emoji)
             
     #deepfloydif try/except
@@ -402,7 +414,7 @@ async def deepfloydif(ctx, *, prompt: str):
         print(f"Error: {e}")
         thread = await ctx.message.create_thread(name=f'deepfloydif Error')
         await thread.send(f"{ctx.author.mention} Overall error with deepfloydif.")
-        await ctx.message.remove_reaction('⏳')
+        await ctx.message.remove_reaction('⏳', bot.user)
         await ctx.message.add_reaction(failure_emoji)
 
 #----------------------------------------------------------------------------------------------------------------------------
@@ -427,7 +439,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         await dfif_command_message.add_reaction(success_emoji)
-        await dfif_command_message.remove_reaction('⏳')
+        await dfif_command_message.remove_reaction('⏳', bot.user)
 
     except Exception as e:
         print(f"Error: {e}")
@@ -438,7 +450,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         await dfif_command_message.add_reaction(failure_emoji)  
-        await dfif_command_message.remove_reaction('⏳')
+        await dfif_command_message.remove_reaction('⏳', bot.user)
         await thread.send(f"Error during stage 2 upscaling.") 
 #----------------------------------------------------------------------------------------------------------------------------
 # react detector for stage 2 ✅