Refresh global gallery after generation

ui/events/run_handlers.py

@@ -94,11 +94,33 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
 
         return ui_dict
 
-    # Helper to run generation and return only the image paths (metadata handled separately in Gallery)
-    def _run_with_meta(*args, progress=gr.Progress(track_tqdm=True)):
+    # Helper to run generation and refresh the global Gallery.
+    def _run_and_refresh(*args, progress=gr.Progress(track_tqdm=True)):
+        """Run the generation pipeline, then return both the tab-specific results
+        and the updated list of images for the Global Gallery.
+        """
         ui_dict = create_ui_inputs_dict(*args)
         results = generate_image_wrapper(ui_dict, progress)
-        return results
+        # Re-scan the output folder for all image files (same logic as layout.load_gallery_images)
+        try:
+            from core.settings import OUTPUT_DIR
+            import os
+            out_dir = os.path.abspath(OUTPUT_DIR)
+            seen = set()
+            gallery_paths = []
+            if os.path.isdir(out_dir):
+                for fname in sorted(os.listdir(out_dir)):
+                    low = fname.lower()
+                    if low.endswith((".png", ".jpg", ".jpeg", ".gif", ".webp")):
+                        base = os.path.splitext(fname)[0]
+                        if base not in seen:
+                            seen.add(base)
+                            gallery_paths.append(os.path.join(out_dir, fname))
+        except Exception as e:
+            print(f"[DEBUG] Failed to refresh global gallery: {e}")
+            gallery_paths = []
+        # Return a tuple: (tab result images, global gallery list)
+        return results, gallery_paths
 
     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')
@@ -110,10 +132,12 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
         # The Gallery component loads images via demo.load() on startup.
         # No need to add it to the run button outputs.
 
+        # When the Run button is pressed, update both the tab-specific result gallery
+        # and the global Gallery tab.
         run_btn.click(
-            fn=_run_with_meta,
+            fn=_run_and_refresh,
             inputs=input_list_flat,
-            outputs=[res_gal]
+            outputs=[res_gal, ui_components.get('gallery')]
         )
         # When a user selects an image from the global Gallery tab, load its generation info
         def _show_meta(selected, *_):