Added support for watermark, and regionalPrompt

2026-03-12 14:21:38 -04:00
parent c17d46d70b
commit aa5f547b99
13 changed files with 1219 additions and 616 deletions
@@ -0,0 +1,80 @@
+from typing import Dict, Any, Tuple
+
+
+class RunwareRegionalPromptingAdvancedFeature:
+    """
+    Runware Regional Prompting Advanced Feature Node
+
+    Builds the advancedFeatures.regionalPrompting configuration object:
+      - injectSteps
+      - backgroundPrompt
+      - baseRatio
+      - regions (connected from the Regions node)
+    """
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "injectSteps": ("INT", {
+                    "tooltip": "Steps to inject regional masks (1-100).",
+                    "default": 15,
+                    "min": 1,
+                    "max": 100,
+                }),
+                "backgroundPrompt": ("STRING", {
+                    "tooltip": "Background prompt applied to uncovered areas.",
+                    "default": "",
+                }),
+                "baseRatio": ("FLOAT", {
+                    "tooltip": "Base prompt attention weight (0.0 - 1.0).",
+                    "default": 0.0,
+                    "min": 0.0,
+                    "max": 1.0,
+                    "step": 0.01,
+                }),
+                "regions": ("RUNWAREREGIONALPROMPTINGREGIONS", {
+                    "tooltip": "Connect Runware Regional Prompting Advanced Feature Regions node to define regional prompts and masks.",
+                }),
+            },
+        }
+
+    RETURN_TYPES: Tuple[str] = ("RUNWAREREGIONALPROMPTINGADVFEATURE",)
+    RETURN_NAMES = ("regionalPrompting",)
+    FUNCTION = "createRegionalPrompting"
+    CATEGORY = "Runware"
+    DESCRIPTION = (
+        "Configure regional prompting advanced feature (injectSteps, backgroundPrompt, "
+        "baseRatio, and regions). Connect output to Runware Image  Advanced Feature Input "
+        "regionalPrompting input."
+    )
+
+    def createRegionalPrompting(self, regions=None, **kwargs) -> tuple:
+        """Build regionalPrompting dict for advancedFeatures.regionalPrompting."""
+        inject_steps = int(kwargs.get("injectSteps", 15))
+        base_ratio = float(kwargs.get("baseRatio", 0.2))
+        background_prompt = (kwargs.get("backgroundPrompt", "") or "").strip()
+
+        config: Dict[str, Any] = {
+            "injectSteps": inject_steps,
+            "baseRatio": base_ratio,
+        }
+
+        if background_prompt:
+            config["backgroundPrompt"] = background_prompt
+
+        if isinstance(regions, list) and regions:
+            config["regions"] = regions
+
+        return (config,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "RunwareRegionalPromptingAdvancedFeature": RunwareRegionalPromptingAdvancedFeature,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "RunwareRegionalPromptingAdvancedFeature": "Runware Regional Prompting Advanced Feature",
+}
+