diff --git a/build_dynamic_prompt.py b/build_dynamic_prompt.py index 29808ad..b372e9f 100644 --- a/build_dynamic_prompt.py +++ b/build_dynamic_prompt.py @@ -3220,7 +3220,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all completeprompt = replacewildcard(completeprompt, insanitylevel, wildcard, attachedlist,True, advancedprompting, artiststyleselector) - + completeprompt = replace_user_wildcards(completeprompt) # prompt strenght stuff # if the given subject already is formed like this ( :1.x) @@ -4933,4 +4933,14 @@ def one_button_superprompt(insanitylevel = 5, prompt = "", seed = -1, override_s - return superpromptresult \ No newline at end of file + return superpromptresult + +def replace_user_wildcards(completeprompt): + user_wildcards_list = re.findall(r'-[\w_]*-', completeprompt) + for user_wildcard in user_wildcards_list: + user_wildcard_clean = user_wildcard.strip("-") + wordlist = csv_to_list(csvfilename=user_wildcard_clean, directory="./userfiles/") + if(wordlist): + completeprompt = completeprompt.replace(user_wildcard, random.choice(wordlist),1) + + return completeprompt \ No newline at end of file diff --git a/prompttester.py b/prompttester.py index 19a7080..f489749 100644 --- a/prompttester.py +++ b/prompttester.py @@ -130,5 +130,5 @@ if __name__ == "__main__": , False #prompt_g_and_l , "SDXL" #base model , "" #preset "All (random)..." - , "superprompter" # superprompter + , "" # superprompter ) \ No newline at end of file