diff --git a/TT_FemaleClothes.py b/TT_FemaleClothes.py deleted file mode 100644 index 8cb6234..0000000 --- a/TT_FemaleClothes.py +++ /dev/null @@ -1,39 +0,0 @@ -import os -import json -import random -from typing import List -import comfy.model_management -from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS - -class TT_FemaleClothes: - @classmethod - def INPUT_TYPES(cls): - options = cls.load_options() - if options: - options.insert(0, "Random Choice") # Add 'Random Choice' at the top - return {"required": {"selection": (options, {"default": options[0] if options else ""}), "seed": ("INT", {"default": 42})}} - - RETURN_TYPES = ("STRING",) - FUNCTION = "output_selection" - CATEGORY = "TT_TextTools" - - @staticmethod - def load_options() -> List[str]: - file_path = os.path.join(os.path.dirname(__file__), "FemaleClothes.txt") - if os.path.exists(file_path): - with open(file_path, "r", encoding="utf-8") as f: - return [line.strip() for line in f.readlines() if line.strip()] - return ["No options found"] - - def output_selection(self, selection: str, seed: int): - if selection == "Random Choice": - options = self.load_options() - if options and "Random Choice" in options: - options.remove("Random Choice") # Remove 'Random Choice' before picking - random.seed(seed) # Set seed for reproducibility - return (random.choice(options) if options else "No options available",) - return (selection,) - -# Register the node -NODE_CLASS_MAPPINGS["TT_FemaleClothes"] = TT_FemaleClothes -NODE_DISPLAY_NAME_MAPPINGS["TT_FemaleClothes"] = "TT Female Clothes"