nodes/sn0w12-ComfyUI-Sn0w-Scripts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix random character not working if not favourite

Browse Source
This commit is contained in:
Lucas Sköld
2024-06-19 15:59:54 +02:00
parent e381a4b9ed
commit a263219d21
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/character_select.py
+2 -6
View File
@@ -111,16 +111,12 @@ class CharacterSelectNode:
return ("", "",)
def sanitize_name(self, name):
# Sanitize the name to match the format used in the exclusion list
return name.lower().replace(" ", "_").replace("(", "_").replace(")", "_").replace("-", "_").replace(",", "_").replace("!", "_").replace(".", "_")
def select_random_character(self):
# Fetching the exclusion settings
exclude_characters = ConfigReader.get_setting("sn0w.ExcludedRandomCharacters", False)
if (exclude_characters == False):
random_character_name = random.choice(list(filtered_characters.keys()))
char_item = filtered_characters[random_character_name]
random_character_name = random.choice(list(self.final_character_dict.keys()))
char_item = self.final_character_dict[random_character_name]
self.logger.log("Random Character: " + str(char_item["name"]), "INFORMATIONAL")
return char_item