Add batch_size to ResolutionMaster outputs and UI

Updated aztoolkit.py to include batch_size in the return values and output names of ResolutionMaster. Modified ResolutionMaster.js to display a pink 'LAT' label for the latent output and clarified the rescale value display.

Author: Azornes

aztoolkit.py

@@ -44,8 +44,8 @@ def INPUT_TYPES(cls):
             },
         }
 
-    RETURN_TYPES = ("INT", "INT", "FLOAT", "LATENT")
-    RETURN_NAMES = ("width", "height", "rescale_factor", "latent")
+    RETURN_TYPES = ("INT", "INT", "FLOAT", "INT", "LATENT")
+    RETURN_NAMES = ("width", "height", "rescale_factor", "batch_size", "latent")
     FUNCTION = "main"
     CATEGORY = "utils/azToolkit"
 
@@ -93,4 +93,4 @@ def main(self, mode, width, height, auto_detect, rescale_mode, rescale_value, ba
         # Generate latent tensor
         latent = torch.zeros([batch_size, 4, height // 8, width // 8], device=self.device)
 
-        return (width, height, rescale_factor, {"samples": latent})
+        return (width, height, rescale_factor, batch_size, {"samples": latent})

js/ResolutionMaster.js

@@ -533,7 +533,7 @@ class ResolutionMasterCanvas {
             ctx.fillStyle = this.hoverElement === 'heightValueArea' ? "#F89" : "#F89";
             ctx.fillText(this.heightWidget.value.toString(), node.size[0] - 20, y_offset_2);
 
-            // Rescale value (non-clickable)
+            // Rescale value (non-clickable, green text)
             ctx.fillStyle = "#9F8";
             ctx.fillText(props.rescaleValue.toFixed(2), node.size[0] - 20, y_offset_3);
 
@@ -558,6 +558,14 @@ class ResolutionMasterCanvas {
 
             ctx.fillStyle = this.hoverElement === 'batchSizeValueArea' ? "#FAB" : "#F8B";
             ctx.fillText(this.batchSizeWidget.value.toString(), node.size[0] - 20, y_offset_4);
+            
+            // Latent output label - pink "lat" text only (non-clickable, read-only)
+            const y_offset_5 = 5 + (LiteGraph.NODE_SLOT_HEIGHT * 4.5);
+            
+            ctx.fillStyle = "#F8B"; // Pink text color (same as batch_size)
+            ctx.font = "bold 12px Arial";
+            ctx.textAlign = "right";
+            ctx.fillText("LAT", node.size[0] - 20, y_offset_5);
         }
     }