From fbf11fd7b15e9845c9846ee4e71e69bf03f2700b Mon Sep 17 00:00:00 2001
From: Ethanfel <ethan.fel@ts-pc.fr>
Date: Sat, 17 Jan 2026 15:30:35 +0100
Subject: [PATCH] Update tab_category_sorter.py

---
 tab_category_sorter.py | 54 +++++++++++++++++++++++++++++-------------
 1 file changed, 38 insertions(+), 16 deletions(-)

diff --git a/tab_category_sorter.py b/tab_category_sorter.py
index eeba3c1..0341be8 100644
--- a/tab_category_sorter.py
+++ b/tab_category_sorter.py
@@ -2,38 +2,50 @@ import streamlit as st
 import os, shutil
 from engine import SorterEngine
 
-def render(path_t, path_out, quality, naming_mode):
-    images = SorterEngine.get_images(path_t)
+def render(path_s, path_o, quality, naming_mode):
+    if not path_s or not os.path.exists(path_s):
+        st.warning("Please provide a valid Source Folder path.")
+        return
+    if not path_o:
+        st.warning("Please provide a Category Output folder.")
+        return
+
+    images = SorterEngine.get_images(path_s)
     categories = SorterEngine.get_categories()
 
     if st.session_state.idx_cat < len(images):
-        curr_p = os.path.join(path_t, images[st.session_state.idx_cat])
+        curr_file = images[st.session_state.idx_cat]
+        curr_p = os.path.join(path_s, curr_file)
         
-        # Filmstrip (Current + Next 6)
+        # 🎞️ Filmstrip Preview (Source script feature)
         st.write("### 🎞️ Filmstrip")
         fs_cols = st.columns(7)
+        # Show current + next 6
         for i, img_name in enumerate(images[st.session_state.idx_cat : st.session_state.idx_cat + 7]):
-            fs_cols[i].image(SorterEngine.compress_for_web(os.path.join(path_t, img_name), 10))
+            fs_cols[i].image(SorterEngine.compress_for_web(os.path.join(path_s, img_name), 10))
 
         st.divider()
 
         col_img, col_btns = st.columns([2, 1])
-        col_img.image(SorterEngine.compress_for_web(curr_p, quality), caption=images[st.session_state.idx_cat])
+        with col_img:
+            st.image(SorterEngine.compress_for_web(curr_p, quality), caption=curr_file)
         
         with col_btns:
-            fid = SorterEngine.get_folder_id(path_t)
-            st.write(f"**Folder ID:** {fid}")
+            fid = SorterEngine.get_folder_id(path_s)
+            st.write(f"**Folder ID:** `{fid}`")
             
-            # Dynamic Category Buttons from SQL
+            # Dynamic Category Buttons from Database
             for cat in categories:
-                if st.button(cat, use_container_width=True):
+                if st.button(cat, use_container_width=True, key=f"btn_{cat}"):
                     _, ext = os.path.splitext(curr_p)
-                    name = images[st.session_state.idx_cat] if naming_mode == "original" else f"{fid}{ext}"
-                    dst_dir = os.path.join(path_out, cat)
+                    # Use naming mode from database profile
+                    name = curr_file if naming_mode == "original" else f"{fid}{ext}"
+                    
+                    dst_dir = os.path.join(path_o, cat)
                     os.makedirs(dst_dir, exist_ok=True)
                     dst_p = os.path.join(dst_dir, name)
                     
-                    # Handle collisions
+                    # Prevent overwriting
                     count = 2
                     final_dst = dst_p
                     while os.path.exists(final_dst):
@@ -42,11 +54,21 @@ def render(path_t, path_out, quality, naming_mode):
                         count += 1
                         
                     shutil.move(curr_p, final_dst)
+                    st.session_state.history.append({'type': 'cat_move', 't_src': curr_p, 't_dst': final_dst})
                     st.toast(f"Moved to {cat}")
                     st.rerun()
-                    
-            if st.button("⏭️ SKIP", type="secondary", use_container_width=True):
+            
+            st.divider()
+            new_cat = st.text_input("➕ Quick Add Category")
+            if st.button("Add Category") and new_cat:
+                SorterEngine.add_category(new_cat)
+                st.rerun()
+
+            if st.button("⏭️ SKIP", use_container_width=True):
                 st.session_state.idx_cat += 1
                 st.rerun()
     else:
-        st.success("Finished Categorizing.")
\ No newline at end of file
+        st.success("Categorization complete for this folder.")
+        if st.button("Reset Category Counter"):
+            st.session_state.idx_cat = 0
+            st.rerun()
\ No newline at end of file