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

Update tab_time_discovery.py

Browse Source
This commit is contained in:
Ethanfel
2026-01-17 13:55:32 +01:00
parent 40f10e50bc
commit 4fb0b9a404
1 changed files with 55 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
tab_time_discovery.py
View File

@@ -1,8 +1,8 @@
import streamlit as st
import os, shutil
import os
from engine import SorterEngine
def render(path_t, path_c, quality, threshold, id_prefix):
def render(path_t, quality, threshold, id_prefix):
target_imgs = SorterEngine.get_images(path_t)
unmatched_t = [f for f in target_imgs if not f.startswith("id")]
@@ -11,35 +11,59 @@ def render(path_t, path_c, quality, threshold, id_prefix):
t_path = os.path.join(path_t, t_file)
t_time = os.path.getmtime(t_path)
best_c_path, min_delta = None, threshold
for c_file in SorterEngine.get_images(path_c):
c_p = os.path.join(path_c, c_file)
# 1. Main Display: Target Image
st.subheader(f"Target: {t_file} ({st.session_state.idx_time + 1}/{len(unmatched_t)})")
st.image(SorterEngine.compress_for_web(t_path, quality))
# 2. Scanning all sibling folders
control_folders = SorterEngine.get_sibling_controls(path_t)
all_matches = []
for folder in control_folders:
for c_file in SorterEngine.get_images(folder):
c_p = os.path.join(folder, c_file)
delta = abs(t_time - os.path.getmtime(c_p))
if delta < min_delta:
min_delta, best_c_path = delta, c_p
if delta <= threshold:
all_matches.append({'path': c_p, 'delta': delta, 'folder': os.path.basename(folder)})
if best_c_path:
st.subheader(f"Time Match Found (Δ {min_delta:.1f}s)")
col1, col2 = st.columns(2)
col1.image(SorterEngine.compress_for_web(t_path, quality), caption=t_file)
col2.image(SorterEngine.compress_for_web(best_c_path, quality), caption=os.path.basename(best_c_path))
all_matches = sorted(all_matches, key=lambda x: x['delta'])
b1, b2, b3 = st.columns(3)
if b1.button("MATCH", type="primary", use_container_width=True):
t_dst, c_dst = SorterEngine.execute_move(t_path, best_c_path, path_t, path_c, id_prefix, "standard")
st.session_state.history.append({'type': 'link_standard', 't_src': t_path, 't_dst': t_dst, 'c_src': best_c_path, 'c_dst': c_dst})
# 3. Actions for Target
col_n, col_s = st.columns(2)
if col_n.button("🚫 NO MATCH (N)", use_container_width=True):
dst = os.path.join(path_t, "selected_target_no_control", t_file)
os.makedirs(os.path.dirname(dst), exist_ok=True)
shutil.move(t_path, dst)
st.session_state.history.append({'type': 'move', 't_src': t_path, 't_dst': dst})
st.rerun()
if b2.button("SOLO", use_container_width=True):
t_dst, c_dst = SorterEngine.execute_move(t_path, best_c_path, path_t, path_c, id_prefix, "solo")
st.session_state.history.append({'type': 'link_solo', 't_src': t_path, 't_dst': t_dst, 'c_src': best_c_path, 'c_dst': c_dst})
st.rerun()
if b3.button("SKIP", use_container_width=True):
if col_s.button("⏩ SKIP (S)", use_container_width=True):
st.session_state.idx_time += 1
st.rerun()
st.divider()
st.write("### 🕒 Time-Sync Matches")
# 4. Result Grid (The Sidebar Logic)
if not all_matches:
st.info("No time-sync matches found in sibling folders.")
else:
st.warning("No time matches found.")
if st.button("SKIP"):
st.session_state.idx_time += 1
for m in all_matches:
with st.container(border=True):
c1, c2 = st.columns([1, 2])
c1.image(SorterEngine.compress_for_web(m['path'], 30))
with c2:
st.write(f"**{os.path.basename(m['path'])}**")
st.write(f"From: `{m['folder']}` | Δ: {m['delta']:.1f}s")
b1, b2 = st.columns(2)
if b1.button("MATCH", key=f"m_{m['path']}", use_container_width=True):
t_dst, c_dst = SorterEngine.execute_match(t_path, m['path'], path_t, id_prefix, "standard")
st.session_state.history.append({'type': 'link_standard', 't_src': t_path, 't_dst': t_dst, 'c_src': m['path'], 'c_dst': c_dst})
st.rerun()
if b2.button("SOLO", key=f"s_{m['path']}", use_container_width=True):
t_dst, c_dst = SorterEngine.execute_match(t_path, m['path'], path_t, id_prefix, "solo")
st.session_state.history.append({'type': 'link_solo', 't_src': t_path, 't_dst': t_dst, 'c_src': m['path'], 'c_dst': c_dst})
st.rerun()
else:
st.success("All unmatched images reviewed.")
st.success("Target folder processing complete!")