From ee09d060fe52af9af5b643688cd9d97758d0384e Mon Sep 17 00:00:00 2001 From: Running Over Glowies <100057094+RunningOverGlowies@users.noreply.github.com> Date: Mon, 8 Dec 2025 13:26:49 +0000 Subject: [PATCH] Delete keyword_generator.py --- keyword_generator.py | 38 -------------------------------------- 1 file changed, 38 deletions(-) delete mode 100644 keyword_generator.py diff --git a/keyword_generator.py b/keyword_generator.py deleted file mode 100644 index 86d2d7a..0000000 --- a/keyword_generator.py +++ /dev/null @@ -1,38 +0,0 @@ -#https://github.com/Black-Lioness/ComfyUI-PromptUtils -import random -import json -import os -from typing import Dict, Tuple - -_CACHE = {} - -def load_json_file(file_name): - if file_name not in _CACHE: - file_path = os.path.join(os.path.dirname(__file__), "data", file_name) - with open(file_path, "r") as file: - _CACHE[file_name] = json.load(file) - return _CACHE[file_name] - -class KeywordGenerator: - def __init__(self, seed: int = None): - self.rng = random.Random(seed) - - @classmethod - def INPUT_TYPES(cls) -> Dict: - return { - "optional": { - "prompt": ("STRING", {"multiline": True, "forceInput": True}), - "keywords_count": ("INT", {"default": 1, "min": 1, "max": 50, "step": 1}), - "seed": ("INT", {"default": random.randint(0, 30000), "min": 0, "max": 30000, "step": 1}), - }, - } - - RETURN_TYPES = ("STRING",) - RETURN_NAMES = ("Keyword(s)",) - FUNCTION = "generate_keyword" - CATEGORY = "Prompt" - - def generate_keyword(self, keywords_count: int, prompt: str = "", seed: int = None) -> str: - if seed is not None: - self.rng.seed(seed) - return ((prompt+" " if prompt != "" else prompt)+", ".join(self.rng.sample(load_json_file("keywords.json"), keywords_count)).lower(),) \ No newline at end of file