[decorators] individual images test

app.py

@@ -60,7 +60,7 @@ async def deepfloydif(ctx, *, prompt: str):
         seed = random.randint(0, 2**32 - 1)
         stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(prompt, "blur", seed, number_of_images, 7.0, 'smart100', 50, api_name="/generate64")
         png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
-
+        '''
         if png_files:
             first_png = png_files[0]
             second_png = png_files[1]
@@ -87,35 +87,19 @@ async def deepfloydif(ctx, *, prompt: str):
             combined_image_path = os.path.join(stage_1_results, 'combined_image.png')
             combined_image.save(combined_image_path)
 
-            # Trigger the second stage prediction
-            #await dfif2(ctx, stage_1_result_path)
-
-
-        with open(combined_image_path, 'rb') as f:
-            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.
-        for emoji in ['1️⃣', '2️⃣', '3️⃣', '4️⃣']:
-            await threadmsg.add_reaction(emoji)   
-
-        reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
-        if str(reaction.emoji) == '1️⃣':
-            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"{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"{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"{ctx.author.mention}Upscaling the fourth image...")
-            index = 3
-            await dfif2(ctx, index, stage_1_result_path, thread)            
+            '''
+        
+        if png_files:
+            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')
+                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}React with 👍 to the image you want to upscale!')         
 
     #deepfloydif try/except
     except Exception as e: