reaction emoji -> await dfif2(ctx, index, stage_1_result_path)

app.py

@@ -41,8 +41,9 @@ bot = commands.Bot(command_prefix='!', intents=intents)
 async def on_ready():
     print('Logged on as', bot.user)
     bot.log_channel = bot.get_channel(1100458786826747945) # 1100458786826747945 = bot-test, 1107006391547342910 = lunarbot server
-        
 
+#---------------------------------------------------------------------------------------------------------------------------------------------- 
+# Stage 1
 @bot.command()
 async def deepfloydif(ctx, *, prompt: str):
     try:
@@ -101,32 +102,34 @@ async def deepfloydif(ctx, *, prompt: str):
         reaction, user = await bot.wait_for('reaction_add', check=check_reaction)
         if str(reaction.emoji) == '1️⃣':
             await ctx.send("You chose the first image!")
+            await dfif2(ctx, 0, stage_1_result_path)
         elif str(reaction.emoji) == '2️⃣':
             await ctx.send("You chose the second image!")
+            await dfif2(ctx, 1, stage_1_result_path)            
         elif str(reaction.emoji) == '3️⃣':
             await ctx.send("You chose the third image!")
+            await dfif2(ctx, 2, stage_1_result_path)            
         elif str(reaction.emoji) == '4️⃣':
-            await ctx.send("You chose the fourth image!")        
+            await ctx.send("You chose the fourth image!")  
+            await dfif2(ctx, 3, stage_1_result_path)            
 
+    #deepfloydif try/except
     except Exception as e:
         print(f"Error: {e}")
         await ctx.reply('An error occurred while processing your request. Please wait 5 seconds before retrying.')
 
-#new stage 2----------------------------------------------------------------------------------------------------------------------------------------------
+#----------------------------------------------------------------------------------------------------------------------------
 # Stage 2
-@bot.command()
-async def dfif2(ctx, index: int, stage_1_result_path, image_paths):
+async def dfif2(ctx, index, stage_1_result_path):
     try:
-        image_path = image_paths[index]
-            
-        selected_index_for_stage_2 = image_path
+        selected_index_for_stage_2 = index
         seed_2 = 0
         guidance_scale_2 = 4
         custom_timesteps_2 = 'smart50'
         number_of_inference_steps_2 = 50
-        result_path = df.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
-        
-        # Process the result_path or perform any additional operations
+        result_path = df.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, 
+                                 guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
+    
         
         with open(result_path, 'rb') as f:
             await ctx.reply('Here is the result of the second stage', file=discord.File(f, 'result.png'))
@@ -134,7 +137,7 @@ async def dfif2(ctx, index: int, stage_1_result_path, image_paths):
     except Exception as e:
         print(f"Error: {e}")
         await ctx.reply('An error occurred while processing stage 2 upscaling. Please try again later.')
-
+#----------------------------------------------------------------------------------------------------------------------------