testing some magic

app.py

@@ -61,6 +61,8 @@ async def deepfloydif(ctx, *, prompt: str):
             guidance_scale = 7
             custom_timesteps_1 = 'smart50'
             number_of_inference_steps = 50
+
+            await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
             
         except Exception as e:
             print(f"Error: {e}")
@@ -71,13 +73,15 @@ async def deepfloydif(ctx, *, prompt: str):
         try:
             stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
                 prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
+            
+            partialpath = stage_1_result_path[5:] #magic for later
         except Exception as e:
             print(f"Error: {e}")
             await ctx.reply('stage 1 error -> during generation')
             await ctx.message.add_reaction('❌')            
             
         try:
-            await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
+
 
             png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
             
@@ -85,16 +89,16 @@ async def deepfloydif(ctx, *, prompt: str):
                 for i, png_file in enumerate(png_files):
                     png_file_path = os.path.join(stage_1_results, png_file)
                     img = Image.open(png_file_path)
-                    image_path = os.path.join(stage_1_results, f'image{i+1}.png')
+                    image_path = os.path.join(stage_1_results, f'{i+1}{partialpath}.png')
                     img.save(image_path)
                     with open(image_path, 'rb') as f:
-                        await thread.send(f'{ctx.author.mention}Image {i+1}', file=discord.File(f, f'image{i+1}.png'))
-                    await asyncio.sleep(1)  # Add a delay between posting each image
+                        await thread.send(f'{ctx.author.mention}Image {i+1}', file=discord.File(f, f'{i+1}{partialpath}.png'))
+                    await asyncio.sleep(1) 
     
             await thread.send(f'{ctx.author.mention}React with 👍 to the image you want to upscale!')             
         except Exception as e:
             print(f"Error: {e}")
-            await ctx.reply('stage 1 error -> threads')
+            await ctx.reply('stage 1 error -> posting images in thread')
             await ctx.message.add_reaction('❌')
             
         
@@ -107,7 +111,7 @@ async def deepfloydif(ctx, *, prompt: str):
 
 #----------------------------------------------------------------------------------------------------------------------------
 # Stage 2
-async def dfif2(index: int, stage_1_result_path, message):
+async def dfif2(index: int, stage_1_result_path, thread):
     try:
         selected_index_for_stage_2 = index
         seed_2 = 0
@@ -119,7 +123,7 @@ async def dfif2(index: int, stage_1_result_path, message):
     
         
         with open(result_path, 'rb') as f:
-            await message.reply(f'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('✅') need to fix this
             
@@ -148,17 +152,32 @@ async def on_reaction_add(reaction, user):
     if required_role not in user.roles:
         return    
 
-    attachment = reaction.message.attachments[0]
     emoji = reaction.emoji
+
+    thread = reaction.message.channel
+    tbd = reaction.message.content
+    #index = 0 decided by index in front of partialpath 
+
+
+    # magic begins
+    attachment = reaction.message.attachments[0]
+    image_name = attachment.filename
+    # we know image_name will be something like 1tmpgtv4qjix.png
+    # remove .png first
+    indexpartialpath = image_name[:-4] # should be 1tmpgtv4qjix 
+    # extract index as an integer (dfif2 needs integer)
+    index = int(indexpartialpath[0]) # should be 1
+    # extract partialpath
+    partialpath = indexpartialpath[1:] # should be tmpgtv4qjix
+    # add /tmp/ to partialpath, save as new variable
+    fullpath = "/tmp/" + partialpath # should be /tmp/tmpgtv4qjix
     
     if emoji == '👍':
         if reaction.message.attachments:
             if user.id == reaction.message.mentions[0].id:  # all we care about is upscaling whatever image this is 
-                stage_1_result_path = "/tmp/s1rp"
-                os.makedirs(stage_1_result_path, exist_ok=True)
-                attachment_path = os.path.join(stage_1_result_path, attachment.filename)
-                await attachment.save(attachment_path)
-                await dfif2(0, attachment_path, message=reaction.message)
+                stage_1_result_path = fullpath
+                index = index
+                await dfif2(index, stage_1_result_path, thread)