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

Allow uploading characters

Browse Source
This commit is contained in:
oobabooga 2023-01-25 15:45:25 -03:00
parent 6388c7fbc0
commit 7a3717b824
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
server.py
View File

@ -408,7 +408,7 @@ if args.chat or args.cai_chat:
f.write(json.dumps({'data': history}, indent=2)) f.write(json.dumps({'data': history}, indent=2))
return Path('logs/conversation.json') return Path('logs/conversation.json')
def load_history(file, name1, name2): def upload_history(file, name1, name2):
global history global history
file = file.decode('utf-8') file = file.decode('utf-8')
try: try:
@ -456,6 +456,20 @@ if args.chat or args.cai_chat:
else: else:
return name2, context, _history return name2, context, _history
def upload_character(file, name1, name2):
global history
file = file.decode('utf-8')
data = json.loads(file)
outfile_name = data["char_name"]
i = 1
while Path(f'characters/{outfile_name}.json').exists():
outfile_name = f'{data["char_name"]}_{i:03d}'
i += 1
with open(Path(f'characters/{outfile_name}.json'), 'w') as f:
f.write(file)
print(f'New character saved to "characters/{outfile_name}.json".')
return outfile_name
suffix = '_pygmalion' if 'pygmalion' in model_name.lower() else '' suffix = '_pygmalion' if 'pygmalion' in model_name.lower() else ''
with gr.Blocks(css=css+".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto}", analytics_enabled=False) as interface: with gr.Blocks(css=css+".h-\[40vh\] {height: 66.67vh} .gradio-container {max-width: 800px; margin-left: auto; margin-right: auto}", analytics_enabled=False) as interface:
if args.cai_chat: if args.cai_chat:
@ -497,6 +511,8 @@ if args.chat or args.cai_chat:
save_btn = gr.Button(value="Click me") save_btn = gr.Button(value="Click me")
with gr.Tab('Upload chat history'): with gr.Tab('Upload chat history'):
upload = gr.File(type='binary') upload = gr.File(type='binary')
with gr.Tab('Upload character'):
upload_char = gr.File(type='binary')
input_params = [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check, history_size_slider] input_params = [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check, history_size_slider]
if args.cai_chat: if args.cai_chat:
@ -514,8 +530,9 @@ if args.chat or args.cai_chat:
textbox.submit(lambda x: "", textbox, textbox, show_progress=False) textbox.submit(lambda x: "", textbox, textbox, show_progress=False)
stop.click(None, None, None, cancels=[gen_event, gen_event2, gen_event3]) stop.click(None, None, None, cancels=[gen_event, gen_event2, gen_event3])
save_btn.click(save_history, inputs=[], outputs=[download]) save_btn.click(save_history, inputs=[], outputs=[download])
upload.upload(load_history, [upload, name1, name2], [])
character_menu.change(load_character, [character_menu, name1, name2], [name2, context, display1]) character_menu.change(load_character, [character_menu, name1, name2], [name2, context, display1])
upload.upload(upload_history, [upload, name1, name2], [])
upload_char.upload(upload_character, [upload_char, name1, name2], [character_menu])
if args.cai_chat: if args.cai_chat:
upload.upload(redraw_html, [name1, name2], [display1]) upload.upload(redraw_html, [name1, name2], [display1])