Update deepfloydif.py

deepfloydif.py

@@ -84,7 +84,7 @@ def combine_images(png_files, stage_1_images, partial_path):
     combined_image.paste(images[3], (images[0].width, images[0].height))
     combined_image_path = os.path.join(stage_1_images, f"{partial_path}.png")
     combined_image.save(combined_image_path)
-
+    return combined_image_path
 
 async def deepfloydif_stage_1(interaction, prompt, client):
     """DeepfloydIF command (generate images with realistic text using slash commands)"""
@@ -113,7 +113,7 @@ async def deepfloydif_stage_1(interaction, prompt, client):
                 png_files = list(glob.glob(f"{stage_1_images}/**/*.png"))
 
                 if png_files:
-                    combine_images(png_files, stage_1_images, partial_path)
+                    combined_image_path = combine_images(png_files, stage_1_images, partial_path)
                     if os.environ.get("TEST_ENV") == "True":
                         print("Images combined for deepfloydif_stage_1")
                     with open(combined_image_path, "rb") as f: