Update run handlers to also update Gallery tab after generation

ui/events/run_handlers.py

@@ -96,8 +96,12 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
     run_btn = ui_components.get(f'run_{prefix}') or ui_components.get(f'{prefix}_run_button')
     res_gal = ui_components.get(f'result_{prefix}') or ui_components.get(f'{prefix}_output_gallery')
     if run_btn and res_gal:
+        outputs = [res_gal]
+        gallery_tab = ui_components.get('gallery')
+        if gallery_tab:
+            outputs.append(gallery_tab)
         run_btn.click(
             fn=lambda *args, progress=gr.Progress(track_tqdm=True): generate_image_wrapper(create_ui_inputs_dict(*args), progress),
             inputs=input_list_flat,
-            outputs=[res_gal]
+            outputs=outputs
         )