nodes/AIrjen-OneButtonPrompt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

overrides added in

Browse Source 
some final improvements
This commit is contained in:
AIrjen
2024-01-01 15:52:31 +01:00
parent 425468534b
commit 8d6747b64b
4 changed files with 32 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
csv_reader.py
+26
View File
@@ -162,11 +162,24 @@ def load_config_csv():
def load_negative_list():
primerlist = []
negativelist = []
primeraddonlist = []
negativeaddonlist = []
script_dir = os.path.dirname(os.path.abspath(__file__))
full_path_default_negative_file = os.path.join(script_dir, "./csvfiles/special_lists/" )
negative_file = full_path_default_negative_file + 'negativewords.csv'
full_path_replace_negative_file = os.path.join(script_dir, "./userfiles/" )
replace_negative_file = full_path_replace_negative_file + 'negativewords_replace.csv'
full_path_addon_negative_file = os.path.join(script_dir, "./userfiles/" )
addon_negative_file = full_path_addon_negative_file + 'negativewords_addon.csv'
if(os.path.isfile(replace_negative_file)):
negative_file = replace_negative_file
with open(negative_file, "r", newline="",encoding="utf8") as file:
reader = csv.DictReader(file, delimiter=";")
primerlist = [row["primer"] for row in reader]
@@ -174,6 +187,19 @@ def load_negative_list():
reader = csv.DictReader(file, delimiter=";")
negativelist = [row["negative"] for row in reader]
if(os.path.isfile(addon_negative_file)):
with open(addon_negative_file, "r", newline="",encoding="utf8") as file:
reader = csv.DictReader(file, delimiter=";")
primeraddonlist = [row["primer"] for row in reader]
with open(addon_negative_file, "r", newline="",encoding="utf8") as file:
reader = csv.DictReader(file, delimiter=";")
negativeaddonlist = [row["negative"] for row in reader]
primerlist += primeraddonlist
negativelist += negativeaddonlist
return primerlist, negativelist
def load_all_artist_and_category():