Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 06:15:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added cancel button, removed generate button on edit page

Browse Source
This commit is contained in:
Chris Malone 2025-09-13 19:25:01 +10:00
parent f18e3daaa3
commit fda245b05e
1 changed files with 84 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
wgp.py
View File

@ -196,6 +196,11 @@ def clean_image_list(gradio_list):
gradio_list = [ convert_image( Image.open(img) if isinstance(img, str) else img ) for img in gradio_list ] gradio_list = [ convert_image( Image.open(img) if isinstance(img, str) else img ) for img in gradio_list ]
return gradio_list return gradio_list
def cancel_edit(state):
state["editing_task_index"] = None
gr.Info("Edit cancelled.")
return gr.Tabs(selected="video_gen")
def edit_task_in_queue( def edit_task_in_queue(
lset_name, lset_name,
image_mode, image_mode,
@ -8215,12 +8220,10 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
if not update_form: if not update_form:
if tab_id == 'edit': if tab_id == 'edit':
edit_btn = gr.Button("Edit") edit_btn = gr.Button("Edit")
generate_btn = gr.Button("Generate", visible=False) cancel_btn = gr.Button("Cancel")
add_to_queue_btn = gr.Button("Add New Prompt To Queue", visible=False)
else: else:
generate_btn = gr.Button("Generate") generate_btn = gr.Button("Generate")
add_to_queue_btn = gr.Button("Add New Prompt To Queue", visible=False) add_to_queue_btn = gr.Button("Add New Prompt To Queue", visible=False)
edit_btn = gr.Button("Save Edit", visible=False)
generate_trigger = gr.Text(visible = False) generate_trigger = gr.Text(visible = False)
add_to_queue_trigger = gr.Text(visible = False) add_to_queue_trigger = gr.Text(visible = False)
@ -8338,6 +8341,7 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
status_trigger.change(refresh_status_async, inputs= [state] , outputs= [gen_status], show_progress_on= [gen_status]) status_trigger.change(refresh_status_async, inputs= [state] , outputs= [gen_status], show_progress_on= [gen_status])
if tab_id == 'generate':
output_trigger.change(refresh_gallery, output_trigger.change(refresh_gallery,
inputs = [state], inputs = [state],
outputs = [output, gen_info, generate_btn, add_to_queue_btn, current_gen_column, current_gen_buttons_row, queue_df, abort_btn, onemorewindow_btn], outputs = [output, gen_info, generate_btn, add_to_queue_btn, current_gen_column, current_gen_buttons_row, queue_df, abort_btn, onemorewindow_btn],
@ -8466,6 +8470,11 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
inputs=edit_inputs_components + [state], inputs=edit_inputs_components + [state],
outputs=[queue_df, main_tabs] outputs=[queue_df, main_tabs]
) )
cancel_btn.click(
fn=cancel_edit,
inputs=[state],
outputs=[main_tabs]
)
refresh_form_trigger.change(fn= fill_inputs, refresh_form_trigger.change(fn= fill_inputs,
inputs=[state], inputs=[state],