From cf1238bbff2a0c3ea7ef16144327928cee90a8bb Mon Sep 17 00:00:00 2001
From: Ethanfel <ethan.fel@ts-pc.fr>
Date: Thu, 22 Jan 2026 15:35:32 +0100
Subject: [PATCH] Update gallery_app.py

---
 gallery_app.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/gallery_app.py b/gallery_app.py
index d2af616..d47a728 100644
--- a/gallery_app.py
+++ b/gallery_app.py
@@ -719,8 +719,14 @@ def build_header():
                 state.load_active_profile()
                 state.active_cat = "control"  # Reset to default category
                 SorterEngine.clear_staging_area()  # Clear staging for new profile
-                refresh_staged_info()
-                refresh_ui()
+                
+                # Auto-load if source path exists
+                if os.path.exists(state.source_dir):
+                    load_images()
+                else:
+                    state.all_images = []
+                    refresh_staged_info()
+                    refresh_ui()
             
             profile_select = ui.select(
                 list(state.profiles.keys()),