fix

app.py

@@ -101,17 +101,21 @@ 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)
+            ctx.send("You chose the first image!")
+            index = 0
+            await dfif2(ctx, index, 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)            
+            ctx.send("You chose the second image!")
+            index = 1
+            await dfif2(ctx, index, 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)            
+            ctx.send("You chose the third image!")
+            index = 2
+            await dfif2(ctx, index, stage_1_result_path)            
         elif str(reaction.emoji) == '4️⃣':
-            await ctx.send("You chose the fourth image!")  
-            await dfif2(ctx, 3, stage_1_result_path)            
+            ctx.send("You chose the fourth image!")
+            index = 3
+            await dfif2(ctx, index, stage_1_result_path)            
 
     #deepfloydif try/except
     except Exception as e:
@@ -120,7 +124,7 @@ async def deepfloydif(ctx, *, prompt: str):
 
 #----------------------------------------------------------------------------------------------------------------------------
 # Stage 2
-async def dfif2(ctx, index, stage_1_result_path):
+async def dfif2(ctx, index: int, stage_1_result_path):
     try:
         selected_index_for_stage_2 = index
         seed_2 = 0