[deepfloydif.py] added debugging

deepfloydif.py

@@ -60,6 +60,8 @@ async def deepfloydif_stage_1(interaction, prompt, client):
         global DEEPFLOYDIF_CHANNEL_ID
         if interaction.user.id != BOT_USER_ID:
             if interaction.channel.id == DEEPFLOYDIF_CHANNEL_ID:
+                if os.environ.get('TEST_ENV') == 'True':
+                    print("Safetychecks passed for deepfloydif_stage_1")
                 await interaction.response.send_message("Working on it!")
                 channel = interaction.channel
                 # interaction.response message can't be used to create a thread, so we create another message
@@ -82,10 +84,14 @@ async def deepfloydif_stage_1(interaction, prompt, client):
                 custom_timesteps_1 = 'smart50'
                 number_of_inference_steps = 50
                 
+                if os.environ.get('TEST_ENV') == 'True':
+                    print("Running deepfloydif_client.submit")                
                 job = deepfloydif_client.submit(prompt, negative_prompt, seed, number_of_images, guidance_scale, custom_timesteps_1, number_of_inference_steps, api_name='/generate64')  # This is not blocking, similar to run_in_executor (but better)
                 while job.done() is False: 
                     pass
                 else:
+                    if os.environ.get('TEST_ENV') == 'True':
+                        print("Safetychecks passed for deepfloydif_stage_1")                    
                     file_paths = job.outputs()
                     file_paths = file_paths[0]
                     stage_1_results = file_paths[0]
@@ -96,6 +102,8 @@ async def deepfloydif_stage_1(interaction, prompt, client):
     
                     if png_files:
                         # take all 4 images and combine them into one large 2x2 image (similar to Midjourney)
+                        if os.environ.get('TEST_ENV') == 'True':
+                            print("Combining images for deepfloydif_stage_1")
                         png_file_index = 0
                         images = load_image(png_files, stage_1_results, png_file_index) 
                         combined_image = Image.new('RGB', (images[0].width * 2, images[0].height * 2))
@@ -105,6 +113,8 @@ async def deepfloydif_stage_1(interaction, prompt, client):
                         combined_image.paste(images[3], (images[0].width, images[0].height))
                         combined_image_path = os.path.join(stage_1_results, f'{partial_path}{dfif_command_message_id}.png')
                         combined_image.save(combined_image_path)
+                        if os.environ.get('TEST_ENV') == 'True':
+                            print("Images combined for deepfloydif_stage_1")                        
     
                         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'{partial_path}{dfif_command_message_id}.png'))                
@@ -141,6 +151,8 @@ async def react_1234(reaction_emojis, combined_image_dfif):
 async def deepfloydif_stage_2(index: int, stage_1_result_path, thread, dfif_command_message_id): 
     """upscaling function for images generated using /deepfloydif"""
     try:
+        if os.environ.get('TEST_ENV') == 'True':
+            print("Running deepfloydif_stage_2")        
         parent_channel = thread.parent
         dfif_command_message = await parent_channel.fetch_message(dfif_command_message_id)
         if index == 0:
@@ -167,6 +179,8 @@ async def deepfloydif_stage_2(index: int, stage_1_result_path, thread, dfif_comm
 async def deepfloydif_stage_2_react_check(reaction, user): 
     """Checks for a reaction in order to call dfif2"""
     try:
+        if os.environ.get('TEST_ENV') == 'True':
+            print("Running deepfloydif_stage_2_react_check")        
         global BOT_USER_ID
         global DEEPFLOYDIF_CHANNEL_ID
         if user.id != BOT_USER_ID: