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

Implement saving/loading chat logs (#9)

Browse Source
This commit is contained in:
oobabooga 2023-01-19 14:03:47 -03:00
parent 39bfea5a22
commit d03b0ad7a8
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
server.py
View File

@ -296,6 +296,21 @@ if args.chat or args.cai_chat:
def clear_html(): def clear_html():
return generate_chat_html([], "", "") return generate_chat_html([], "", "")
def redraw_html(name1, name2):
global history
return generate_chat_html(history, name1, name2)
def save_history():
if not Path('logs').exists():
Path('logs').mkdir()
with open(Path('logs/conversation.json'), 'w') as f:
f.write(json.dumps({'data': history}))
return Path('logs/conversation.json')
def load_history(file):
global history
history = json.loads(file.decode('utf-8'))['data']
if 'pygmalion' in model_name.lower(): if 'pygmalion' in model_name.lower():
context_str = settings['context_pygmalion'] context_str = settings['context_pygmalion']
name1_str = settings['name1_pygmalion'] name1_str = settings['name1_pygmalion']
@ -329,6 +344,14 @@ if args.chat or args.cai_chat:
context = gr.Textbox(value=context_str, lines=2, label='Context') context = gr.Textbox(value=context_str, lines=2, label='Context')
with gr.Row(): with gr.Row():
check = gr.Checkbox(value=settings['stop_at_newline'], label='Stop generating at new line character?') check = gr.Checkbox(value=settings['stop_at_newline'], label='Stop generating at new line character?')
with gr.Row():
with gr.Column():
gr.Markdown("Upload chat history")
upload = gr.File(type='binary')
with gr.Column():
gr.Markdown("Download chat history")
save_btn = gr.Button(value="Click me")
download = gr.File()
if args.cai_chat: if args.cai_chat:
gen_event = btn.click(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=args.no_stream, api_name="textgen") gen_event = btn.click(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=args.no_stream, api_name="textgen")
@ -345,6 +368,11 @@ 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]) stop.click(None, None, None, cancels=[gen_event, gen_event2])
save_btn.click(save_history, inputs=[], outputs=[download])
upload.upload(load_history, [upload], [])
upload.upload(redraw_html, [name1, name2], [display1])
elif args.notebook: elif args.notebook:
with gr.Blocks(css=css, analytics_enabled=False) as interface: with gr.Blocks(css=css, analytics_enabled=False) as interface:
gr.Markdown(description) gr.Markdown(description)