tweak

comfy-nodes/comfy_std_output_image.py

@@ -9,7 +9,7 @@ import folder_paths
 class ComfyDeployStdOutputImage:
     def __init__(self):
         self.output_dir = folder_paths.get_output_directory()
-        self.type = "std_output"
+        self.type = "output"
         self.prefix_append = ""
         self.compress_level = 4
 
@@ -36,7 +36,7 @@ class ComfyDeployStdOutputImage:
 
     OUTPUT_NODE = True
 
-    CATEGORY = "std_output"
+    CATEGORY = "output"
     DESCRIPTION = "Saves the input images to your ComfyUI output directory."
 
     def run(

custom_routes.py

@@ -1229,6 +1229,7 @@ async def send_json_override(self, event, data, sid=None):
             # if class_type == "PreviewImage":
             #     logger.info("Skipping preview image")
             # else:
+            if class_type == "ComfyDeployStdOutputImage":
                 await update_run_with_output(
                     prompt_id,
                     data.get("output"),
@@ -1765,10 +1766,7 @@ async def update_run_with_output(
     have_upload_media = False
     if data is not None:
         have_upload_media = (
-            "images" in data
-            and isinstance(data.get("images"), list)
-            and len(data["images"]) > 0
-            and data["images"][0].get("type") == "std_output"
+            "images" in data or "files" in data or "gifs" in data or "mesh" in data
         )
     if bypass_upload and have_upload_media:
         print(