Update app.py

app.py

@@ -214,56 +214,69 @@ async def dfif(interaction: discord.Interaction, prompt: str):
                 await thread.send(f'{interaction.user.mention}Generating images in thread, can take ~1 minute...')
 
                 # generation
+              
+                '''
                 job = df.submit(prompt, negative_prompt, seed, number_of_images, guidance_scale, 
                                 custom_timesteps_1, number_of_inference_steps, api_name='/generate64')
                 while not job.done():
                     time.sleep(0.1)
-                else:
-                    file_paths = job.outputs()
+                else:                
+                '''
+                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:]
+                png_files = list(glob.glob(f"{stage_1_results}/**/*.png"))
+
+                '''
+                     file_paths = job.outputs()
                     file_paths = file_paths[0] # tmp12345678.json
                     stage_1_results = file_paths[0]
                     stage_1_result_path = file_paths[2]
                     partialpath = stage_1_result_path[5:]
                     
-                    png_files = list(glob.glob(f"{stage_1_results}/**/*.png"))
+                    png_files = list(glob.glob(f"{stage_1_results}/**/*.png"))               
+                '''
 
-                    if png_files:
-                        first_png = png_files[0]
-                        second_png = png_files[1]
-                        third_png = png_files[2]
-                        fourth_png = png_files[3]
-                    
-                        first_png_path = os.path.join(stage_1_results, first_png)
-                        second_png_path = os.path.join(stage_1_results, second_png)
-                        third_png_path = os.path.join(stage_1_results, third_png)
-                        fourth_png_path = os.path.join(stage_1_results, fourth_png)
-                    
-                        img1 = Image.open(first_png_path)
-                        img2 = Image.open(second_png_path)
-                        img3 = Image.open(third_png_path)
-                        img4 = Image.open(fourth_png_path)
-                    
-                        combined_image = Image.new('RGB', (img1.width * 2, img1.height * 2))
-                    
-                        combined_image.paste(img1, (0, 0))
-                        combined_image.paste(img2, (img1.width, 0))
-                        combined_image.paste(img3, (0, img1.height))
-                        combined_image.paste(img4, (img1.width, img1.height))
-                    
-                        combined_image_path = os.path.join(stage_1_results, f'{partialpath}{dfif_command_message_id}.png')
-                        combined_image.save(combined_image_path)
 
-                        with open(combined_image_path, 'rb') as f:
-                            combined_image_dfif = await thread.send(f'{interaction.user.mention}React with the image number you want to upscale!', file=discord.File(
-                                f, f'{partialpath}{dfif_command_message_id}.png')) # named something like: tmpgtv4qjix1111269940599738479.png                
+                if png_files:
+                    first_png = png_files[0]
+                    second_png = png_files[1]
+                    third_png = png_files[2]
+                    fourth_png = png_files[3]
+                
+                    first_png_path = os.path.join(stage_1_results, first_png)
+                    second_png_path = os.path.join(stage_1_results, second_png)
+                    third_png_path = os.path.join(stage_1_results, third_png)
+                    fourth_png_path = os.path.join(stage_1_results, fourth_png)
+                
+                    img1 = Image.open(first_png_path)
+                    img2 = Image.open(second_png_path)
+                    img3 = Image.open(third_png_path)
+                    img4 = Image.open(fourth_png_path)
+                
+                    combined_image = Image.new('RGB', (img1.width * 2, img1.height * 2))
+                
+                    combined_image.paste(img1, (0, 0))
+                    combined_image.paste(img2, (img1.width, 0))
+                    combined_image.paste(img3, (0, img1.height))
+                    combined_image.paste(img4, (img1.width, img1.height))
+                
+                    combined_image_path = os.path.join(stage_1_results, f'{partialpath}{dfif_command_message_id}.png')
+                    combined_image.save(combined_image_path)
+
+                    with open(combined_image_path, 'rb') as f:
+                        combined_image_dfif = await thread.send(f'{interaction.user.mention}React with the image number you want to upscale!', file=discord.File(
+                            f, f'{partialpath}{dfif_command_message_id}.png')) # named something like: tmpgtv4qjix1111269940599738479.png                
 
 
-                        emoji_list = ['↖️', '↗️', '↙️', '↘️']
-                        await react1234(emoji_list, combined_image_dfif)
-                            
+                    emoji_list = ['↖️', '↗️', '↙️', '↘️']
+                    await react1234(emoji_list, combined_image_dfif)
                         
-                        await message.remove_reaction('<a:loading:1121820108189339738>', client.user)
-                        await message.add_reaction('✅')
+                    
+                    await message.remove_reaction('<a:loading:1121820108189339738>', client.user)
+                    await message.add_reaction('✅')