From 46e7f6eb25c5101f652cb64079f0dc5abb29cc10 Mon Sep 17 00:00:00 2001
From: Maxed-Out-99 <maxedout99999@gmail.com>
Date: Mon, 15 Dec 2025 15:16:04 -0800
Subject: [PATCH] Add Save Image MXD node for saving and previewing images

Introduces the SaveImage_MXD node, allowing users to either save images to disk, preview them, or both, via a single node. Updates NODE_CLASS_MAPPINGS and NODE_DISPLAY_NAME_MAPPINGS to register the new node.
---
 maxedoutnodes.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/maxedoutnodes.py b/maxedoutnodes.py
index fa5931d..2521dec 100644
--- a/maxedoutnodes.py
+++ b/maxedoutnodes.py
@@ -1029,6 +1029,49 @@ class LoadImageWithPromptsMXD:
     
 ########################################################################################################################
 
+from nodes import PreviewImage, SaveImage
+class SaveImage_MXD:
+    TITLE = "Save Image MXD"
+    CATEGORY = "MXD/Image"
+    OUTPUT_NODE = True
+    FUNCTION = "save"
+
+    DESCRIPTION = """
+    - Save images to your ComfyUI output folder, or preview without saving.
+
+    - Use this when you want one simple node instead of choosing
+      between PreviewImage and SaveImage.
+    """
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": ("IMAGE", {"tooltip": "Images to preview and/or save."}),
+                "filename_prefix": ("STRING", {
+                    "default": "ComfyUI",
+                    "tooltip": "File name prefix. Tip: you can use a subfolder like 'tests/my_run'."
+                }),
+                "mode": ([
+                    "Save + Preview",
+                    "Preview only"
+                ], {
+                    "default": "Save + Preview",
+                    "tooltip": "Choose whether to write files to disk or only show the preview."
+                }),
+            },
+            "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
+        }
+
+    RETURN_TYPES = ()
+    OUTPUT_TOOLTIPS = ("Saves and/or previews the images.",)
+
+    def save(self, images, filename_prefix, mode, prompt=None, extra_pnginfo=None):
+        if mode.startswith("Preview"):
+            return PreviewImage().save_images(images, filename_prefix, prompt, extra_pnginfo)
+        return SaveImage().save_images(images, filename_prefix, prompt, extra_pnginfo)
+
+
 
 # NODE MAPPING
 NODE_CLASS_MAPPINGS = {
@@ -1046,6 +1089,7 @@ NODE_CLASS_MAPPINGS = {
     "Load Image Batch MXD": LoadImageBatchMXD,
     "LoadImageWithPromptsMXD": LoadImageWithPromptsMXD,
     "ZImageTurboEmptyLatentImage": ZImageTurboEmptyLatentImage,
+    "Save Image MXD": SaveImage_MXD,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -1063,4 +1107,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Load Image Batch MXD": "Load Image Batch MXD",
     "LoadImageWithPromptsMXD": "Load Image MXD",
     "ZImageTurboEmptyLatentImage": "ZImageTurbo Empty Latent Image MXD",
+    "Save Image MXD": "Save Image MXD",
 }