added the GenderControlOutput node

Author: zhangp365

README.md

@@ -36,6 +36,9 @@ This node, ColorCorrectOfUtils, is an extension of the original [ColorCorrect](h
 ## ModifyTextGender
 This node adjusts the text to describe the gender based on the input. If the gender input is 'M', the text will be adjusted to describe as male; if the gender input is 'F', it will be adjusted to describe as female.
 
+## GenderControlOutput
+This node determines the output based on the input gender. If the gender input is 'M', it will output male-specific text, float, and integer values. If the gender input is 'F', it will output female-specific text, float, and integer values.
+
 ## SplitMask
 This node splits one mask into two masks of the same size according to the area of the submasks. If there are more than two areas, it will select the two largest submasks.
 

nodes.py

@@ -314,6 +314,42 @@ def gender_swap(text, gender, gender_map):
         return ' '.join(words)
 
 
+class GenderControlOutput:
+    """
+    This node will modify the prompt string according gender. gender words include M, F
+    """
+    @ classmethod
+    def INPUT_TYPES(cls):
+        return {"required": {
+            "gender_prior": (["", "M", "F"],),
+            "male_text": ("STRING",  {"multiline": True, "defaultBehavior": "input"}),
+            "male_float": ("FLOAT", {"default": 1, "step": 0.1}),  
+            "male_int": ("INT", {"default": 1, "step": 1}),
+            "female_text": ("STRING",  {"multiline": True, "defaultBehavior": "input"}),
+            "female_float": ("FLOAT", {"default": 1, "step": 0.1}),  
+            "female_int": ("INT", {"default": 1, "step": 1}),
+        },
+            "optional": {
+            "gender_alternative": ("STRING", {"forceInput": True}),           
+            }
+        }
+
+    RETURN_TYPES = ("STRING","FLOAT","INT","BOOLEAN","BOOLEAN")
+    RETURN_NAMES = ("gender_text","float","int","is_male","is_female")
+    FUNCTION = "fun"
+    CATEGORY = "utils/text/operations"
+
+    @ staticmethod
+    def fun(gender_prior,male_text,male_float,male_int,female_text,female_float,female_int, gender_alternative=None):
+        gender= gender_prior if gender_prior else gender_alternative
+        if gender is None or gender.upper() not in ["M", "F"]:
+            raise Exception("can't get any gender input.")
+        if gender.upper()== "M":
+            return (male_text, male_float, male_int, True, False)
+        else:
+            return (female_text, female_float, female_int, False, True)
+        
+
 class ImageConcanateOfUtils:
     @classmethod
     def INPUT_TYPES(s):
@@ -931,6 +967,7 @@ def forward(self, image, upscale_model, tile_size=512, threshold_of_xl_area=0.9)
     "ImageBatchOneOrMore": ImageBatchOneOrMore,
     "ConcatTextOfUtils": ConcatTextOfUtils,
     "ModifyTextGender": ModifyTextGender,
+    "GenderControlOutput": GenderControlOutput,
     "IntAndIntAddOffsetLiteral": IntAndIntAddOffsetLiteral,
     "IntMultipleAddLiteral": IntMultipleAddLiteral,
     "ImageConcanateOfUtils": ImageConcanateOfUtils,
@@ -954,6 +991,7 @@ def forward(self, image, upscale_model, tile_size=512, threshold_of_xl_area=0.9)
     "ImageBatchOneOrMore": "Batch Images One or More",
     "ConcatTextOfUtils": "Concat text",
     "ModifyTextGender": "Modify Text Gender",
+    "GenderControlOutput": "Gender control output",
     "IntAndIntAddOffsetLiteral": "Int And Int Add Offset Literal",
     "IntMultipleAddLiteral": "Int Multiple and Add Literal",
     "ImageConcanateOfUtils": "Image Concanate of utils",

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-utils-nodes"
-description = "Nodes:LoadImageWithSwitch, ImageBatchOneOrMore, ModifyTextGender, ImageCompositeMaskedWithSwitch, ColorCorrectOfUtils, SplitMask, MaskFastGrow, CheckpointLoaderSimpleWithSwitch, ImageResizeTo8x, MatchImageRatioToPreset, UpscaleImageWithModelIfNeed etc."
-version = "1.0.9"
+description = "Nodes:LoadImageWithSwitch, ImageBatchOneOrMore, ModifyTextGender, GenderControlOutput, ImageCompositeMaskedWithSwitch, ColorCorrectOfUtils, SplitMask, MaskFastGrow, CheckpointLoaderSimpleWithSwitch, ImageResizeTo8x, MatchImageRatioToPreset, UpscaleImageWithModelIfNeed etc."
+version = "1.1.0"
 license = { file = "LICENSE" }
 dependencies = []