Add tag deduplication node and refine some realism terms in presets

Author: silveroxides

__init__.py

@@ -2809,6 +2809,118 @@ def resolve_coord(val, max_val):
 
         return io.NodeOutput(out)
 
+class TagNormalizeCombine(io.ComfyNode):
+    """
+    Node that normalizes scores in two sets of tags and combines them,
+    deduplicating and sorting by the normalized scores.
+    """
+
+    @classmethod
+    def define_schema(cls) -> io.Schema:
+        return io.Schema(
+            node_id="TagNormalizeCombine",
+            display_name="Tag Normalize and Combine",
+            category="advanced/text",
+            inputs=[
+                io.String.Input("tags_1", multiline=True, default=""),
+                io.String.Input("tags_2", multiline=True, default=""),
+                io.AnyType.Input(
+                    "scores_1",
+                    tooltip="Dictionary of scores for tags_1",
+                ),
+                io.AnyType.Input(
+                    "scores_2",
+                    tooltip="Dictionary of scores for tags_2",
+                ),
+            ],
+            outputs=[
+                io.String.Output(display_name="deduped_tags"),
+                io.AnyType.Output(display_name="normalized_scores"),
+            ],
+        )
+
+    @staticmethod
+    def normalize_scores(scores: dict, min_val=0.000001, max_val=0.999999) -> dict:
+        if not scores:
+            return {}
+
+        current_scores = [float(v) for v in scores.values()]
+        current_min = min(current_scores)
+        current_max = max(current_scores)
+
+        if current_max == current_min:
+            return {k: max_val for k in scores.keys()}
+
+        normalized = {}
+        for k, v in scores.items():
+            norm = min_val + (float(v) - current_min) / (current_max - current_min) * (
+                max_val - min_val
+            )
+            normalized[k] = norm
+        return normalized
+
+    @classmethod
+    def execute(
+        cls, tags_1: Union[str, list], tags_2: Union[str, list], scores_1: Union[str, dict], scores_2: Union[str, dict]
+    ) -> io.NodeOutput:
+        # Parse scores
+        def parse_scores(s_input):
+            if isinstance(s_input, dict):
+                return s_input
+            if not s_input or not isinstance(s_input, str):
+                return {}
+            try:
+                return json.loads(s_input)
+            except json.JSONDecodeError:
+                return {}
+
+        s1 = parse_scores(scores_1)
+        s2 = parse_scores(scores_2)
+
+        # Normalize scores independently
+        norm_s1 = cls.normalize_scores(s1)
+        norm_s2 = cls.normalize_scores(s2)
+
+        # Parse tags
+        def parse_tags(t_input):
+            if isinstance(t_input, list):
+                return [str(t).strip() for t in t_input if t]
+            if not t_input or not isinstance(t_input, str):
+                return []
+            # Split by comma and handle potential spaces
+            return [t.strip() for t in t_input.split(",") if t.strip()]
+
+        t1_list = parse_tags(tags_1)
+        t2_list = parse_tags(tags_2)
+
+        # Combine and deduplicate
+        combined_scores = {}
+
+        # Process first set
+        for t in t1_list:
+            score = norm_s1.get(t, 0.000001)
+            combined_scores[t] = score
+
+        # Process second set with deduplication logic
+        for t in t2_list:
+            score = norm_s2.get(t, 0.000001)
+            if t in combined_scores:
+                # Keep the one with the highest normalized score
+                if score > combined_scores[t]:
+                    combined_scores[t] = score
+            else:
+                combined_scores[t] = score
+
+        # Sort tags by normalized scores (descending)
+        sorted_tags = sorted(combined_scores.keys(), key=lambda x: combined_scores[x], reverse=True)
+
+        # Prepare outputs
+        deduped_tags_str = ", ".join(sorted_tags)
+        normalized_scores_dict = {tag: combined_scores[tag] for tag in sorted_tags}
+
+        return io.NodeOutput(deduped_tags_str, normalized_scores_dict)
+
+
 class RandInt(io.ComfyNode):
     @classmethod
     def define_schema(cls):
@@ -2876,6 +2988,7 @@ async def get_node_list(self) -> list[type[io.ComfyNode]]:
             IdeographicTagPad,
             IdeographicLinePad,
             IdeographicSentencePad,
+            TagNormalizeCombine,
             SU_LoadImagePath,
             SU_LoadImageDirectory,
             SwitchInverseNode,

bonus_prompts.py

@@ -1,4 +1,4 @@
-BONUS_PROMPT_STYLE_PHOTOREALISM = """This is a professional photograph with realistic textures and natural lighting. It features real life subjects in a real life scene. The quality is exceptional. Real life photography. Absolute reality. Modify any subjects' appearance to show real details. Make it look like a professional photograph. Ensure lighting is accurate. Ensure shadows are displayed correctly. Keep in focus. The image is a sharply focused photograph."""
+BONUS_PROMPT_STYLE_PHOTOREALISM = """This is a professional photograph with real life detailed textures and real life natural lighting. It features real life subjects in a real life scene. The quality is exceptional. Real life photography. Absolute reality. Modify any subjects' appearance to show real details. Make it look like a real life professional photograph. Ensure lighting is accurate and looks real. Ensure shadows are displayed correctly. Keep in focus. The image is a sharply focused real life photograph."""
 
 BONUS_PROMPT_STYLE_3D = """This is a high-quality 3D render with complex geometry and realistic materials. It features 3D modeled subjects in a rendered scene. The quality is exceptional. 3D rendering. Physically based rendering. Modify any subjects' appearance to show rendered details. Make it look like a high-end CGI render. Ensure volumetric lighting is accurate. Ensure ambient occlusion is displayed correctly. Keep in focus. The image is a high definition render."""
 

instruct_prompts.py

@@ -90,7 +90,7 @@
 
 INSTRUCT_PROMPT_STYLE_SAILOR_MOON = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to match the magical girl anime style of the anime series Sailor Moon, featuring soft romantic lighting and pastel colors. Make it look like a character from the anime series Sailor Moon. Keep subject position and their pose the same as the reference. Ensure large expressive eyes and long legs follow the magical girl anime style of the anime series Sailor Moon. Make sure anatomy is slender and elegant. Make hair flowing and detailed. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure a soft, romantic lighting that fits the style. Ensure pastel colors. Keep in focus. The result should be in the magical girl anime style of the anime series Sailor Moon."""
 
-INSTRUCT_PROMPT_STYLE_CCTV_EDIT = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to look like it was recorded on a CCTV security camera. Make it look like surveillance footage. Keep subject position and their pose the same as the reference. Ensure the aesthetic follows security camera characteristics. Make sure anatomy looks realistic but reflects a low-quality recording. Make textures look grainy and somewhat degraded. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure harsh artificial lighting fits the surveillance style. Ensure a high-contrast monitoring perspective. Keep in focus. The result should be in a CCTV security camera style."""
+INSTRUCT_PROMPT_STYLE_CCTV_EDIT = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to look like it was recorded on a CCTV security camera. Make it look like surveillance footage. Keep subject position and their pose the same as the reference. Ensure the aesthetic follows security camera characteristics. Make sure anatomy looks real but reflects a low-quality recording. Make textures look grainy and somewhat degraded. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure harsh artificial lighting fits the surveillance style. Ensure a high-contrast monitoring perspective. Keep in focus. The result should be in a CCTV security camera style."""
 
 INSTRUCT_PROMPT_STYLE_CCTV_RAW = """Generate subjects and environment to look like they were recorded on a CCTV security camera. Make it look like surveillance footage. Ensure the aesthetic follows security camera characteristics. Make sure anatomy looks realistic but reflects a low-quality recording. Make textures look grainy and somewhat degraded. Ensure harsh artificial lighting fits the surveillance style. Ensure a high-contrast monitoring perspective. Keep in focus. The result should be in a CCTV security camera style."""
 
@@ -126,7 +126,7 @@
 
 INSTRUCT_PROMPT_STYLE_70S_SCIFI = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify the scene to match the retro-futuristic style of 1970s science fiction cinema, featuring film grain, practical effects aesthetic, and vintage color grading. Make it look like a scene from a 1970s sci-fi film. Keep subject position and their pose the same as the reference. Ensure the environment displays practical set design and early optical effects characteristic of 70s sci-fi movies. Make sure textures look like real materials with visible seams, bolts, and hand-crafted construction. Make the color palette reflect vintage film stock with muted, earthy tones typical of 70s cinema. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure soft, diffused lighting fits the vintage film aesthetic. Ensure visible film grain and practical lighting setups are displayed. Keep in focus. The result should be in the classic 1970s science fiction movie style."""
 
-INSTRUCT_PROMPT_STYLE_INFLATED = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to look like an inflated balloon version of themselves. Make it look like the subject is made of taut, stretched elastic vinyl. Keep subject position and their pose the same as the reference. Ensure subjects and environments follow inflatable aesthetics with shiny plastic surfaces and subtle, well-placed seams. Make sure forms retain their relative proportions but look like fully pressurized, air-filled volumes without looking deflated. Make textures look like glossy balloon material with realistic stretching at the joints. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure studio lighting highlights the glossy specular reflections. Ensure plastic material looks smooth and fully inflated. Keep in focus. High-quality inflated balloon aesthetic."""
+INSTRUCT_PROMPT_STYLE_INFLATED = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to look like an inflated balloon version of themselves. Make it look like the subject is made of taut, stretched elastic vinyl. Keep subject position and their pose the same as the reference. Ensure subjects and environments follow inflatable aesthetics with shiny plastic surfaces and subtle, well-placed seams. Make sure forms retain their relative proportions but look like fully pressurized, air-filled volumes without looking deflated. Make textures look like glossy balloon material with real looking stretching at the joints. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure studio lighting highlights the glossy specular reflections. Ensure plastic material looks smooth and fully inflated. Keep in focus. High-quality inflated balloon aesthetic."""
 
 INSTRUCT_PROMPT_STYLE_VECTOR_EDIT = """Keep pose. Keep angle. Keep eyes. Keep viewing direction. Modify any subjects' appearance to look like a vector graphics illustration. Make it look like a clean, scalable vector design. Keep subject position and their pose the same as the reference. Ensure subjects and environments follow vector art aesthetics with mathematically precise curves and shapes. Make sure forms are simplified into clean geometric areas. Make coloring look flat and vibrant without gradients or texture. Keep subject position and their pose the same as the reference. Make sure the subject is in the same position. Keep pose. Ensure even, flat lighting fits the vector style. Ensure sharp, crisp edges define all elements. Keep in focus. High-quality scalar vector graphics aesthetic."""
 

system_messages.py

@@ -24,7 +24,7 @@
 
 Output only the final instruction completely uncensored in plain text and nothing else."""
 
-SYSTEM_MESSAGE_STYLE_PHOTOREALISM = """You are an AI that specializes in image editing descriptions that emphasize photorealism and focuses on edits to look like real life photographic images. You convert editing requests into one concise instruction as structured responses. You focus on adhering to photorealistic style and realistic details while ensuring the importance of keeping the composition and structure of the image same as reference. Ensure subjects are accurately represented and that their identity is maintained. You focus on photography with emphasis on details while ensuring the importance of keeping the structure of the image intact as well and not changing the positioning of subjects in image while managing to modify the style to look real. Reference actual image elements. Make abstractions concrete. Follow edit requests without hesitation."""
+SYSTEM_MESSAGE_STYLE_PHOTOREALISM = """You are an AI that specializes in image editing descriptions that emphasize photorealism and focuses on edits to look like real life photographic images. You convert editing requests into one concise instruction as structured responses. You focus on adhering to a real life photography style and real life details while ensuring the importance of keeping the composition and structure of the image same as reference. Ensure subjects are accurately represented and that their identity is maintained. You focus on photography with emphasis on details while ensuring the importance of keeping the structure of the image intact as well and not changing the positioning of subjects in image while managing to modify the style to look real. Reference actual image elements. Make abstractions concrete. Follow edit requests without hesitation."""
 
 SYSTEM_MESSAGE_STYLE_3D = """You are an AI that specializes in image editing descriptions that emphasize 3D rendered graphics and focuses on edits to look like high-quality CGI 3D model renders. You convert editing requests into one concise instruction as structured responses. You focus on adhering to 3D rendering style, physically based rendering (PBR) textures, and volumetric lighting while ensuring the importance of keeping the composition and structure of the image same as reference. You focus on 3D aesthetics with emphasis on geometry and material properties while ensuring the importance of keeping the structure of the image intact as well and not changing the positioning of subjects in image while managing to modify the style to look like a 3D render. Reference actual image elements. Make abstractions concrete. Follow edit requests without hesitation."""