Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,520 Bytes
1b80e0f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import random
_choice = ["YES", "NO"]
_range = ["Fixed", "Random"]
class DynamicPrompt:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(s):
return {"required":
{
"variable_prompt": ("STRING", {"forceInput": True} ),
"cached": (_choice,{"default": "NO"} ),
"number_of_random_tag": (_range,{"default": "Random"} ),
"fixed_number_of_random_tag": ("INT", {"default": 1, "min": 1}),
},
"optional":{
"fixed_prompt": ("STRING", {"forceInput": True} ),
}
}
RETURN_TYPES = ("STRING",)
FUNCTION = "prompt_generator"
CATEGORY = "N-Suite/Conditioning"
OUTPUT_NODE = True
"""
Generates a prompt based on fixed_prompt and variable_prompt.
Args:
fixed_prompt (str): The fixed prompt for generating the prompt.
variable_prompt (str): The variable prompt for generating the prompt.
Returns:
tuple: A tuple containing the generated prompt.
"""
def prompt_generator(self, variable_prompt,cached,number_of_random_tag,fixed_number_of_random_tag, fixed_prompt = ""):
prompt = ""
if fixed_prompt == "undefined":
fixed_prompt = ""
#if not an int
if not str(fixed_prompt):
fixed_prompt = ""
if variable_prompt == "undefined":
variable_prompt = ""
#if not an int
if not str(variable_prompt):
variable_prompt = ""
try:
if variable_prompt.strip() != "":
add_feature = []
if len(variable_prompt.split(",")) > 0:
variable_prompt_list = variable_prompt.split(",")
if number_of_random_tag == "Random":
random_range = random.randint(1, len(variable_prompt_list))
else:
if fixed_number_of_random_tag <= len(variable_prompt_list):
random_range = fixed_number_of_random_tag
else:
random_range = len(variable_prompt_list)
for i in range(random_range):
len_list = len(variable_prompt_list)-1
add_feature.append(variable_prompt_list.pop(random.randint(0, len_list)).strip())
# random.choice(variable_prompt.split(","))
if fixed_prompt.strip() != "":
prompt = f"{fixed_prompt},{ ','.join(add_feature) }"
else:
prompt = ','.join(add_feature)
else:
print("Warning: Variable prompt is empty.")
except Exception as e:
print(f"Error: something went wrong.\n {e}")
prompt = fixed_prompt
return {"ui": {"text": prompt}, "result": (prompt,)}
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
"DynamicPrompt [n-suite]": DynamicPrompt
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"DynamicPrompt [n-suite]": "Dynamic Prompt [🅝-🅢🅤🅘🅣🅔]"
}
|