| import json | |
| import os | |
| import re | |
| from internals.data.dataAccessor import getCharacters | |
| from internals.util.config import root_dir | |
| class Avatar: | |
| __avatars = {} | |
| def load_local(self, model_dir): | |
| self.__find_available_characters(model_dir) | |
| if len(self.__avatars.items()) > 0: | |
| print("Local characters", self.__avatars) | |
| def fetch_from_network(self, model_id: int): | |
| if not model_id: | |
| return | |
| characters = getCharacters(str(model_id)) | |
| if characters is not None: | |
| for character in characters: | |
| item = { | |
| "avatarName": str(character["title"]).lower(), | |
| "codename": character["tag"], | |
| "extraPrompt": character["extraData"]["extraPrompt"], | |
| } | |
| self.__avatars[item["avatarName"]] = item | |
| def add_code_names(self, prompt): | |
| array_of_objects = self.__avatars.values() | |
| for obj in array_of_objects: | |
| prompt = ( | |
| re.sub( | |
| r"\b" + obj["avatarName"] + r"\b", | |
| obj["extraPrompt"], | |
| prompt, | |
| flags=re.IGNORECASE, | |
| ) | |
| + " " | |
| ) | |
| print(prompt) | |
| return prompt | |
| def __find_available_characters(self, path: str): | |
| if os.path.exists(path + "/characters.json"): | |
| print(path) | |
| try: | |
| print("Loading characters") | |
| with open(path + "/characters.json") as f: | |
| data = json.load(f) | |
| print("Characters: ", data) | |
| if "avatarName" in data[0]: | |
| for item in data: | |
| self.__avatars[item["avatarName"]] = item | |
| print("Avatars", self.__avatars) | |
| else: | |
| print("Invalid characters.json file") | |
| except Exception as e: | |
| print("Error Loading characters", e) | |