import sys import os import folder_paths from datetime import datetime import uuid import platform custom_nodes_path = os.path.join(folder_paths.base_path, "custom_nodes") onebuttonprompt_path = os.path.join(custom_nodes_path, "OneButtonPrompt") sys.path.append(onebuttonprompt_path) from .build_dynamic_prompt import * from .csv_reader import * from .one_button_presets import OneButtonPresets OBPresets = OneButtonPresets() allpresets = [OBPresets.RANDOM_PRESET_OBP] + list(OBPresets.opb_presets.keys()) artists = ["all", "all (wild)", "none", "popular", "greg mode", "3D", "abstract", "angular", "anime" ,"architecture", "art nouveau", "art deco", "baroque", "bauhaus", "cartoon", "character", "children's illustration", "cityscape", "cinema", "clean", "cloudscape", "collage", "colorful", "comics", "cubism", "dark", "detailed", "digital", "expressionism", "fantasy", "fashion", "fauvism", "figurativism", "gore", "graffiti", "graphic design", "high contrast", "horror", "impressionism", "installation", "landscape", "light", "line drawing", "low contrast", "luminism", "magical realism", "manga", "melanin", "messy", "monochromatic", "nature", "nudity", "photography", "pop art", "portrait", "primitivism", "psychedelic", "realism", "renaissance", "romanticism", "scene", "sci-fi", "sculpture", "seascape", "space", "stained glass", "still life", "storybook realism", "street art", "streetscape", "surrealism", "symbolism", "textile", "ukiyo-e", "vibrant", "watercolor", "whimsical"] artifyartists = ["all", "all (wild)", "popular", "greg mode", "3D", "abstract", "angular", "anime" ,"architecture", "art nouveau", "art deco", "baroque", "bauhaus", "cartoon", "character", "children's illustration", "cityscape", "cinema", "clean", "cloudscape", "collage", "colorful", "comics", "cubism", "dark", "detailed", "digital", "expressionism", "fantasy", "fashion", "fauvism", "figurativism", "gore", "graffiti", "graphic design", "high contrast", "horror", "impressionism", "installation", "landscape", "light", "line drawing", "low contrast", "luminism", "magical realism", "manga", "melanin", "messy", "monochromatic", "nature", "nudity", "photography", "pop art", "portrait", "primitivism", "psychedelic", "realism", "renaissance", "romanticism", "scene", "sci-fi", "sculpture", "seascape", "space", "stained glass", "still life", "storybook realism", "street art", "streetscape", "surrealism", "symbolism", "textile", "ukiyo-e", "vibrant", "watercolor", "whimsical"] imagetypes = ["all", "all - force multiple", "all - anime", "none", "photograph", "octane render","digital art","concept art", "painting", "portrait", "anime", "only other types", "only templates mode", "dynamic templates mode", "art blaster mode", "quality vomit mode", "color cannon mode", "unique art mode", "massive madness mode", "photo fantasy mode", "subject only mode", "fixed styles mode", "the tokinator"] subjects =["all", "object", "animal", "humanoid", "landscape", "concept"] genders = ["all", "male", "female"] emojis = [False, True] models = ["SD1.5", "SDXL", "Stable Cascade", "Anime Model"] prompt_enhancers = ["none", "superprompt-v1"] subjects =["------ all"] subjectsubtypesobject = ["all"] subjectsubtypeshumanoid = ["all"] subjectsubtypesconcept = ["all"] #subjectsubtypesobject = ["all", "generic objects", "vehicles", "food", "buildings", "space", "flora"] #subjectsubtypeshumanoid = ["all", "generic humans", "generic human relations", "celebrities e.a.", "fictional characters", "humanoids", "based on job or title", "based on first name"] #subjectsubtypesconcept = ["all", "event", "the X of Y concepts", "lines from poems", "lines from songs"] amountofflufflist = ["none", "dynamic", "short", "medium", "long"] fluff_reverse_polarity = [False,True] artifymodeslist = ["standard", "remix", "super remix turbo"] artifyamountofartistslist = ["random", "0", "1", "2", "3", "4", "5"] superprompterstyleslist = ['all'] superprompterstyleslist += csv_to_list("superprompter_styles") # Load up stuff for personal artists list, if any # find all artist files starting with personal_artits in userfiles script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory userfilesfolder = os.path.join(script_dir, "./userfiles/" ) for filename in os.listdir(userfilesfolder): if(filename.endswith(".csv") and filename.startswith("personal_artists") and filename != "personal_artists_sample.csv"): name = os.path.splitext(filename)[0] name = name.replace("_"," ",-1).lower() # directly insert into the artists list artists.insert(2, name) # on startup, check if we have a config file, or else create it config = load_config_csv() # load subjects stuff from config generatevehicle = True generateobject = True generatefood = True generatebuilding = True generatespace = True generateflora = True generateanimal = True generatebird = True generatecat = True generatedog = True generateinsect = True generatepokemon = True generatemarinelife = True generatemanwoman = True generatemanwomanrelation = True generatemanwomanmultiple = True generatefictionalcharacter = True generatenonfictionalcharacter = True generatehumanoids = True generatejob = True generatefirstnames = True generatelandscape = True generatelocation = True generatelocationfantasy = True generatelocationscifi = True generatelocationvideogame = True generatelocationbiome = True generatelocationcity = True generateevent = True generateconcepts = True generatepoemline = True generatesongline = True generatecardname = True generateepisodetitle = True generateconceptmixer = True for item in config: # objects if item[0] == 'subject_vehicle' and item[1] != 'on': generatevehicle = False if item[0] == 'subject_object' and item[1] != 'on': generateobject = False if item[0] == 'subject_food' and item[1] != 'on': generatefood = False if item[0] == 'subject_building' and item[1] != 'on': generatebuilding = False if item[0] == 'subject_space' and item[1] != 'on': generatespace = False if item[0] == 'subject_flora' and item[1] != 'on': generateflora = False # animals if item[0] == 'subject_animal' and item[1] != 'on': generateanimal = False if item[0] == 'subject_bird' and item[1] != 'on': generatebird = False if item[0] == 'subject_cat' and item[1] != 'on': generatecat = False if item[0] == 'subject_dog' and item[1] != 'on': generatedog = False if item[0] == 'subject_insect' and item[1] != 'on': generateinsect = False if item[0] == 'subject_pokemon' and item[1] != 'on': generatepokemon = False if item[0] == 'subject_marinelife' and item[1] != 'on': generatemarinelife = False # humanoids if item[0] == 'subject_manwoman' and item[1] != 'on': generatemanwoman = False if item[0] == 'subject_manwomanrelation' and item[1] != 'on': generatemanwomanrelation = False if item[0] == 'subject_manwomanmultiple' and item[1] != 'on': generatemanwomanmultiple = False if item[0] == 'subject_fictional' and item[1] != 'on': generatefictionalcharacter = False if item[0] == 'subject_nonfictional' and item[1] != 'on': generatenonfictionalcharacter = False if item[0] == 'subject_humanoid' and item[1] != 'on': generatehumanoids = False if item[0] == 'subject_job' and item[1] != 'on': generatejob = False if item[0] == 'subject_firstnames' and item[1] != 'on': generatefirstnames = False # landscape if item[0] == 'subject_location' and item[1] != 'on': generatelocation = False if item[0] == 'subject_location_fantasy' and item[1] != 'on': generatelocationfantasy = False if item[0] == 'subject_location_scifi' and item[1] != 'on': generatelocationscifi = False if item[0] == 'subject_location_videogame' and item[1] != 'on': generatelocationvideogame = False if item[0] == 'subject_location_biome' and item[1] != 'on': generatelocationbiome = False if item[0] == 'subject_location_city' and item[1] != 'on': generatelocationcity = False # concept if item[0] == 'subject_event' and item[1] != 'on': generateevent = False if item[0] == 'subject_concept' and item[1] != 'on': generateconcepts = False if item[0] == 'subject_poemline' and item[1] != 'on': generatepoemline = False if item[0] == 'subject_songline' and item[1] != 'on': generatesongline = False if item[0] == 'subject_cardname' and item[1] != 'on': generatecardname = False if item[0] == 'subject_episodetitle' and item[1] != 'on': generateepisodetitle = False if item[0] == 'subject_conceptmixer' and item[1] != 'on': generateconceptmixer = False # build up all subjects we can choose based on the loaded config file if(generatevehicle or generateobject or generatefood or generatebuilding or generatespace or generateflora): subjects.append("--- object - all") if(generateobject): subjects.append("object - generic") if(generatevehicle): subjects.append("object - vehicle") if(generatefood): subjects.append("object - food") if(generatebuilding): subjects.append("object - building") if(generatespace): subjects.append("object - space") if(generateflora): subjects.append("object - flora") if(generateanimal or generatebird or generatecat or generatedog or generateinsect or generatepokemon or generatemarinelife): subjects.append("--- animal - all") if(generateanimal): subjects.append("animal - generic") if(generatebird): subjects.append("animal - bird") if(generatecat): subjects.append("animal - cat") if(generatedog): subjects.append("animal - dog") if(generateinsect): subjects.append("animal - insect") if(generatemarinelife): subjects.append("animal - marine life") if(generatepokemon): subjects.append("animal - pokémon") if(generatemanwoman or generatemanwomanrelation or generatefictionalcharacter or generatenonfictionalcharacter or generatehumanoids or generatejob or generatemanwomanmultiple): subjects.append("--- human - all") if(generatemanwoman): subjects.append("human - generic") if(generatemanwomanrelation): subjects.append("human - relations") if(generatenonfictionalcharacter): subjects.append("human - celebrity") if(generatefictionalcharacter): subjects.append("human - fictional") if(generatehumanoids): subjects.append("human - humanoids") if(generatejob): subjects.append("human - job/title") if(generatefirstnames): subjects.append("human - first name") if(generatemanwomanmultiple): subjects.append("human - multiple") if(generatelandscape or generatelocation or generatelocationfantasy or generatelocationscifi or generatelocationvideogame or generatelocationbiome or generatelocationcity): subjects.append("--- landscape - all") if(generatelocation): subjects.append("landscape - generic") if(generatelocationfantasy): subjects.append("landscape - fantasy") if(generatelocationscifi): subjects.append("landscape - sci-fi") if(generatelocationvideogame): subjects.append("landscape - videogame") if(generatelocationbiome): subjects.append("landscape - biome") if(generatelocationcity): subjects.append("landscape - city") if(generateevent or generateconcepts or generatepoemline or generatesongline or generatecardname or generateepisodetitle or generateconceptmixer): subjects.append("--- concept - all") if(generateevent): subjects.append("concept - event") if(generateconcepts): subjects.append("concept - the x of y") if(generatepoemline): subjects.append("concept - poem lines") if(generatesongline): subjects.append("concept - song lines") if(generatecardname): subjects.append("concept - card names") if(generateepisodetitle): subjects.append("concept - episode titles") if(generateconceptmixer): subjects.append("concept - mixer") # do the same for the subtype subjects # subjectsubtypesobject = ["all"] # subjectsubtypeshumanoid = ["all"] # subjectsubtypesconcept = ["all"] # objects first if(generateobject): subjectsubtypesobject.append("generic objects") if(generatevehicle): subjectsubtypesobject.append("vehicles") if(generatefood): subjectsubtypesobject.append("food") if(generatebuilding): subjectsubtypesobject.append("buildings") if(generatespace): subjectsubtypesobject.append("space") if(generateflora): subjectsubtypesobject.append("flora") # humanoids (should I review descriptions??) if(generatemanwoman): subjectsubtypeshumanoid.append("generic humans") if(generatemanwomanrelation): subjectsubtypeshumanoid.append("generic human relations") if(generatenonfictionalcharacter): subjectsubtypeshumanoid.append("celebrities e.a.") if(generatefictionalcharacter): subjectsubtypeshumanoid.append("fictional characters") if(generatehumanoids): subjectsubtypeshumanoid.append("humanoids") if(generatejob): subjectsubtypeshumanoid.append("based on job or title") if(generatefirstnames): subjectsubtypeshumanoid.append("based on first name") if(generatemanwomanmultiple): subjectsubtypeshumanoid.append("multiple humans") # concepts if(generateevent): subjectsubtypesconcept.append("event") if(generateconcepts): subjectsubtypesconcept.append("the X of Y concepts") if(generatepoemline): subjectsubtypesconcept.append("lines from poems") if(generatesongline): subjectsubtypesconcept.append("lines from songs") if(generatecardname): subjectsubtypesconcept.append("names from card based games") if(generateepisodetitle): subjectsubtypesconcept.append("episode titles from tv shows") if(generateconceptmixer): subjectsubtypesconcept.append("concept mixer") class OneButtonPrompt: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "insanitylevel": ("INT", { "default": 5, "min": 1, #Minimum value "max": 10, #Maximum value "step": 1 #Slider's step }), }, "optional": { "artist": (artists, {"default": "all"}), "imagetype": (imagetypes, {"default": "all"}), "imagemodechance": ("INT", { "default": 20, "min": 1, #Minimum value "max": 100, #Maximum value "step": 1 #Slider's step }), "subject": (subjects, {"default": "------ all"}), "custom_subject": ("STRING", { "multiline": False, #True if you want the field to look like the one on the ClipTextEncode node "default": "" }), "custom_outfit": ("STRING", { "multiline": False, # This is the overwrite for an outfit, super nice "default": "" }), "prompt_prefix": ("STRING", { "multiline": False, # prefix the prompt "default": "" }), "prompt_suffix": ("STRING", { "multiline": False, # Suffix of the prompt "default": "" }), "humanoids_gender": (genders, {"default": "all"}), "emojis":(emojis, {"default": False}), "base_model":(models, {"default": "SDXL"}), "prompt_enhancer":(prompt_enhancers, {"default": "none"}), "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING","STRING", "STRING") RETURN_NAMES = ("prompt","prompt_g", "prompt_l") FUNCTION = "Comfy_OBP" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP(self, insanitylevel, custom_subject, seed, artist, imagetype, subject, imagemodechance, humanoids_gender, emojis, custom_outfit, base_model, prompt_enhancer, prompt_prefix, prompt_suffix): generatedpromptlist = build_dynamic_prompt(insanitylevel,subject,artist,imagetype,False,"",prompt_prefix,prompt_suffix,1,"",custom_subject,True,"",imagemodechance, humanoids_gender,"all", "all", "all", False, emojis, seed, custom_outfit, True, base_model, "", prompt_enhancer) #print(generatedprompt) generatedprompt = generatedpromptlist[0] prompt_g = generatedpromptlist[1] prompt_l = generatedpromptlist[2] return (generatedprompt, prompt_g, prompt_l) class CreatePromptVariant: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "prompt_input": ("STRING", {"default": '', "multiline": True}), }, "optional": { "insanitylevel": ("INT", { "default": 5, "min": 1, #Minimum value "max": 10, #Maximum value "step": 1 #Slider's step }), "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("prompt",) FUNCTION = "Comfy_OBP_PromptVariant" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_PromptVariant(self, prompt_input, insanitylevel, seed): generatedprompt = createpromptvariant(prompt_input, insanitylevel) print(generatedprompt) return (generatedprompt,) # Let us create our own prompt saver. Not everyone has WAS installed class SavePromptToFile: def __init__(self): self.output_dir = folder_paths.get_output_directory() self.type = "output" self.prefix_append = "" @classmethod def INPUT_TYPES(s): return { "required": { "filename_prefix": ("STRING", {"default": "Prompt"}), "positive_prompt": ("STRING",{"multiline": True}), "negative_prompt": ("STRING",{"multiline": True}), }, "optional": { "prompt_g": ("STRING",{"multiline": True}), "prompt_l": ("STRING",{"multiline": True}), }, } OUTPUT_NODE = True RETURN_TYPES = () FUNCTION = "saveprompttofile" CATEGORY = "OneButtonPrompt" def saveprompttofile(self, positive_prompt, prompt_g, prompt_l, negative_prompt, filename_prefix): # Some stuff for the prefix filename_prefix += self.prefix_append # turns out there is some hardcoded stuff on saveimage we have to kind of repeat here # Find the %date:yyyy-M-d% pattern using regular expression pattern = r'%date:([^\%]+)%' match = re.search(pattern, filename_prefix) if match: # Extract the date format from the match date_format = match.group(1) # Get the current date current_date = datetime.now() # convert the ComfyUI standard into Python standard format. # What a crazy way of doing this # first lol, I got to make sure it doesn't overlap things date_format = date_format.replace('M', 'X') date_format = date_format.replace('m', 'Z') # This is so bad # lets make it even worse, it work differently on windows than in Linux if(platform.system() == 'Windows'): date_format = date_format.replace('yyyy', '%Y') date_format = date_format.replace('yy', '%#y') date_format = date_format.replace('X', '%#m') date_format = date_format.replace('d', '%#d') date_format = date_format.replace('h', '%#H') date_format = date_format.replace('Z', '%#M') date_format = date_format.replace('s', '%#S') else: date_format = date_format.replace('yyyy', '%Y') date_format = date_format.replace('yy', '%-y') date_format = date_format.replace('X', '%-m') date_format = date_format.replace('d', '%-d') date_format = date_format.replace('h', '%-H') date_format = date_format.replace('Z', '%-M') date_format = date_format.replace('s', '%-S') # Format the date using the extracted format formatted_date = current_date.strftime(date_format) # Replace the matched pattern with the formatted date filename_prefix = re.sub(pattern, formatted_date, filename_prefix) full_output_folder, filename_short, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir) # make the filename, from from a to the first comma # find the index of the first comma after "of a" or end of the prompt if(positive_prompt.find("of a ") != -1): start_index = positive_prompt.find("of a ") + len("of a ") end_index = positive_prompt.find(",", start_index) if(end_index == -1): end_index=len(positive_prompt) else: start_index = 0 end_index = 128 # extract the desired substring using slicing filename = positive_prompt[start_index:end_index] # cleanup some unsafe things in the filename filename = filename.replace("\"", "") filename = filename.replace("[", "") filename = filename.replace("|", "") filename = filename.replace("]", "") filename = filename.replace("<", "") filename = filename.replace(">", "") filename = filename.replace(":", "_") filename = filename.replace(".", "") filename = re.sub(r'[0-9]+', '', filename) safe_characters = set("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.") # Use regular expression to filter out any characters not in the whitelist filename = re.sub(r"[^{}]+".format(re.escape(''.join(safe_characters))), '', filename) if(filename==""): filename = str(uuid.uuid4()) if(filename_prefix == ""): # create a datetime object for the current date and time # if there is no prefix now = datetime.now() filenamecomplete = now.strftime("%Y%m%d%H%M%S") + "_" + filename.replace(" ", "_").strip() + ".txt" else: # lol since we insert a file, the counter of the image goes up by 1. # So we add 1 here, so the prompt file matches the image file formatted_counter = str(counter + 1).zfill(5) filenamecomplete = filename_short + "_" + formatted_counter + "_" + filename.replace(" ", "_").strip() + ".txt" directoryandfilename = os.path.abspath(os.path.join(full_output_folder, filenamecomplete)) with open(directoryandfilename, 'w', encoding="utf-8") as file: file.write("prompt: " + positive_prompt + "\n") if(len(prompt_g) > 0): file.write("prompt_g: " + prompt_g + "\n") if(len(prompt_l) > 0): file.write("prompt_l: " + prompt_l + "\n") file.write("negative prompt: " + negative_prompt + "\n") return ("done") class OneButtonPreset: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "OneButtonPreset": (allpresets, {"default": "Standard"}), }, "optional": { "base_model":(models, {"default": "SDXL"}), "prompt_enhancer":(prompt_enhancers, {"default": "none"}), "preset_prefix": ("STRING", { "multiline": False, # prefix the preset "default": "" }), "preset_suffix": ("STRING", { "multiline": False, # Suffix of the preset "default": "" }), "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("prompt",) FUNCTION = "Comfy_OBP_OneButtonPreset" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_OneButtonPreset(self, OneButtonPreset, seed, base_model, prompt_enhancer, preset_prefix, preset_suffix): # load the stuff if(OneButtonPreset == OBPresets.RANDOM_PRESET_OBP): selected_opb_preset = OBPresets.get_obp_preset("Standard") else: selected_opb_preset = OBPresets.get_obp_preset(OneButtonPreset) insanitylevel=selected_opb_preset["insanitylevel"] subject=selected_opb_preset["subject"] artist=selected_opb_preset["artist"] chosensubjectsubtypeobject=selected_opb_preset["chosensubjectsubtypeobject"] chosensubjectsubtypehumanoid=selected_opb_preset["chosensubjectsubtypehumanoid"] chosensubjectsubtypeconcept=selected_opb_preset["chosensubjectsubtypeconcept"] chosengender=selected_opb_preset["chosengender"] imagetype=selected_opb_preset["imagetype"] imagemodechance=selected_opb_preset["imagemodechance"] givensubject=selected_opb_preset["givensubject"] smartsubject=selected_opb_preset["smartsubject"] givenoutfit=selected_opb_preset["givenoutfit"] prefixprompt=selected_opb_preset["prefixprompt"] suffixprompt=selected_opb_preset["suffixprompt"] giventypeofimage=selected_opb_preset["giventypeofimage"] antistring=selected_opb_preset["antistring"] generatedprompt = build_dynamic_prompt(insanitylevel=insanitylevel, forcesubject=subject, artists=artist, subtypeobject=chosensubjectsubtypeobject, subtypehumanoid=chosensubjectsubtypehumanoid, subtypeconcept=chosensubjectsubtypeconcept, gender=chosengender, imagetype=imagetype, imagemodechance=imagemodechance, givensubject=givensubject, smartsubject=smartsubject, overrideoutfit=givenoutfit, prefixprompt=prefixprompt, suffixprompt=suffixprompt, giventypeofimage=giventypeofimage, antivalues=antistring, advancedprompting=False, hardturnoffemojis=True, seed=seed, base_model=base_model, OBP_preset=OneButtonPreset, prompt_enhancer=prompt_enhancer, preset_prefix=preset_prefix, preset_suffix=preset_suffix, ) return (generatedprompt,) class AutoNegativePrompt: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "postive_prompt": ("STRING", {"default": '', "multiline": True}), }, "optional": { "base_negative": ("STRING", { "multiline": True, #True if you want the field to look like the one on the ClipTextEncode node "default": "text, watermark" }), "enhancenegative": ("INT", { "default": 0, "min": 0, #Minimum value "max": 1, #Maximum value "step": 1, #Slider's step }), "insanitylevel": ("INT", { "default": 0, "min": 0, #Minimum value "max": 10, #Maximum value "step": 1 #Slider's step }), "base_model":(models, {"default": "SDXL"}), "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("negative_prompt",) FUNCTION = "Comfy_OBP_AutoNegativePrompt" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_AutoNegativePrompt(self, postive_prompt, insanitylevel, enhancenegative,base_negative, seed, base_model): generatedprompt = build_dynamic_negative(postive_prompt, insanitylevel, enhancenegative, base_negative, base_model=base_model) print("Generated negative prompt: " + generatedprompt) return (generatedprompt,) class OneButtonArtify: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "prompt": ("STRING", {"default": '', "multiline": True}), "artist": (artifyartists, {"default": "all"}), "amount_of_artists": (artifyamountofartistslist, {"default": "1"}), "artify_mode": (artifymodeslist, {"default": "standard"}) }, "optional": { "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("artified_prompt",) FUNCTION = "Comfy_OBP_Artify" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_Artify(self, prompt, artist, amount_of_artists,artify_mode, seed): # artify here artified_prompt = artify_prompt(prompt=prompt, artists=artist, amountofartists=amount_of_artists, mode=artify_mode, seed=seed) print("Artified prompt: " + artified_prompt) return (artified_prompt,) class OneButtonFlufferize: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "prompt": ("STRING", {"default": '', "multiline": True}), "amount_of_fluff": (amountofflufflist, {"default": "dynamic"}), "reverse_polarity": (fluff_reverse_polarity, {"default": False}), }, "optional": { "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("fluffed_prompt",) FUNCTION = "Comfy_OBP_Flufferize" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_Flufferize(self, prompt, amount_of_fluff, reverse_polarity, seed): # artify here fluffed_prompt = flufferizer(prompt=prompt, amountoffluff=amount_of_fluff, reverse_polarity=reverse_polarity, seed=seed) print("Fluffed prompt: " + fluffed_prompt) return (fluffed_prompt,) class OneButtonSuperPrompt: def __init__(self): pass @classmethod def INPUT_TYPES(s): return { "required": { "prompt": ("STRING", {"default": '', "multiline": True}), "insanitylevel": ("INT", { "default": 5, "min": 1, #Minimum value "max": 10, #Maximum value "step": 1 #Slider's step }), "superpromptstyle": (superprompterstyleslist, {"default": "all"}), }, "optional": { "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}), }, } RETURN_TYPES = ("STRING",) RETURN_NAMES = ("super_prompt",) FUNCTION = "Comfy_OBP_SuperPrompt" #OUTPUT_NODE = False CATEGORY = "OneButtonPrompt" def Comfy_OBP_SuperPrompt(self, insanitylevel, prompt, superpromptstyle, seed): OBPsuperprompt = one_button_superprompt(insanitylevel=insanitylevel, prompt=prompt, seed=seed, superpromptstyle=superpromptstyle) print("Super prompt: " + OBPsuperprompt) return (OBPsuperprompt,) # A dictionary that contains all nodes you want to export with their names # NOTE: names should be globally unique NODE_CLASS_MAPPINGS = { "OneButtonPrompt": OneButtonPrompt, "OneButtonPreset": OneButtonPreset, "OneButtonArtify": OneButtonArtify, "CreatePromptVariant": CreatePromptVariant, "SavePromptToFile": SavePromptToFile, "AutoNegativePrompt": AutoNegativePrompt, "OneButtonFlufferize": OneButtonFlufferize, "OneButtonSuperPrompt": OneButtonSuperPrompt, } # A dictionary that contains the friendly/humanly readable titles for the nodes NODE_DISPLAY_NAME_MAPPINGS = { "OneButtonPrompt": "One Button Prompt", "OneButtonPreset": "One Button Preset", "OneButtonArtify": "One Button Artify", "CreatePromptVariant": "Create Prompt Variant", "SavePromptToFile": "Save Prompt To File", "AutoNegativePrompt": "Auto Negative Prompt", "OneButtonFlufferize": "One Button Flufferize", "OneButtonSuperPrompt": "One Button SuperPrompt", }