Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Save TavernAI characters with TavernAI- prefix

Browse Source
This commit is contained in:
oobabooga 2023-01-28 21:01:56 -03:00
parent 833a1138fa
commit e779fd795f
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.py
View File

@ -515,7 +515,7 @@ if args.chat or args.cai_chat:
else: else:
return name2, context, history['visible'] return name2, context, history['visible']
def upload_character(json_file, img): def upload_character(json_file, img, tavern=False):
json_file = json_file if type(json_file) == str else json_file.decode('utf-8') json_file = json_file if type(json_file) == str else json_file.decode('utf-8')
data = json.loads(json_file) data = json.loads(json_file)
outfile_name = data["char_name"] outfile_name = data["char_name"]
@ -523,6 +523,8 @@ if args.chat or args.cai_chat:
while Path(f'characters/{outfile_name}.json').exists(): while Path(f'characters/{outfile_name}.json').exists():
outfile_name = f'{data["char_name"]}_{i:03d}' outfile_name = f'{data["char_name"]}_{i:03d}'
i += 1 i += 1
if tavern:
outfile_name = f'TavernAI-{outfile_name}'
with open(Path(f'characters/{outfile_name}.json'), 'w') as f: with open(Path(f'characters/{outfile_name}.json'), 'w') as f:
f.write(json_file) f.write(json_file)
if img is not None: if img is not None:
@ -538,7 +540,7 @@ if args.chat or args.cai_chat:
_json = json.loads(decoded_string) _json = json.loads(decoded_string)
_json = {"char_name": _json['name'], "char_persona": _json['description'], "char_greeting": _json["first_mes"], "example_dialogue": _json['mes_example'], "world_scenario": _json['scenario']} _json = {"char_name": _json['name'], "char_persona": _json['description'], "char_greeting": _json["first_mes"], "example_dialogue": _json['mes_example'], "world_scenario": _json['scenario']}
_json['example_dialogue'] = _json['example_dialogue'].replace('{{user}}', name1).replace('{{char}}', _json['char_name']) _json['example_dialogue'] = _json['example_dialogue'].replace('{{user}}', name1).replace('{{char}}', _json['char_name'])
return upload_character(json.dumps(_json), img) return upload_character(json.dumps(_json), img, tavern=True)
def upload_your_profile_picture(img): def upload_your_profile_picture(img):
img = Image.open(io.BytesIO(img)) img = Image.open(io.BytesIO(img))