revert to stable [1.0

app.py

@@ -53,11 +53,10 @@ async def deepfloydif(ctx, *, prompt: str):
         prompt = re.sub(r'[^\w\s]', '', prompt)  # Remove special characters
 
         def check_reaction(reaction, user):
-            return user == ctx.author and str(reaction.emoji) in ['1️⃣', '2️⃣', '3️⃣', '4️⃣']   # is this used?
+            return user == ctx.author and str(reaction.emoji) in ['1️⃣', '2️⃣', '3️⃣', '4️⃣']   
 
         await ctx.message.add_reaction('👍')
-
-        #generate images----------------------------------------------------
+        
         number_of_images = 4
         current_time = int(time.time())
         random.seed(current_time)
@@ -65,7 +64,6 @@ async def deepfloydif(ctx, *, prompt: str):
         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')]
 
-        #combine images into one image--------------------------------------
         if png_files:
             first_png = png_files[0]
             second_png = png_files[1]
@@ -91,13 +89,13 @@ async def deepfloydif(ctx, *, prompt: str):
 
             combined_image_path = os.path.join(stage_1_results, 'combined_image.png')
             combined_image.save(combined_image_path)
-            
-        # open combined image in a thread, mention the user
+
+            # Trigger the second stage prediction
+            #await dfif2(ctx, stage_1_result_path)
+
         with open(combined_image_path, 'rb') as f:
-            thread_channel = await ctx.channel.create_thread(name='Image Thread')
-            sent_message = await thread_channel.send('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️⃣']:
@@ -119,8 +117,7 @@ async def deepfloydif(ctx, *, prompt: str):
         elif str(reaction.emoji) == '4️⃣':
             await ctx.reply("Upscaling the fourth image...")
             index = 3
-            await dfif2(ctx, index, stage_1_result_path)  
-          
+            await dfif2(ctx, index, stage_1_result_path)            
 
     #deepfloydif try/except
     except Exception as e:
@@ -149,12 +146,7 @@ async def dfif2(ctx, index: int, stage_1_result_path):
 #----------------------------------------------------------------------------------------------------------------------------
 
 
-
-
-
-
-
-
+        
 
 def run_bot():
     bot.run(DISCORD_TOKEN)