Upload wildcard_1.py

wildcard_1.py

@@ -0,0 +1,325 @@
+import random
+
+
+class Wildcard_1:
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                # Seed controls the whole combo (same seed -> same pair)
+                "seed": ("INT", {
+                    "default": 0,
+                    "min": 0,
+                    "max": 0xffffffffffffffff,
+                    "step": 1
+                }),
+            }
+        }
+
+    # Two text outputs: female + male
+    RETURN_TYPES = ("STRING", "STRING",)
+    RETURN_NAMES = ("female_prompt", "male_prompt",)
+    FUNCTION = "generate"
+    CATEGORY = "wildcards"
+
+    def generate(self, seed: int):
+        rnd = random.Random(int(seed))
+
+        # -------------------------
+        # Category 1: fixed gender token
+        # -------------------------
+        category1_f = "1girl"
+        category1_m = "1boy"
+
+        # -------------------------
+        # Category 2: POV of camera (8 directions, LEFTSIDE/RIGHTSIDE uppercase)
+        # -------------------------
+        pov_options = [
+            "straight-on, frontview",
+            "diagonal LEFTSIDE, three quarter view LEFTSIDE",
+            "sideview LEFTSIDE, from LEFTSIDE side",
+            "frontview POV",
+            "sideview RIGHTSIDE, from RIGHTSIDE side",
+            "diagonal RIGHTSIDE, three quarter view RIGHTSIDE",
+            "frontview, low angle POV, from below",
+            "straight-on, from below, low angle view",
+        ]
+        pov = rnd.choice(pov_options)
+
+        # -------------------------
+        # Category 3: woman / man / girl / boy / grandma / grandpa
+        # (chosen via female word, then mapped to male counterpart)
+        # -------------------------
+        female_roles = ["girl", "woman", "loli"]
+        role_f = rnd.choice(female_roles)
+        role_m = {
+            "girl": "boy",
+            "woman": "man",
+            "loli": "shota",
+        }[role_f]
+
+        # -------------------------
+        # Category 4: social + inner personality (two tokens)
+        # second word is the same and gender-correct
+        # e.g. "rich woman, sadistic woman"
+        # -------------------------
+        social_descriptors = [
+            "rich", "poor", "noble", "commoner",
+            "merchant", "warrior", "scholar", "artist",
+            "assassin", "guard", "leader", "servant",
+            "clumsy", "elegant", "streetwise", "aristocratic",
+            "rural", "city", "rebellious", "stoic",
+        ]
+        inner_descriptors = [
+            "sadistic", "kind", "gentle", "cold", "calculating",
+            "cheerful", "melancholic", "serious", "playful",
+            "paranoid", "confident", "insecure", "arrogant",
+            "humble", "curious", "apathetic", "obsessive",
+            "protective", "jealous", "ambitious",
+        ]
+        social = rnd.choice(social_descriptors)
+        inner = rnd.choice(inner_descriptors)
+
+        identity_f1 = f"{social} {role_f}"
+        identity_f2 = f"{inner} {role_f}"
+        identity_m1 = f"{social} {role_m}"
+        identity_m2 = f"{inner} {role_m}"
+
+        # -------------------------
+        # Skin tone (shared)
+        # -------------------------
+        skin_tones = [
+            "very pale skincolor",
+            "fair skincolor",
+            "light olive skincolor",
+            "olive skincolor",
+            "tan skincolor",
+            "dark-tan skincolor",
+            "brown skincolor",
+            "dark-brown skincolor",
+        ]
+        skin = rnd.choice(skin_tones)
+
+        # -------------------------
+        # Category 5: eyes = color + style + gaze direction
+        # (no eyewear)
+        # -------------------------
+        eye_colors = [
+            "blue eyes",
+            "green eyes",
+            "hazel eyes",
+            "amber eyes",
+            "grey eyes",
+            "dark-brown eyes",
+            "light-brown eyes",
+            "violet eyes",
+        ]
+        eye_styles = [
+            "sharp eyes",
+            "narrow eyes",
+            "gentle eyes",
+            "wide eyes",
+            "sleepy eyes",
+            "intense eyes",
+            "cold eyes",
+            "warm eyes",
+            "mischievous eyes",
+            "tired eyes",
+        ]
+        eye_directions = [
+            "",
+        ]
+        eye_color = rnd.choice(eye_colors)
+        eye_style = rnd.choice(eye_styles)
+        eye_direction = rnd.choice(eye_directions)
+        eyes = ", ".join([eye_color, eye_style, eye_direction])
+
+        # -------------------------
+        # Category 6: hair (color, length, hairstyle)
+        # Hair colors: avoid ambiguous "red hair", use "ginger" etc.
+        # -------------------------
+        hair_colors = [
+            "black hair",
+            "ginger hair",
+            "dark-brown hair",
+            "light-brown hair",
+            "chestnut-brown hair",
+            "platinum-blonde hair",
+            "ash-blonde hair",
+            "golden-blonde hair",
+            "silver hair",
+            "white hair",
+            "grey hair",
+            "blue-dyed hair",
+            "pink-dyed hair",
+            "purple-dyed hair",
+        ]
+        hair_lengths = [
+            "very short hair",
+            "short hair",
+            "medium-length hair",
+            "long hair",
+            "very long hair",
+        ]
+        hair_styles = [
+            "straight hair",
+            "wavy hair",
+            "curly hair",
+            "messy hair",
+            "neatly combed hair",
+            "braided hair",
+            "pony tail hair",
+            "twin braids hair",
+            "bun hair",
+        ]
+        hair = ", ".join([
+            rnd.choice(hair_colors),
+            rnd.choice(hair_lengths),
+            rnd.choice(hair_styles),
+        ])
+
+        # -------------------------
+        # Category 7: top garment
+        # color + topgarment, material + topgarment, adjective + topgarment
+        # e.g. "red shirt, linen shirt, torn shirt"
+        # -------------------------
+        top_colors = [
+            "red", "blue", "green", "yellow", "black",
+            "white", "grey", "brown", "purple", "pink", "beige",
+        ]
+        top_types = ["shirt", "blouse", "hoodie", "jacket", "sweater", "coat", "vest", "tank top"]
+        top_materials_by_type = {
+            "shirt": ["linen", "cotton", "silk"],
+            "blouse": ["linen", "cotton", "silk", "chiffon"],
+            "hoodie": ["cotton", "fleece", "synthetic"],
+            "jacket": ["leather", "denim", "canvas"],
+            "sweater": ["wool", "cotton", "cashmere"],
+            "coat": ["wool", "cotton"],
+            "vest": ["leather", "denim", "cotton"],
+            "tank top": ["cotton", "synthetic"],
+        }
+        top_adjectives = [
+            "torn", "elegant", "simple", "expensive",
+            "casual", "formal", "wrinkled", "clean",
+            "ornate", "plain", "striped", "checkered",
+            "logo-print", "oversized", "fitted", "loose",
+        ]
+
+        top_type = rnd.choice(top_types)
+        top_color = rnd.choice(top_colors)
+        top_material = rnd.choice(top_materials_by_type[top_type])
+        top_adj = rnd.choice(top_adjectives)
+
+        top = ", ".join([
+            f"{top_color} {top_type}",
+            f"{top_material} {top_type}",
+            f"{top_adj} {top_type}",
+        ])
+
+        # -------------------------
+        # Category 8: bottom garment (NO SKIRT)
+        # color + bottom, material + bottom, adjective + bottom
+        # -------------------------
+        bottom_colors = [
+            "black", "blue", "dark-blue", "grey",
+            "brown", "white", "khaki", "olive", "burgundy",
+        ]
+        bottom_types = ["pants", "jeans", "shorts", "cargo pants", "leggings"]
+        bottom_materials_by_type = {
+            "pants": ["wool", "cotton", "denim", "leather"],
+            "jeans": ["denim"],
+            "shorts": ["cotton", "denim"],
+            "cargo pants": ["cotton", "canvas"],
+            "leggings": ["cotton", "synthetic"],
+        }
+        bottom_adjectives = [
+            "slim", "baggy", "loose", "tight",
+            "high-waist", "low-waist", "ripped", "clean",
+            "casual", "formal", "wrinkled", "plain",
+            "patterned", "striped", "folded-hem",
+        ]
+
+        bottom_type = rnd.choice(bottom_types)
+        bottom_color = rnd.choice(bottom_colors)
+        bottom_material = rnd.choice(bottom_materials_by_type[bottom_type])
+        bottom_adj = rnd.choice(bottom_adjectives)
+
+        bottom = ", ".join([
+            f"{bottom_color} {bottom_type}",
+            f"{bottom_material} {bottom_type}",
+            f"{bottom_adj} {bottom_type}",
+        ])
+
+        # -------------------------
+        # Category 9: footwear
+        # color + footwear, adjective + footwear, adjective + footwear
+        # -------------------------
+        foot_colors = ["black", "brown", "grey", "white", "red", "blue", "tan"]
+        footwear_types = [
+            "boots", "sneakers", "shoes", "sandals",
+            "loafers", "ankle boots", "knee-high boots",
+        ]
+        footwear_adjectives = [
+            "leather", "worn", "polished", "dirty",
+            "sporty", "elegant", "heavy", "thick-soled",
+            "lace-up", "strapped", "high-heeled", "flat",
+        ]
+
+        footwear = rnd.choice(footwear_types)
+        foot_color = rnd.choice(foot_colors)
+        foot_adj1 = rnd.choice(footwear_adjectives)
+        remaining = [a for a in footwear_adjectives if a != foot_adj1]
+        foot_adj2 = rnd.choice(remaining) if remaining else foot_adj1
+
+        feet = ", ".join([
+            f"{foot_color} {footwear}",
+            f"{foot_adj1} {footwear}",
+            f"{foot_adj2} {footwear}",
+        ])
+
+
+        # -------------------------
+        # Assemble prompts
+        # -------------------------
+        parts_f = [
+            category1_f,      # 1girl
+            pov,              # POV string
+            role_f,           # woman/girl/grandma
+            identity_f1,      # social descriptor
+            identity_f2,      # inner personality
+            skin,             # skincolor
+            eyes,             # eye color + style + direction
+            hair,             # hair
+            top,              # top garment
+            bottom,           # bottom garment (no skirt)
+            feet,             # footwear
+        ]
+
+        parts_m = [
+            category1_m,      # 1boy
+            pov,
+            role_m,           # man/boy/grandpa
+            identity_m1,
+            identity_m2,
+            skin,
+            eyes,
+            hair,
+            top,
+            bottom,
+            feet,
+        ]
+
+        female_prompt = ", ".join(parts_f) + "."
+        male_prompt = ", ".join(parts_m) + "."
+
+        return (female_prompt, male_prompt)
+
+
+# ComfyUI requires these mappings at module level
+NODE_CLASS_MAPPINGS = {
+    "Wildcard_1": Wildcard_1,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Wildcard_1": "Wildcard_1",
+}