check marks / failure reactions for more responsiveness

app.py

@@ -56,9 +56,9 @@ async def deepfloydif(ctx, *, prompt: str):
             return user == ctx.author and str(reaction.emoji) in ['1️⃣', '2️⃣', '3️⃣', '4️⃣']   
 
         await ctx.message.add_reaction('👍')
-        thread = await ctx.message.create_thread(name=f'Image Upscaling Thread {ctx.author.mention}')
+        thread = await ctx.message.create_thread(name=f'{ctx.author.} Image Upscaling Thread ')
         # create thread -> send new message inside thread + combined_image -> add reactions -> dfif2
-        await thread.send(f'Creating thread...{ctx.author.mention}')
+        await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
         
         number_of_images = 4
         current_time = int(time.time())
@@ -114,7 +114,7 @@ async def deepfloydif(ctx, *, prompt: str):
 
         #4
         with open(combined_image_path, 'rb') as f:
-            threadmsg = await thread.send(f'Here is the combined image. React with the image number you want to upscale! {ctx.author.mention}', file=discord.File(f, 'combined_image.png'))          
+            threadmsg = await thread.send(f'{ctx.author.mention}React with the image number you want to upscale!', file=discord.File(f, 'combined_image.png'))          
             
         # bot reacts with appropriate emojis to the post, both showing the user what options they have,
         # as well as showing which post to react to.
@@ -123,19 +123,19 @@ async def deepfloydif(ctx, *, prompt: str):
 
         reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
         if str(reaction.emoji) == '1️⃣':
-            await thread.send(f"Upscaling the first image...{ctx.author.mention}")
+            await thread.send(f"{ctx.author.mention}Upscaling the first image...")
             index = 0
             await dfif2(ctx, index, stage_1_result_path, thread)
         elif str(reaction.emoji) == '2️⃣':
-            await thread.send(f"Upscaling the second image...{ctx.author.mention}")
+            await thread.send(f"{ctx.author.mention}Upscaling the second image...")
             index = 1
             await dfif2(ctx, index, stage_1_result_path, thread)            
         elif str(reaction.emoji) == '3️⃣':
-            await thread.send(f"Upscaling the third image...{ctx.author.mention}")
+            await thread.send(f"{ctx.author.mention}Upscaling the third image...")
             index = 2
             await dfif2(ctx, index, stage_1_result_path, thread)            
         elif str(reaction.emoji) == '4️⃣':
-            await thread.send(f"Upscaling the fourth image...{ctx.author.mention}")
+            await thread.send(f"{ctx.author.mention}Upscaling the fourth image...")
             index = 3
             await dfif2(ctx, index, stage_1_result_path, thread)            
 
@@ -143,6 +143,7 @@ async def deepfloydif(ctx, *, prompt: str):
     except Exception as e:
         print(f"Error: {e}")
         await ctx.reply('An error occurred while processing your request. Please wait 5 seconds before retrying.')
+        await ctx.message.add_reaction('❌')
 
 #----------------------------------------------------------------------------------------------------------------------------
 # Stage 2
@@ -158,12 +159,17 @@ async def dfif2(ctx, index: int, stage_1_result_path, thread):
     
         
         with open(result_path, 'rb') as f:
-            await thread.send(f'Here is the result of the second stage {ctx.author.mention}', file=discord.File(f, 'result.png'))
+            await thread.send(f'{ctx.author.mention}Here is the upscaled image! :) ', file=discord.File(f, 'result.png'))
             #await ctx.reply('Here is the result of the second stage', file=discord.File(f, 'result.png'))
+            await ctx.message.add_reaction('✔️')
+            
+
+        
 
     except Exception as e:
         print(f"Error: {e}")
         await ctx.reply('An error occurred while processing stage 2 upscaling. Please try again later.')
+        await ctx.message.add_reaction('❌')
 #----------------------------------------------------------------------------------------------------------------------------