reorganized functions, added combined_image_path to the thread

app.py

@@ -92,41 +92,14 @@ 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)
-
-
-
-
-
-        await create_thread_dfif(ctx)
+            # creates a thread, passing the combined_image_path in
+            await create_thread_dfif(ctx, combined_image_path)
             
 
         #with open(combined_image_path, 'rb') as f:
         #    sent_message = await ctx.reply('Here is the combined image. 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 sent_message.add_reaction(emoji)            
-
-        reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
-        if str(reaction.emoji) == '1️⃣':
-            await ctx.reply("Upscaling the first image...")
-            index = 0
-            await dfif2(ctx, index, stage_1_result_path)
-        elif str(reaction.emoji) == '2️⃣':
-            await ctx.reply("Upscaling the second image...")
-            index = 1
-            await dfif2(ctx, index, stage_1_result_path)            
-        elif str(reaction.emoji) == '3️⃣':
-            await ctx.reply("Upscaling the third image...")
-            index = 2
-            await dfif2(ctx, index, stage_1_result_path)            
-        elif str(reaction.emoji) == '4️⃣':
-            await ctx.reply("Upscaling the fourth image...")
-            index = 3
-            await dfif2(ctx, index, stage_1_result_path)            
+          
 
     #deepfloydif try/except
     except Exception as e:
@@ -153,10 +126,33 @@ async def dfif2(ctx, index: int, stage_1_result_path):
         print(f"Error: {e}")
         await ctx.reply('An error occurred while processing stage 2 upscaling. Please try again later.')
 #----------------------------------------------------------------------------------------------------------------------------
-async def create_thread_dfif(ctx):
+async def create_thread_dfif(ctx, combined_image_path):
     with open(combined_image_path, 'rb') as f:
         thread_channel = await ctx.channel.create_thread(name='Image Thread')
-        sent_message = await ctx.reply('Here is the combined image. React with the image number you want to upscale!', file=discord.File(f, 'combined_image.png'))        
+        sent_message = await ctx.reply('Here is the combined image. 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 sent_message.add_reaction(emoji)            
+
+        reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
+        if str(reaction.emoji) == '1️⃣':
+            await ctx.reply("Upscaling the first image...")
+            index = 0
+            await dfif2(ctx, index, stage_1_result_path)
+        elif str(reaction.emoji) == '2️⃣':
+            await ctx.reply("Upscaling the second image...")
+            index = 1
+            await dfif2(ctx, index, stage_1_result_path)            
+        elif str(reaction.emoji) == '3️⃣':
+            await ctx.reply("Upscaling the third image...")
+            index = 2
+            await dfif2(ctx, index, stage_1_result_path)            
+        elif str(reaction.emoji) == '4️⃣':
+            await ctx.reply("Upscaling the fourth image...")
+            index = 3
+            await dfif2(ctx, index, stage_1_result_path)  
 #----------------------------------------------------------------------------------------------------------------------------