116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
import streamlit as st
|
|
import db
|
|
from models import Video
|
|
from views import show_video_thumbnail
|
|
from controllers import label_widget
|
|
from datetime import date
|
|
|
|
st.set_page_config(page_title="Dance Video Manager", layout="wide")
|
|
st.title("💃 Dance Video Explorer")
|
|
|
|
# --- Barre latérale : filtres dynamiques ---
|
|
st.sidebar.header("⚙️ Filtres et affichage")
|
|
max_height = st.sidebar.slider("Hauteur maximale (px)", 100, 800, 300, 50)
|
|
st.markdown(f"""
|
|
<style>
|
|
img, video {{
|
|
max-height: {max_height}px !important;
|
|
object-fit: contain;
|
|
border-radius: 8px;
|
|
transition: all 0.3s ease-in-out;
|
|
}}
|
|
.unlabeled {{
|
|
border: 3px solid #f39c12;
|
|
box-shadow: 0 0 10px #f39c12;
|
|
border-radius: 10px;
|
|
padding: 5px;
|
|
margin-bottom: 10px;
|
|
}}
|
|
</style>
|
|
""", unsafe_allow_html=True)
|
|
|
|
all_labels = db.load_labels()
|
|
unique_days = db.get_unique_days()
|
|
unique_difficulties = db.get_unique_difficulties()
|
|
unique_addresses = db.get_unique_addresses()
|
|
|
|
selected_labels = st.sidebar.multiselect("Filtrer par labels", all_labels)
|
|
|
|
day_filter = st.sidebar.selectbox(
|
|
"Jour de la semaine",
|
|
["Tous"] + unique_days
|
|
)
|
|
|
|
difficulty_filter = st.sidebar.selectbox(
|
|
"Niveau de difficulté",
|
|
["Tous"] + unique_difficulties
|
|
)
|
|
|
|
address_keyword = st.sidebar.selectbox(
|
|
"Adresse (mot-clé)",
|
|
[""] + unique_addresses
|
|
)
|
|
|
|
start_date = st.sidebar.date_input("Date de début", value=None)
|
|
end_date = st.sidebar.date_input("Date de fin", value=None)
|
|
show_unlabeled_only = st.sidebar.checkbox("Afficher uniquement les vidéos sans labels")
|
|
|
|
# --- Filtrage SQL ---
|
|
day_value = None if day_filter == "Tous" else day_filter
|
|
df_videos = db.search_videos(
|
|
label_names=selected_labels,
|
|
day_of_week=day_value,
|
|
address_keyword=address_keyword if address_keyword else None,
|
|
start_date=start_date.isoformat() if start_date else None,
|
|
end_date=end_date.isoformat() if end_date else None,
|
|
difficulty=difficulty_filter
|
|
)
|
|
|
|
if df_videos.empty:
|
|
st.warning("Aucune vidéo trouvée avec ces critères.")
|
|
st.stop()
|
|
|
|
# --- Chargement des labels et affichage lazy ---
|
|
video_labels_map = {row["file_name"]: db.load_video_labels(row["file_name"]) for _, row in df_videos.iterrows()}
|
|
if show_unlabeled_only:
|
|
df_videos = df_videos[df_videos["file_name"].apply(lambda fn: not video_labels_map.get(fn))]
|
|
|
|
videos = [Video(**row) for _, row in df_videos.iterrows()]
|
|
|
|
# --- Lazy loading ---
|
|
page_size = 20
|
|
if "video_page" not in st.session_state:
|
|
st.session_state.video_page = 1
|
|
|
|
start = 0
|
|
end = st.session_state.video_page * page_size
|
|
visible_videos = videos[start:end]
|
|
|
|
for video in visible_videos:
|
|
preselected = video_labels_map.get(video.file_name, [])
|
|
css_class = "unlabeled" if not preselected else ""
|
|
with st.container():
|
|
st.markdown(f"<div class='{css_class}'>", unsafe_allow_html=True)
|
|
col3 = show_video_thumbnail(video)
|
|
with col3:
|
|
label_widget(video, preselected=preselected)
|
|
# --- Sélecteur de difficulté ---
|
|
new_level = st.selectbox(
|
|
"🎚 Niveau de difficulté",
|
|
["Tout niveau", "Débutant", "Intermédiaire", "Avancé", "Star"],
|
|
index=["Tout niveau", "Débutant", "Intermédiaire", "Avancé", "Star"].index(video.difficulty_display),
|
|
key=f"diff_{video.file_name}"
|
|
)
|
|
if new_level != video.difficulty_display:
|
|
db.update_video_difficulty(video.file_name, new_level)
|
|
st.success(f"Niveau mis à jour pour {video.file_name}")
|
|
st.markdown("</div>", unsafe_allow_html=True)
|
|
|
|
# --- Bouton "Voir plus" ---
|
|
if end < len(videos):
|
|
if st.button("📦 Charger plus de vidéos"):
|
|
st.session_state.video_page += 1
|
|
st.rerun()
|
|
else:
|
|
st.info("✅ Toutes les vidéos sont affichées.")
|