Refresh current task fix

This commit is contained in:
DeepBeepMeep 2025-04-17 13:17:49 +02:00
parent 30f2889a35
commit 5e38aa5c67

4
wgp.py
View File

@ -2818,6 +2818,8 @@ def process_tasks(state, progress=gr.Progress()):
gen_in_progress = True gen_in_progress = True
gen["in_progress"] = True gen["in_progress"] = True
yield "Generating Video"
prompt_no = 0 prompt_no = 0
while len(queue) > 0: while len(queue) > 0:
prompt_no += 1 prompt_no += 1
@ -4132,7 +4134,7 @@ def generate_video_tab(update_form = False, state_dict = None, ui_defaults = Non
outputs= queue_df outputs= queue_df
).then(fn=prepare_generate_video, ).then(fn=prepare_generate_video,
inputs= [state], inputs= [state],
outputs= [generate_btn, add_to_queue_btn, current_gen_column], outputs= [generate_btn, add_to_queue_btn, current_gen_column]
).then(fn=process_tasks, ).then(fn=process_tasks,
inputs= [state], inputs= [state],
outputs= [gen_status], outputs= [gen_status],