Update gallery_app.py

gallery_app.py

@@ -78,7 +78,7 @@ class AppState:
 
     def get_categories(self) -> List[str]:
         """Get list of categories, ensuring active_cat exists."""
-        cats = SorterEngine.get_categories() or ["Default"]
+        cats = SorterEngine.get_categories(self.profile_name) or ["control"]
         if self.active_cat not in cats:
             self.active_cat = cats[0]
         return cats
@@ -340,7 +340,7 @@ def render_sidebar():
             
             def add_category():
                 if new_cat_input.value:
-                    SorterEngine.add_category(new_cat_input.value)
+                    SorterEngine.add_category(new_cat_input.value, state.profile_name)
                     state.active_cat = new_cat_input.value
                     refresh_staged_info()
                     render_sidebar()
@@ -350,7 +350,7 @@ def render_sidebar():
         # Delete category
         with ui.expansion('Danger Zone', icon='warning').classes('w-full text-red-400 mt-2'):
             def delete_category():
-                SorterEngine.delete_category(state.active_cat)
+                SorterEngine.delete_category(state.active_cat, state.profile_name)
                 refresh_staged_info()
                 render_sidebar()
             
@@ -492,16 +492,50 @@ def build_header():
         with ui.row().classes('w-full items-center gap-4 no-wrap px-4'):
             ui.label('🖼️ NiceSorter').classes('text-xl font-bold shrink-0 text-green-400')
             
-            # Profile selector
+            # Profile selector with add/delete
-            profile_names = list(state.profiles.keys())
+            profile_select = ui.select(
-            
+                list(state.profiles.keys()), 
-            def change_profile(e):
+                value=state.profile_name,
-                state.profile_name = e.value
+                on_change=lambda e: (
-                state.load_active_profile()
+                    setattr(state, 'profile_name', e.value),
+                    state.load_active_profile(),
                     load_images()
+                )
+            ).props('dark dense options-dense borderless').classes('w-32')
             
-            ui.select(profile_names, value=state.profile_name, on_change=change_profile) \
+            def add_profile():
-                .props('dark dense options-dense borderless').classes('w-32')
+                async def create():
+                    name = await dialog
+                    if name and name not in state.profiles:
+                        state.profiles[name] = {"tab5_source": "/storage", "tab5_out": "/storage"}
+                        state.profile_name = name
+                        state.load_active_profile()
+                        profile_select.options = list(state.profiles.keys())
+                        profile_select.value = name
+                        ui.notify(f"Profile '{name}' created", type='positive')
+                
+                with ui.dialog() as dialog, ui.card().classes('p-4'):
+                    ui.label('New Profile Name').classes('font-bold')
+                    name_input = ui.input(placeholder='Profile name').props('autofocus')
+                    with ui.row().classes('w-full justify-end gap-2 mt-2'):
+                        ui.button('Cancel', on_click=lambda: dialog.submit(None)).props('flat')
+                        ui.button('Create', on_click=lambda: dialog.submit(name_input.value)).props('color=green')
+                asyncio.create_task(create())
+            
+            def delete_profile():
+                if len(state.profiles) <= 1:
+                    ui.notify("Cannot delete the last profile", type='warning')
+                    return
+                del state.profiles[state.profile_name]
+                state.profile_name = list(state.profiles.keys())[0]
+                state.load_active_profile()
+                profile_select.options = list(state.profiles.keys())
+                profile_select.value = state.profile_name
+                load_images()
+                ui.notify("Profile deleted", type='info')
+            
+            ui.button(icon='add', on_click=add_profile).props('flat round dense color=green').tooltip('New profile')
+            ui.button(icon='delete', on_click=delete_profile).props('flat round dense color=red').tooltip('Delete profile')
             
             # Source and output paths
             with ui.row().classes('flex-grow gap-2'):