[decorators] big test, custom path, individual image reacts

app.py

@@ -19,6 +19,9 @@ from pytz import timezone
 #
 import asyncio
 
+import shutil # for doing image movement magic
+
+
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
 df = Client("huggingface-projects/IF", DFIF_TOKEN)
@@ -44,8 +47,9 @@ async def on_ready():
 async def deepfloydif(ctx, *, prompt: str):
     try:
         try:
-            def check_reaction(reaction, user):
-                return user == ctx.author and str(reaction.emoji) in ['1️⃣', '2️⃣', '3️⃣', '4️⃣']   
+            channelcheck = bot.get_channel(channel_id)
+            if channelcheck != 1100458786826747945: # bot-test
+                return  
     
             await ctx.message.add_reaction('👍')
             thread = await ctx.message.create_thread(name=f'{ctx.author} Image Upscaling Thread ')
@@ -107,7 +111,7 @@ async def deepfloydif(ctx, *, prompt: str):
 
 #----------------------------------------------------------------------------------------------------------------------------
 # Stage 2
-async def dfif2(ctx, index: int, stage_1_result_path, thread):
+async def dfif2(index: int, stage_1_result_path, thread):
     try:
         selected_index_for_stage_2 = index
         seed_2 = 0
@@ -119,9 +123,9 @@ async def dfif2(ctx, index: int, stage_1_result_path, thread):
     
         
         with open(result_path, 'rb') as f:
-            await thread.send(f'{ctx.author.mention}Here is the upscaled image! :) ', file=discord.File(f, 'result.png'))
+            await thread.send(f'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('✔️')
+            #await ctx.message.add_reaction('✅') need to fix this
             
 
         
@@ -131,7 +135,33 @@ async def dfif2(ctx, index: int, stage_1_result_path, thread):
         await ctx.reply('An error occured in stage 2')
         await ctx.message.add_reaction('❌')
 #----------------------------------------------------------------------------------------------------------------------------
+@bot.event
+async def on_reaction_add(reaction, user):
+    # safety checks first
+    if user.bot:
+        return    
+        
+    # do they have the required (verified) role?
+    guild = reaction.message.guild
+    member = guild.get_member(user.id)
 
+    # Check if the user has the desired role
+    required_role_id = 897376942817419265  # huggingfolks for now
+    required_role = guild.get_role(required_role_id)
+
+    if required_role not in member.roles:
+        return    
+
+    attachment = reaction.message.attachments[0]
+    emoji = reaction.emoji
+    
+    if reaction.emoji.name == '👍':
+        if message.attachments:
+            if user.id == message.mentions[0].id:  # all we care about is upscaling whatever image this is 
+                stage_1_result_path = "/tmp/s1rp"
+                attachment_path = os.path.join(stage_1_result_path, attachment.filename)
+                await attachment.save(attachment_path)
+                await dfif2(0, attachment_path, thread=reaction.message.reference.resolved)