From 07701db0bb5ae38100c3ae95b5af89d42e17b718 Mon Sep 17 00:00:00 2001
From: Thomas Ward <teward@thomas-ward.net>
Date: Fri, 24 May 2024 03:00:39 -0400
Subject: [PATCH] Add image.py objects to available mappings

---
 _mappings.py          | 8 ++++++++
 saving.py => image.py | 0
 2 files changed, 8 insertions(+)
 rename saving.py => image.py (100%)

diff --git a/_mappings.py b/_mappings.py
index fd76c0d..df9a5c6 100644
--- a/_mappings.py
+++ b/_mappings.py
@@ -1,6 +1,8 @@
 from .generation import TWCUI_Util_GenerationParameters, TWCUI_Util_CommonSDXLResolutions
 from .generation import TWCUI_Util_GenerationPrompts
 
+from .image import TWCUI_Util_SaveImage
+
 from .util import TWCUI_Util_StringLiteral, TWCUI_Util_MultilineStringLiteral
 from .util import TWCUI_Util_IntLiteral, TWCUI_Util_FloatLiteral
 
@@ -10,6 +12,9 @@ NODE_CLASS_MAPPINGS = {
     "TWCUI_Util_GenerationParameters": TWCUI_Util_GenerationParameters,
     "TWCUI_Util_GenerationPrompts": TWCUI_Util_GenerationPrompts,
 
+    # Image category
+    "TWCUI_Util_SaveImage": TWCUI_Util_SaveImage,
+
     # Util category
     "TWCUI_Util_FloatLiteral": TWCUI_Util_FloatLiteral,
     "TWCUI_Util_IntLiteral": TWCUI_Util_IntLiteral,
@@ -23,6 +28,9 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "TWCUI_Util_GenerationParameters": "Generation Parameters",
     "TWCUI_Util_GenerationPrompts": "Prompts",
 
+    # Image category
+    "TWCUI_Util_SaveImage": "Save Image",
+
     # Util category
     "TWCUI_Util_FloatLiteral": "FLOAT Literal",
     "TWCUI_Util_IntLiteral": "INTEGER Literal",
diff --git a/saving.py b/image.py
similarity index 100%
rename from saving.py
rename to image.py