fixing some redundant thread creation, adding both auto_archive_duration=60 as well as manual instant archiving upon error / stage 2 completion [dfif]

app.py

@@ -205,10 +205,12 @@ async def disney(ctx):
                 await thread.send(f'{ctx.author.mention} Here is the {style} version of it', file=discord.File(im))
                 await ctx.message.add_reaction('<:agree:1098629085955113011>') # img + face 
                 await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
+                await thread.edit(archived=True)
             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('<:disagree:1098628957521313892>') 
                 await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
+                await thread.edit(archived=True)
                 
     except Exception as e: # no generation / img + no face
         await fullqueue(e, thread)       
@@ -216,7 +218,7 @@ async def disney(ctx):
         await thread.send(f"{ctx.author.mention} Error: {e}")
         await ctx.message.add_reaction('<:disagree:1098628957521313892>') 
         await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
-        
+        await thread.edit(archived=True)        
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # Spider-Verse ✅
 @bot.command()
@@ -232,18 +234,20 @@ async def spidey(ctx):
                 im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, style)
                 await thread.send(f'{ctx.author.mention} Here is the {style} version of it', file=discord.File(im))
                 await ctx.message.add_reaction('<:agree:1098629085955113011>') # img + face    
-                await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)                
+                await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user) 
+                await thread.edit(archived=True)                
             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('<:disagree:1098628957521313892>')  
                 await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
+                await thread.edit(archived=True)                
     except Exception as e: # no generation / img + no face
         await fullqueue(e, thread)       
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention} Error: {e}")
         await ctx.message.add_reaction('<:disagree:1098628957521313892>') 
         await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
-        
+        await thread.edit(archived=True)        
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
 # sketch ✅
 @bot.command()
@@ -259,17 +263,20 @@ async def sketch(ctx):
                 im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, 'sketch')
                 await thread.send(f'{ctx.author.mention} Here is the sketch version of it', file=discord.File(im))
                 await ctx.message.add_reaction('<:agree:1098629085955113011>') # img + face 
-                await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)                
+                await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user) 
+                await thread.edit(archived=True)                
             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('<:disagree:1098628957521313892>') 
                 await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
+                await thread.edit(archived=True)                
     except Exception as e: # no generation / img + no face
         await fullqueue(e, thread)  
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention} Error: {e}")
         await ctx.message.add_reaction('<:disagree:1098628957521313892>') 
         await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
+        await thread.edit(archived=True)        
 #----------------------------------------------------------------------------------------------------------------------------------------------
 async def fullqueue(e, thread):
     error_message = str(e)
@@ -338,12 +345,13 @@ async def deepfloydif(ctx, *, prompt: str):
             
         except Exception as e:
             print(f"Error: {e}")
-            thread = await ctx.message.create_thread(name=f'DFIF1 Error')
+            if thread is Null:
+                thread = await ctx.message.create_thread(name=f'DFIF1 Error')
             await thread.send(f"{ctx.author.mention} Error during stage 1 pre-generation.")
             await fullqueue(e, thread)  
             await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
             await ctx.message.add_reaction('<:disagree:1098628957521313892>')
-            
+            await thread.edit(archived=True)            
         #generation✅-------------------------------------------------------
         try:
             #stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
@@ -361,11 +369,13 @@ async def deepfloydif(ctx, *, prompt: str):
             
         except Exception as e:
             print(f"Error: {e}")
-            thread = await ctx.message.create_thread(name=f'Generation Error')
+            if thread is Null:
+                thread = await ctx.message.create_thread(name=f'Generation Error')
             await thread.send(f"{ctx.author.mention} Error during stage 1 generation.")
             await fullqueue(e, thread)  
             await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
-            await ctx.message.add_reaction('<:disagree:1098628957521313892>')            
+            await ctx.message.add_reaction('<:disagree:1098628957521313892>')
+            await thread.edit(archived=True)            
         #posting images✅----------------------------------------------------------------    
         try:
             #await thread.send(f'✅combining images...') 
@@ -426,7 +436,7 @@ async def deepfloydif(ctx, *, prompt: str):
             await fullqueue(e, thread)  
             await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
             await ctx.message.add_reaction('<:disagree:1098628957521313892>')
-            
+            await thread.edit(archived=True)            
     #deepfloydif try/except
     except Exception as e:
         print(f"Error: {e}")
@@ -436,7 +446,7 @@ async def deepfloydif(ctx, *, prompt: str):
         await fullqueue(e, thread)  
         await ctx.message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
         await ctx.message.add_reaction('<:disagree:1098628957521313892>')
-
+        await thread.edit(archived=True)
 #----------------------------------------------------------------------------------------------------------------------------
 # Stage 2 ✅
 async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
@@ -464,6 +474,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         await dfif_command_message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
         await dfif_command_message.add_reaction('<:agree:1098629085955113011>')
+        await thread.edit(archived=True)
 
     except Exception as e:
         print(f"Error: {e}")
@@ -472,7 +483,8 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         await dfif_command_message.remove_reaction('<a:loading:1114111677990981692>', bot.user)
         await dfif_command_message.add_reaction('<:disagree:1098628957521313892>')  
         await thread.send(f"Error during stage 2 upscaling.") 
-        await fullqueue(e, thread)  
+        await fullqueue(e, thread) 
+        await thread.edit(archived=True)
 #----------------------------------------------------------------------------------------------------------------------------
 # react detector for stage 2 ✅
 @bot.event
@@ -522,7 +534,7 @@ async def on_reaction_add(reaction, user):    # ctx = await bot.get_context(reac
                         await dfif2(index, stage_1_result_path, thread, dfif_command_message_id)
 
     except Exception as e:
-        print(f"Error: {e} testing emoji behavior with reacts")
+        print(f"Error: {e} (known error, does not cause issues, fix later)")
 
 
 #----------------------------------------------------------------------------------------------------------------------------