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 [🅝-🅢🅤🅘🅣🅔]"
}