added delete last turn button and logic
Browse filesadded the button, the click event and the inference function for deleting the last turn
app.py
CHANGED
@@ -192,7 +192,13 @@ examples = [
|
|
192 |
def clear_chat():
|
193 |
return [], []
|
194 |
|
195 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
196 |
def process_example(args):
|
197 |
for [x, y] in generate(args):
|
198 |
pass
|
@@ -262,6 +268,7 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
|
|
262 |
send_button = gr.Button("Send", elem_id="send-btn", visible=True)
|
263 |
|
264 |
# regenerate_button = gr.Button("Regenerate", elem_id="send-btn", visible=True)
|
|
|
265 |
|
266 |
clear_chat_button = gr.Button("Clear chat", elem_id="clear-btn", visible=True)
|
267 |
|
@@ -364,6 +371,7 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
|
|
364 |
outputs=[chatbot, history, last_user_message, user_message],
|
365 |
)
|
366 |
|
|
|
367 |
clear_chat_button.click(clear_chat, outputs=[chatbot, history])
|
368 |
selected_model.change(clear_chat, outputs=[chatbot, history])
|
369 |
# share_button.click(None, [], [], _js=share_js)
|
|
|
192 |
def clear_chat():
|
193 |
return [], []
|
194 |
|
195 |
+
def delete_last_turn(chat, history):
|
196 |
+
if chat and history:
|
197 |
+
chat.pop(-1)
|
198 |
+
history.pop(-1)
|
199 |
+
history.pop(-1)
|
200 |
+
return chat, history
|
201 |
+
|
202 |
def process_example(args):
|
203 |
for [x, y] in generate(args):
|
204 |
pass
|
|
|
268 |
send_button = gr.Button("Send", elem_id="send-btn", visible=True)
|
269 |
|
270 |
# regenerate_button = gr.Button("Regenerate", elem_id="send-btn", visible=True)
|
271 |
+
delete_turn_button = gr.Button("Delete last turn", elem_id="delete-btn", visible=True)
|
272 |
|
273 |
clear_chat_button = gr.Button("Clear chat", elem_id="clear-btn", visible=True)
|
274 |
|
|
|
371 |
outputs=[chatbot, history, last_user_message, user_message],
|
372 |
)
|
373 |
|
374 |
+
delete_turn_button.click(delete_last_turn, [chatbot, history], [chatbot, history])
|
375 |
clear_chat_button.click(clear_chat, outputs=[chatbot, history])
|
376 |
selected_model.change(clear_chat, outputs=[chatbot, history])
|
377 |
# share_button.click(None, [], [], _js=share_js)
|