| from PIL import Image |
| from bw_utils import get_child_paths,lang_detect,is_image,decode_base64,save_json_file,create_dir |
| import PIL.PngImagePlugin |
| import os |
| import json |
|
|
| card_dir = "./data/sillytavern_cards" |
| names = os.listdir(card_dir) |
|
|
| for name in names: |
| path = os.path.join(card_dir, name) |
| role_code = name.split('.')[0].replace(" ","_") |
| if is_image(path): |
| with open(path, 'rb',encoding="utf-8") as f: |
| image = Image.open(f) |
| card_info = json.loads(decode_base64(image.text['chara'])) |
| language = lang_detect(card_info['data']['description']) |
| role_info = { |
| "role_code": f"{role_code}-{language}", |
| "role_name": card_info['data']['name'], |
| "source": "", |
| "profile": card_info['data']['description'], |
| "nickname": card_info['data']['name'], |
| "relation": {}, |
| "card_data": card_info['data'] |
| } |
| create_dir(f"./data/roles/sillytavern/{role_code}") |
| save_json_file(os.path.join(f"./data/roles/sillytavern/{role_code}","role_info.json"),role_info) |
| image.save(os.path.join(f"./data/roles/sillytavern/{role_code}",f"icon.png")) |
| print(f"{name} converted successfully.") |
| |
