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

fix gallery disappearing while next queued item generating

Browse Source
This commit is contained in:
Chris Malone 2025-03-30 06:03:45 +11:00
parent 519b38493d
commit 6fc7f762a1
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gradio_server.py
View File

@ -1176,7 +1176,12 @@ def build_callback(taskid, state, pipe, num_inference_steps, repeats):
return update_progress return update_progress
def refresh_gallery(state): def refresh_gallery(state):
return state return gr.update(value=state.get("file_list", []))
def refresh_gallery_on_trigger(state):
if(state.get("update_gallery", False)):
state['update_gallery'] = False
return gr.update(value=state.get("file_list", []))
def finalize_gallery(state): def finalize_gallery(state):
choice = 0 choice = 0
@ -2318,10 +2323,6 @@ def generate_video_tab(image2video=False):
return gr.update(), gr.update(value=image_data_to_show), gr.update(visible=True) return gr.update(), gr.update(value=image_data_to_show), gr.update(visible=True)
else: else:
return gr.update(), gr.update(), gr.update(visible=False) return gr.update(), gr.update(), gr.update(visible=False)
def refresh_gallery_on_trigger(state):
if(state.get("update_gallery", False)):
state['update_gallery'] = False
return gr.update(value=state.get("file_list", []))
selected_indices = gr.State([]) selected_indices = gr.State([])
queue_df.select( queue_df.select(
fn=handle_selection, fn=handle_selection,