Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2026-01-11 16:53:34 +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 ]
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(
lset_name,
image_mode,
@ -8215,12 +8220,10 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
if not update_form:
if tab_id == 'edit':
edit_btn = gr.Button("Edit")
generate_btn = gr.Button("Generate", visible=False)
add_to_queue_btn = gr.Button("Add New Prompt To Queue", visible=False)
cancel_btn = gr.Button("Cancel")
else:
generate_btn = gr.Button("Generate")
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)
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])
if tab_id == 'generate':
output_trigger.change(refresh_gallery,
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],
@ -8466,6 +8470,11 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
inputs=edit_inputs_components + [state],
outputs=[queue_df, main_tabs]
)
cancel_btn.click(
fn=cancel_edit,
inputs=[state],
outputs=[main_tabs]
)
refresh_form_trigger.change(fn= fill_inputs,
inputs=[state],