[dfif] def inference(prompt):

app.py

@@ -89,7 +89,7 @@ async def safetychecks(ctx):
         print(f"Error: safetychecks failed somewhere, command will not continue.")
         await ctx.message.reply(f"<@811235357663297546> SC failed somewhere") # this will always ping, as long as the bot has access to the channel
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
-# jojo
+# jojo ✅
 @bot.command()
 async def jojo(ctx):
     # img + face    ✅
@@ -122,7 +122,7 @@ async def jojo(ctx):
         await ctx.message.add_reaction('❌')            
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
-# Disney
+# Disney ❌
 @bot.command()
 async def disney(ctx):
     try:
@@ -145,7 +145,7 @@ async def disney(ctx):
         await ctx.message.add_reaction('❌') 
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
-# Spider-Verse
+# Spider-Verse ✅
 @bot.command()
 async def spiderverse(ctx):
     try:
@@ -168,7 +168,7 @@ async def spiderverse(ctx):
         await ctx.message.add_reaction('❌') 
         
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
-# sketch
+# sketch ✅
 @bot.command()
 async def sketch(ctx):
     try:    
@@ -189,8 +189,23 @@ async def sketch(ctx):
         print(f"Error: {e}")
         await thread.send(f"{ctx.author.mention}Error: {e}")
         await ctx.message.add_reaction('❌') 
-#----------------------------------------------------------------------------------------------------------------------------------------------     
-# Stage 1
+#----------------------------------------------------------------------------------------------------------------------------------------------
+# deepfloydif stage 1 generation ❌
+def inference(prompt):
+  negative_prompt = ''
+  seed = random.randint(0, 1000)
+  #seed = 1
+  number_of_images = 4
+  guidance_scale = 7
+  custom_timesteps_1 = 'smart50'
+  number_of_inference_steps = 50
+
+  stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
+      prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
+
+  return [stage_1_results, stage_1_param_path, stage_1_result_path]
+#----------------------------------------------------------------------------------------------------------------------------------------------    
+# Stage 1 ❌
 @bot.command()
 async def deepfloydif(ctx, *, prompt: str):
     try:
@@ -219,21 +234,25 @@ async def deepfloydif(ctx, *, prompt: str):
             print(f"Error: {e}")
             await ctx.reply('stage 1 error -> pre generation')
             await ctx.message.add_reaction('❌')
-
+            
+        #generation❌-------------------------------------------------------
         try:
             #stage_1_results, stage_1_param_path, stage_1_result_path = df.predict(
             #    prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
 
-            stage_1_results, stage_1_param_path, stage_1_result_path = await asyncio.get_running_loop().run_in_executor(
-                None, df.predict, prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps,
-                api_name)
+            # run blocking function in executor
+            loop = asyncio.get_running_loop()
+            result = await loop.run_in_executor(None, inference, prompt)
+            stage_1_results = result[0]
+            stage_1_result_path = result[2]
             
             partialpath = stage_1_result_path[5:] #magic for later
+            
         except Exception as e:
             print(f"Error: {e}")
             await ctx.reply('stage 1 error -> during generation')
             await ctx.message.add_reaction('❌')            
-            
+        #posting images----------------------------------------------------------------    
         try:
             png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
 
@@ -298,7 +317,7 @@ async def deepfloydif(ctx, *, prompt: str):
         await ctx.message.add_reaction('❌')
 
 #----------------------------------------------------------------------------------------------------------------------------
-# Stage 2
+# Stage 2 ❌
 async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
     try:
         await thread.send(f"inside dfif2, upscaling")  
@@ -342,33 +361,11 @@ async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id
         #await ctx.reply('An error occured in stage 2') need to fix
         #await ctx.message.add_reaction('❌')
 #----------------------------------------------------------------------------------------------------------------------------
+# react detector for stage 2 ❌
 @bot.event
 async def on_reaction_add(reaction, user):    # ctx = await bot.get_context(reaction.message)? could try later, might simplify
     try:
         # safety checks first ❌
-        '''
-        if user.bot:
-            return   
-
-        #offline bot check ❌
-        offline_bot_role_id = 1103676632667017266
-        bot_member = reaction.message.guild.get_member(bot.user.id)
-        if any(role.id == offline_bot_role_id for role in bot_member.roles):            
-            return
-            
-        # verified role check ❌
-        guild = reaction.message.guild            
-        required_role_id = 900063512829755413  # @verified for now
-        required_role = guild.get_role(required_role_id)
-        if required_role not in user.roles:
-            return    
-
-        #channel check ❌
-        if reaction.message.channel.id != 1100458786826747945: 
-            return        
-        
-        '''
-
         thread = reaction.message.channel
         threadparentid = thread.parent.id