Fix some events

This commit is contained in:
oobabooga 2025-06-18 10:27:38 -07:00
parent 8b7eb5c87c
commit 7cb2b1bfdb
2 changed files with 9 additions and 3 deletions

View file

@ -1918,10 +1918,16 @@ def handle_send_instruction_click(state):
output = generate_chat_prompt("Input", state)
return output
if state["show_two_notebook_columns"]:
return gr.update(), output, ""
else:
return output, gr.update(), gr.update()
def handle_send_chat_click(state):
output = generate_chat_prompt("", state, _continue=True)
return output, output, ""
if state["show_two_notebook_columns"]:
return gr.update(), output, ""
else:
return output, gr.update(), gr.update()

View file

@ -344,7 +344,7 @@ def create_event_handlers():
shared.gradio['send_instruction_to_notebook'].click(
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(
chat.handle_send_instruction_click, gradio('interface_state'), gradio('textbox-notebook'), show_progress=False).then(
chat.handle_send_instruction_click, gradio('interface_state'), gradio('textbox-notebook', 'textbox-default', 'output_textbox'), show_progress=False).then(
None, None, None, js=f'() => {{{ui.switch_tabs_js}; switch_to_notebook()}}')
shared.gradio['send-chat-to-notebook'].click(