From 6fc7f762a19cd51ce1fca1303dd800041da7ffa2 Mon Sep 17 00:00:00 2001 From: Chris Malone Date: Sun, 30 Mar 2025 06:03:45 +1100 Subject: [PATCH] fix gallery disappearing while next queued item generating --- gradio_server.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/gradio_server.py b/gradio_server.py index f712ef1..caee14b 100644 --- a/gradio_server.py +++ b/gradio_server.py @@ -1176,7 +1176,12 @@ def build_callback(taskid, state, pipe, num_inference_steps, repeats): return update_progress 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): 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) else: 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([]) queue_df.select( fn=handle_selection,