diff --git a/app.py b/app.py index a005494..549659e 100644 --- a/app.py +++ b/app.py @@ -32,7 +32,7 @@ with st.sidebar: st.rerun() # --- TABS --- -t1, t2, t3, t4 = st.tabs(["🕒 1. Discovery", "🆔 2. ID Review", "♻️ 3. Unused", "📂 4. Category Sorter"]) +t1, t2, t3, t4, t5 = st.tabs(["🕒 Discovery", "🆔 Review", "♻️ Unused", "📂 Categorizer", "🖼️ Gallery Sorter"]) with t1: path_t1 = st.text_input("Discovery Target", value=p_data.get("tab1_target", "/storage"), key="t1_input") @@ -61,4 +61,10 @@ with t4: # This now correctly accepts the 'mode' argument SorterEngine.save_tab_paths(selected_profile, t4_s=path_t4_s, t4_o=path_t4_o, mode=mode) - tab_category_sorter.render(path_t4_s, path_t4_o, quality, mode) \ No newline at end of file + tab_category_sorter.render(path_t4_s, path_t4_o, quality, mode) + +with t5: + path_t5 = st.text_input("Gallery Source (PNG)", value=p_data.get("tab5_source", "/storage"), key="t5_input") + if path_t5 != p_data.get("tab5_source"): + SorterEngine.save_tab_paths(selected_profile, t5_s=path_t5) # Ensure engine supports t5_s + tab_gallery_sorter.render(path_t5, p_data.get("tab4_out"), quality) \ No newline at end of file