Ethanfel/sorting-sorted
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update tab_category_sorter.py

Browse Source
This commit is contained in:
Ethanfel
2026-01-17 15:10:08 +01:00
parent 48d8c740d0
commit 991a71fc68
1 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tab_category_sorter.py
View File

@@ -9,18 +9,22 @@ def render(path_t, path_out, quality, naming_mode):
if st.session_state.idx_cat < len(images): if st.session_state.idx_cat < len(images):
curr_p = os.path.join(path_t, images[st.session_state.idx_cat]) curr_p = os.path.join(path_t, images[st.session_state.idx_cat])
# Filmstrip # Filmstrip (Current + Next 6)
st.write("### 🎞️ Filmstrip") st.write("### 🎞️ Filmstrip")
fs_cols = st.columns(7) fs_cols = st.columns(7)
for i, img_name in enumerate(images[st.session_state.idx_cat : st.session_state.idx_cat + 7]): 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_t, img_name), 10))
st.divider()
col_img, col_btns = st.columns([2, 1]) col_img, col_btns = st.columns([2, 1])
col_img.image(SorterEngine.compress_for_web(curr_p, quality)) col_img.image(SorterEngine.compress_for_web(curr_p, quality), caption=images[st.session_state.idx_cat])
with col_btns: with col_btns:
fid = SorterEngine.get_folder_id(path_t) fid = SorterEngine.get_folder_id(path_t)
st.write(f"**Folder ID:** {fid}") st.write(f"**Folder ID:** {fid}")
# Dynamic Category Buttons from SQL
for cat in categories: for cat in categories:
if st.button(cat, use_container_width=True): if st.button(cat, use_container_width=True):
_, ext = os.path.splitext(curr_p) _, ext = os.path.splitext(curr_p)
@@ -28,10 +32,21 @@ def render(path_t, path_out, quality, naming_mode):
dst_dir = os.path.join(path_out, cat) dst_dir = os.path.join(path_out, cat)
os.makedirs(dst_dir, exist_ok=True) os.makedirs(dst_dir, exist_ok=True)
dst_p = os.path.join(dst_dir, name) dst_p = os.path.join(dst_dir, name)
shutil.move(curr_p, dst_p)
st.session_state.history.append({'type': 'cat_move', 't_src': curr_p, 't_dst': dst_p}) # Handle collisions
count = 2
final_dst = dst_p
while os.path.exists(final_dst):
root, ext = os.path.splitext(name)
final_dst = os.path.join(dst_dir, f"{root}_{count}{ext}")
count += 1
shutil.move(curr_p, final_dst)
st.toast(f"Moved to {cat}")
st.rerun() st.rerun()
if st.button("SKIP"):
if st.button("⏭️ SKIP", type="secondary", use_container_width=True):
st.session_state.idx_cat += 1 st.session_state.idx_cat += 1
st.rerun() st.rerun()
else: st.success("Finished Categorizing.") else:
st.success("Finished Categorizing.")