Add Refresh button to Gallery tab to manually reload thumbnails

ui/layout.py

@@ -43,8 +43,12 @@ def build_ui(event_handler_function):
 
             # New Gallery tab to display generated images
             with gr.TabItem("Gallery", id=5):
+                # Refresh button to manually reload thumbnails from the output folder.
+                refresh_btn = gr.Button("Refresh Gallery", variant="secondary")
                 gallery = gr.Gallery(label="Generated Images", show_label=False, columns=3, height="auto", type="filepath")
                 ui_components["gallery"] = gallery
+                # Bind the refresh button to the same loader used on startup.
+                refresh_btn.click(fn=load_gallery_images, outputs=gallery)
 
             ui_components["tabs"] = tabs
             ui_components["image_gen_tabs"] = tabs