This commit is contained in:
lllyasviel
2023-10-17 00:10:32 -07:00
parent ef4e120d0a
commit e3ac360396
2 changed files with 3 additions and 8 deletions
+1 -1
View File
@@ -1 +1 @@
version = '2.1.689' version = '2.1.690'
+2 -7
View File
@@ -1,8 +1,6 @@
import os import os
import json import json
import re import re
import random
from modules.util import get_files_from_folder from modules.util import get_files_from_folder
@@ -121,11 +119,8 @@ def apply_wildcards(wildcard_text, seed=None, directory=wildcards_path):
placeholders = re.findall(r'__(\w+)__', wildcard_text) placeholders = re.findall(r'__(\w+)__', wildcard_text)
for placeholder in placeholders: for placeholder in placeholders:
try: try:
with open(os.path.join(directory, f'{placeholder}.txt'), encoding='utf-8') as f: words = open(os.path.join(directory, f'{placeholder}.txt'), encoding='utf-8').read().splitlines()
words = f.read().splitlines() wildcard_text = wildcard_text.replace(f'__{placeholder}__', words[int(seed) % len(words)])
rng = random.Random(seed)
wildcard_text = re.sub(rf'__{placeholder}__', rng.choice(words), wildcard_text)
except IOError: except IOError:
print(f'Error: could not open wildcard file {placeholder}.txt, using as normal word.') print(f'Error: could not open wildcard file {placeholder}.txt, using as normal word.')
wildcard_text = wildcard_text.replace(f'__{placeholder}__', placeholder) wildcard_text = wildcard_text.replace(f'__{placeholder}__', placeholder)