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

Add "regenerate" button to the chat

Browse Source
This commit is contained in:
oobabooga 2023-01-22 02:19:58 -03:00
parent 5b50db9dee
commit 62b533f344
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
server.py
View File

@ -322,6 +322,16 @@ if args.chat or args.cai_chat:
for history in chatbot_wrapper(text, tokens, inference_settings, selected_model, name1, name2, context, check, history_size): for history in chatbot_wrapper(text, tokens, inference_settings, selected_model, name1, name2, context, check, history_size):
yield generate_chat_html(history, name1, name2, character) yield generate_chat_html(history, name1, name2, character)
def regenerate_wrapper(text, tokens, inference_settings, selected_model, name1, name2, context, check, history_size):
last = history.pop()
text = last[0]
if args.cai_chat:
for i in cai_chatbot_wrapper(text, tokens, inference_settings, selected_model, name1, name2, context, check, history_size):
yield i
else:
for i in chatbot_wrapper(text, tokens, inference_settings, selected_model, name1, name2, context, check, history_size):
yield i
def remove_last_message(name1, name2): def remove_last_message(name1, name2):
history.pop() history.pop()
if args.cai_chat: if args.cai_chat:
@ -411,9 +421,10 @@ if args.chat or args.cai_chat:
textbox = gr.Textbox(lines=2, label='Input') textbox = gr.Textbox(lines=2, label='Input')
btn = gr.Button("Generate") btn = gr.Button("Generate")
with gr.Row(): with gr.Row():
btn2 = gr.Button("Clear history")
stop = gr.Button("Stop") stop = gr.Button("Stop")
btn3 = gr.Button("Remove last message") btn_regenerate = gr.Button("Regenerate")
btn_remove_last = gr.Button("Remove last")
btn_clear = gr.Button("Clear history")
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
@ -447,17 +458,19 @@ if args.chat or args.cai_chat:
if args.cai_chat: if args.cai_chat:
gen_event = btn.click(cai_chatbot_wrapper, input_params, display1, show_progress=args.no_stream, api_name="textgen") gen_event = btn.click(cai_chatbot_wrapper, input_params, display1, show_progress=args.no_stream, api_name="textgen")
gen_event2 = textbox.submit(cai_chatbot_wrapper, input_params, display1, show_progress=args.no_stream) gen_event2 = textbox.submit(cai_chatbot_wrapper, input_params, display1, show_progress=args.no_stream)
btn2.click(clear_html, [], display1, show_progress=False) btn_clear.click(clear_html, [], display1, show_progress=False)
else: else:
gen_event = btn.click(chatbot_wrapper, input_params, display1, show_progress=args.no_stream, api_name="textgen") gen_event = btn.click(chatbot_wrapper, input_params, display1, show_progress=args.no_stream, api_name="textgen")
gen_event2 = textbox.submit(chatbot_wrapper, input_params, display1, show_progress=args.no_stream) gen_event2 = textbox.submit(chatbot_wrapper, input_params, display1, show_progress=args.no_stream)
btn2.click(lambda x: "", display1, display1, show_progress=False) btn_clear.click(lambda x: "", display1, display1, show_progress=False)
gen_event3 = btn_regenerate.click(regenerate_wrapper, input_params, display1, show_progress=args.no_stream)
btn2.click(clear) btn_clear.click(clear)
btn3.click(remove_last_message, [name1, name2], display1, show_progress=False) btn_remove_last.click(remove_last_message, [name1, name2], display1, show_progress=False)
btn.click(lambda x: "", textbox, textbox, show_progress=False) btn.click(lambda x: "", textbox, textbox, show_progress=False)
btn_regenerate.click(lambda x: "", textbox, textbox, show_progress=False)
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, gen_event3])
save_btn.click(save_history, inputs=[], outputs=[download]) save_btn.click(save_history, inputs=[], outputs=[download])
upload.upload(load_history, [upload], []) upload.upload(load_history, [upload], [])
character_menu.change(load_character, [character_menu, name1, name2], [name2, context, display1]) character_menu.change(load_character, [character_menu, name1, name2], [name2, context, display1])