Initial commit

2025-12-29 20:50:25 +00:00
commit 2417dcc090
50 changed files with 1596 additions and 0 deletions
--- a/nodes/character_reader/character_reader.py
+++ b/nodes/character_reader/character_reader.py
@@ -0,0 +1,117 @@
+import json
+import os
+import random
+
+class CharacterJsonReader:
+    @classmethod
+    def INPUT_TYPES(s):
+        base_path = os.path.dirname(os.path.realpath(__file__))
+        char_dir = os.path.join(base_path, "characters")
+        
+        characters = []
+        if os.path.exists(char_dir):
+            characters = sorted([f for f in os.listdir(char_dir) if f.endswith('.json')])
+        
+        return {
+            "required": {
+                "character_file": (characters, ),
+                "selection_mode": (["manual", "sequential", "random"], {"default": "manual"}),
+                "repeat_count": ("INT", {"default": 1, "min": 1, "max": 100, "step": 1}),
+            }
+        }
+
+    @classmethod
+    def IS_CHANGED(s, character_file, selection_mode, repeat_count):
+        if selection_mode == "random" or selection_mode == "sequential":
+            return float("nan")
+        return character_file
+
+    RETURN_TYPES = (
+        "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING", "STRING",
+        "STRING", "STRING", "STRING", "STRING", "STRING", "STRING",
+        "STRING", "STRING", "STRING", "STRING",
+        "STRING", "FLOAT", "STRING"
+    )
+    RETURN_NAMES = (
+        "name", "base_specs", "hair", "eyes", "expression", "hands", "arms", "torso", "pelvis", "legs", "feet", "distinguishing_marks",
+        "inner_layer", "outer_layer", "lower_body", "footwear", "gloves", "accessories",
+        "aesthetic", "primary_color", "secondary_color", "tertiary_color",
+        "lora_name", "lora_weight", "lora_triggers"
+    )
+    
+    FUNCTION = "read_character"
+    CATEGORY = "AODH Pack"
+
+    _current_index = 0
+    _current_count = 0
+    _last_selection = None
+
+    def read_character(self, character_file, selection_mode, repeat_count):
+        base_path = os.path.dirname(os.path.realpath(__file__))
+        char_dir = os.path.join(base_path, "characters")
+        
+        characters = sorted([f for f in os.listdir(char_dir) if f.endswith('.json')])
+        
+        if not characters:
+            return ("",) * 22 + ("", 0.0, "")
+
+        if selection_mode == "manual":
+            selected_file = character_file
+        elif selection_mode == "sequential":
+            if self.__class__._current_count >= repeat_count:
+                self.__class__._current_index += 1
+                self.__class__._current_count = 0
+            selected_file = characters[self.__class__._current_index % len(characters)]
+            self.__class__._current_count += 1
+        elif selection_mode == "random":
+            if self.__class__._current_count >= repeat_count or self.__class__._last_selection is None or self.__class__._last_selection not in characters:
+                self.__class__._last_selection = random.choice(characters)
+                self.__class__._current_count = 0
+            selected_file = self.__class__._last_selection
+            self.__class__._current_count += 1
+        else:
+            selected_file = character_file
+
+        file_path = os.path.join(char_dir, selected_file)
+        
+        with open(file_path, 'r') as f:
+            data = json.load(f)
+            
+        identity = data.get("identity", {})
+        wardrobe = data.get("wardrobe", {})
+        styles = data.get("styles", {})
+        lora = data.get("lora", {})
+        
+        def get_val(obj, key, prepend_comma=True):
+            val = obj.get(key, "")
+            if val and prepend_comma and not val.startswith(","):
+                val = ", " + val
+            return val
+
+        return (
+            get_val(data, "character_id", prepend_comma=False),
+            get_val(identity, "base_specs"),
+            get_val(identity, "hair"),
+            get_val(identity, "eyes"),
+            get_val(identity, "expression"),
+            get_val(identity, "hands"),
+            get_val(identity, "arms"),
+            get_val(identity, "torso"),
+            get_val(identity, "pelvis"),
+            get_val(identity, "legs"),
+            get_val(identity, "feet"),
+            get_val(identity, "distinguishing_marks"),
+            get_val(wardrobe, "inner_layer"),
+            get_val(wardrobe, "outer_layer"),
+            get_val(wardrobe, "lower_body"),
+            get_val(wardrobe, "footwear"),
+            get_val(wardrobe, "gloves"),
+            get_val(wardrobe, "accessories"),
+            get_val(styles, "aesthetic"),
+            get_val(styles, "primary_color"),
+            get_val(styles, "secondary_color"),
+            get_val(styles, "tertiary_color"),
+            get_val(lora, "lora_name", prepend_comma=False),
+            float(lora.get("lora_weight", 0.0) if lora.get("lora_weight") else 0.0),
+            get_val(lora, "lora_triggers"),
+        )