[reactions] custom emoji, channel check

app.py

@@ -23,6 +23,7 @@ import shutil # for doing image movement magic
 
 #todo
 # 4 -> combined image
+# loading emoji? animated emojis?
 # tasks for concurrent coroutines
 # ratelimits
 
@@ -194,7 +195,7 @@ async def deepfloydif(ctx, *, prompt: str):
     try:
         try:
             if await safetychecks(ctx): #✅
-                await ctx.message.add_reaction('🤖')
+                await ctx.message.add_reaction('🤖') # loading emoji?
                 dfif_command_message_id = ctx.message.id # we will use this in some magic later on
                 thread = await ctx.message.create_thread(name=f'{ctx.author} DeepfloydIF Image Upscaling Thread ')
                 # create thread -> send new message inside thread + combined_image -> add reactions -> dfif2
@@ -303,11 +304,18 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         with open(result_path, 'rb') as f:
             await thread.send(f'Here is the upscaled image! :) ', file=discord.File(f, 'result.png'))
 
+        # using custom emoji that looks nicer
+        emoji_guild = thread.guild  
+        confirm_emoji_id = 1098629085955113011
+        confirm_emoji = discord.utils.get(guild.emojis, id=emoji_id)
+        
         # assuming dfif2 is always inside a thread, we can always exit the thread to find the channel with the original command,
         # which allows us to react confirm on that message.
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
-        await dfif_command_message.add_reaction('✅')
+
+        # reacting to original !deepfloydif command + using a custom emoji to do it
+        await dfif_command_message.add_reaction(confirm_emoji)
         await thread.send(f"upscale posted")             
             #await ctx.reply('Here is the result of the second stage', file=discord.File(f, 'result.png'))
             #await ctx.message.add_reaction('✅') need to fix this
@@ -325,7 +333,7 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         #await ctx.message.add_reaction('❌')
 #----------------------------------------------------------------------------------------------------------------------------
 @bot.event
-async def on_reaction_add(reaction, user):
+async def on_reaction_add(reaction, user):    # ctx = await bot.get_context(reaction.message)? could try later, might simplify
     try:
         # safety checks first ❌
         '''
@@ -351,38 +359,39 @@ async def on_reaction_add(reaction, user):
         
         '''
         if not user.bot: 
-            # 811235357663297546 =  lunarflu
-            if reaction.message.attachments:
-                if user.id == 811235357663297546:  #  if user.id == reaction.message.mentions[0].id:           
-                    # magic begins
-                    await reaction.message.channel.send("reaction detected")
-                    attachment = reaction.message.attachments[0]
-                    image_name = attachment.filename # named something like: tmpgtv4qjix1111269940599738479.png
-                    # remove .png first
-                    partialpathmessageid = image_name[:-4] # should be tmpgtv4qjix1111269940599738479 
-                    # extract partialpath, messageid
-                    partialpath = partialpathmessageid[:11] # tmpgtv4qjix
-                    messageid = partialpathmessageid[11:] # 1111269940599738479
-                    # add /tmp/ to partialpath, save as new variable
-                    fullpath = "/tmp/" + partialpath # should be /tmp/tmpgtv4qjix
-                    await reaction.message.channel.send(f"fullpath extracted, {fullpath}")        
-                    emoji = reaction.emoji
-                    
-                    if emoji == "1️⃣":
-                        index = 0
-                    elif emoji == "2️⃣":
-                        index = 1
-                    elif emoji == "3️⃣":
-                        index = 2
-                    elif emoji == "4️⃣":
-                        index = 3       
-                    await reaction.message.channel.send(f"index extracted, {index}")         
-                    index = index
-                    stage_1_result_path = fullpath
-                    thread = reaction.message.channel
-                    dfif_command_message_id = messageid
-                    await reaction.message.channel.send(f"calling dfif2")  
-                    await dfif2(index, stage_1_result_path, thread, dfif_command_message_id)
+            if reaction.message.channel.id == 1100458786826747945:
+                # 811235357663297546 =  lunarflu
+                if reaction.message.attachments:
+                    if user.id == 811235357663297546:  #  if user.id == reaction.message.mentions[0].id:           
+                        # magic begins
+                        await reaction.message.channel.send("reaction detected")
+                        attachment = reaction.message.attachments[0]
+                        image_name = attachment.filename # named something like: tmpgtv4qjix1111269940599738479.png
+                        # remove .png first
+                        partialpathmessageid = image_name[:-4] # should be tmpgtv4qjix1111269940599738479 
+                        # extract partialpath, messageid
+                        partialpath = partialpathmessageid[:11] # tmpgtv4qjix
+                        messageid = partialpathmessageid[11:] # 1111269940599738479
+                        # add /tmp/ to partialpath, save as new variable
+                        fullpath = "/tmp/" + partialpath # should be /tmp/tmpgtv4qjix
+                        await reaction.message.channel.send(f"fullpath extracted, {fullpath}")        
+                        emoji = reaction.emoji
+                        
+                        if emoji == "1️⃣":
+                            index = 0
+                        elif emoji == "2️⃣":
+                            index = 1
+                        elif emoji == "3️⃣":
+                            index = 2
+                        elif emoji == "4️⃣":
+                            index = 3       
+                        await reaction.message.channel.send(f"index extracted, {index}")         
+                        index = index
+                        stage_1_result_path = fullpath
+                        thread = reaction.message.channel
+                        dfif_command_message_id = messageid
+                        await reaction.message.channel.send(f"calling dfif2")  
+                        await dfif2(index, stage_1_result_path, thread, dfif_command_message_id)
         
         '''
 
@@ -414,7 +423,10 @@ async def on_reaction_add(reaction, user):
         
         #await ctx.reply('An error occured in stage 2') need to fix
         #await ctx.message.add_reaction('❌')
-#----------------------------------------------------------------------------------------------------------------------------        
+#---------------------------------------------------------------------------------------------------------------------------- 
+async def safetychecks(ctx):
+
+    
 def run_bot():
     bot.run(DISCORD_TOKEN)