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

Add "remove last message" button to chat

Browse Source
This commit is contained in:
oobabooga 2023-01-15 03:19:09 -03:00
parent fd220f827f
commit 13b04c1b94
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server.py
View File

@ -223,6 +223,10 @@ elif args.chat:
history.append((text, reply)) history.append((text, reply))
return history return history
def remove_last_message():
history.pop()
return history
def clear(): def clear():
global history global history
history = [] history = []
@ -257,10 +261,15 @@ elif args.chat:
display1 = gr.Chatbot() display1 = gr.Chatbot()
textbox = gr.Textbox(lines=2, label='Input') textbox = gr.Textbox(lines=2, label='Input')
btn = gr.Button("Generate") btn = gr.Button("Generate")
btn2 = gr.Button("Clear history") with gr.Row():
with gr.Column():
btn3 = gr.Button("Remove last message")
with gr.Column():
btn2 = gr.Button("Clear history")
btn.click(chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True, api_name="textgen") btn.click(chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True, api_name="textgen")
textbox.submit(chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True) textbox.submit(chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True)
btn3.click(remove_last_message, [], display1, show_progress=False)
btn2.click(clear) btn2.click(clear)
btn.click(lambda x: "", textbox, textbox, show_progress=False) btn.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)