Upload app.py

app.py

@@ -0,0 +1,462 @@
+import os
+import io
+import json
+import yaml
+import shutil
+from pathlib import Path
+from passlib.hash import bcrypt
+try:
+    import tomllib
+except ModuleNotFoundError:  # Py<3.11
+    import tomli as tomllib
+import streamlit as st
+from moviepy.editor import VideoFileClip
+
+from database import set_db_path, init_schema, get_user, create_video, update_video_status, list_videos, get_video, get_all_users, upsert_result, get_results, add_feedback, get_feedback_for_video, get_feedback_stats
+from api_client import APIClient
+from utils import ensure_dirs, save_bytes, save_text, human_size
+
+
+# -- Move DB ---
+os.environ["STREAMLIT_DATA_DIRECTORY"] = "/tmp/.streamlit"
+Path("/tmp/.streamlit").mkdir(parents=True, exist_ok=True)
+Path("/tmp/data").mkdir(parents=True, exist_ok=True)
+source_db = "init_data/veureu.db"
+target_db = "/tmp/data/app.db"
+if not os.path.exists(target_db) and os.path.exists(source_db):
+    shutil.copy(source_db, target_db)
+
+static_videos = Path(__file__).parent / "videos"
+runtime_videos = Path("/tmp/data/videos")
+if not runtime_videos.exists():
+    shutil.copytree(static_videos, runtime_videos, dirs_exist_ok=True)
+
+
+# --- Config ---
+def _load_yaml(path="config.yaml") -> dict:
+    with open(path, "r", encoding="utf-8") as f:
+        cfg = yaml.safe_load(f) or {}
+    # interpolación sencilla de ${VARS} si las usas en el YAML
+    def _subst(s: str) -> str:
+        return os.path.expandvars(s) if isinstance(s, str) else s
+
+    # aplica sustitución en los campos que te interesan
+    if "api" in cfg:
+        cfg["api"]["base_url"] = _subst(cfg["api"].get("base_url", ""))
+        cfg["api"]["token"]    = _subst(cfg["api"].get("token", ""))
+
+    if "storage" in cfg and "root_dir" in cfg["storage"]:
+        cfg["storage"]["root_dir"] = _subst(cfg["storage"]["root_dir"])
+
+    if "sqlite" in cfg and "path" in cfg["sqlite"]:
+        cfg["sqlite"]["path"] = _subst(cfg["sqlite"]["path"])
+
+    return cfg
+
+CFG = _load_yaml("config.yaml")
+
+# Ajuste de variables según tu esquema YAML
+DATA_DIR = CFG.get("storage", {}).get("root_dir", "data")
+BACKEND_BASE_URL = CFG.get("api", {}).get("base_url", "http://localhost:8000")
+USE_MOCK = bool(CFG.get("app", {}).get("use_mock", False))  # si no la tienes en el yaml, queda False
+API_TOKEN = CFG.get("api", {}).get("token") or os.getenv("API_SHARED_TOKEN")
+
+os.makedirs(DATA_DIR, exist_ok=True)
+ensure_dirs(DATA_DIR)
+DB_PATH = os.path.join(DATA_DIR, "app.db")
+set_db_path(DB_PATH)
+init_schema()
+
+api = APIClient(BACKEND_BASE_URL, use_mock=USE_MOCK, data_dir=DATA_DIR, token=API_TOKEN)
+
+st.set_page_config(page_title="Veureu — Audiodescripció", page_icon="🎬", layout="wide")
+
+# --- Session: auth ---
+# print("Usuarios disponibles:", get_all_users()) # Descomentar para depurar
+if "user" not in st.session_state:
+    st.session_state.user = None  # dict with {username, role, id(optional)}
+
+def require_login():
+    if not st.session_state.user:
+        st.info("Por favor, inicia sesión para continuar.")
+        login_form()
+        st.stop()
+
+def verify_password(password: str, pw_hash: str) -> bool:
+    try:
+        return bcrypt.verify(password, pw_hash)
+    except Exception:
+        return False
+        
+# --- Sidebar (only after login) ---
+role = st.session_state.user["role"] if st.session_state.user else None
+with st.sidebar:
+    st.title("Veureu")
+    if st.session_state.user:
+        st.write(f"Usuari: **{st.session_state.user['username']}** (rol: {st.session_state.user['role']})")
+        if st.button("Tancar sessió"):
+            st.session_state.user = None
+            st.rerun()
+    if st.session_state.user:
+        page = st.radio("Navegació", ["Analitzar video-transcripcions","Processar vídeo nou","Estadístiques"], index=0)
+    else:
+        page = None
+
+# --- Pre-login screen ---
+if not st.session_state.user:
+    st.title("Veureu — Audiodescripció")
+    def login_form():
+        st.subheader("Inici de sessió")
+        username = st.text_input("Usuari")
+        password = st.text_input("Contrasenya", type="password")
+        if st.button("Entrar", type="primary"):
+            row = get_user(username)
+            if row and verify_password(password, row["pw_hash"]):
+                st.session_state.user = {"id": row["id"], "username": row["username"], "role": row["role"]}
+                st.success(f"Benvingut/da, {row['username']}")
+                st.rerun()
+            else:
+                st.error("Credencials invàlides")
+    login_form()
+    st.stop()
+
+# --- Pages ---
+if page == "Processar vídeo nou":
+    require_login()
+    if role != "verd":
+        st.error("No tens permisos per processar nous vídeos. Canvia d'usuari o sol·licita permisos.")
+        st.stop()
+
+    st.header("Processar un nou clip de vídeo")
+
+    # Inicializar el estado de la página si no existe
+    if 'video_uploaded' not in st.session_state:
+        st.session_state.video_uploaded = None
+    if 'characters_detected' not in st.session_state:
+        st.session_state.characters_detected = None
+    if 'characters_saved' not in st.session_state:
+        st.session_state.characters_saved = False
+
+    # --- 1. Subida del vídeo ---
+    MAX_SIZE_MB = 20
+    MAX_DURATION_S = 240 # 4 minutos
+
+    uploaded_file = st.file_uploader("Puja un clip de vídeo (MP4, < 20MB, < 4 minuts)", type=["mp4"], key="video_uploader")
+
+    if uploaded_file is not None:
+        # Resetear el estado si se sube un nuevo archivo
+        if st.session_state.video_uploaded is None or uploaded_file.name != st.session_state.video_uploaded.get('original_name'):
+            st.session_state.video_uploaded = {'original_name': uploaded_file.name, 'status': 'validating'}
+            st.session_state.characters_detected = None
+            st.session_state.characters_saved = False
+
+        # --- Validación y Procesamiento ---
+        if st.session_state.video_uploaded['status'] == 'validating':
+            is_valid = True
+            # 1. Validar tamaño
+            if uploaded_file.size > MAX_SIZE_MB * 1024 * 1024:
+                st.error(f"El vídeo supera el límit de {MAX_SIZE_MB}MB.")
+                is_valid = False
+
+            if is_valid:
+                with st.spinner("Processant el vídeo..."):
+                    # Guardar temporalmente para analizarlo
+                    with open("temp_video.mp4", "wb") as f:
+                        f.write(uploaded_file.getbuffer())
+                    
+                    clip = VideoFileClip("temp_video.mp4")
+                    duration = clip.duration
+
+                    # 2. Validar y truncar duración
+                    was_truncated = False
+                    if duration > MAX_DURATION_S:
+                        clip = clip.subclip(0, MAX_DURATION_S)
+                        was_truncated = True
+
+                    # Crear carpeta y guardar el vídeo final
+                    video_name = Path(uploaded_file.name).stem
+                    video_dir = Path("/tmp/data/videos") / video_name
+                    video_dir.mkdir(parents=True, exist_ok=True)
+                    final_video_path = video_dir / f"{video_name}.mp4"
+                    clip.write_videofile(str(final_video_path), codec="libx264", audio_codec="aac")
+
+                    clip.close()
+                    os.remove("temp_video.mp4")
+
+                    # Actualizar estado
+                    st.session_state.video_uploaded.update({
+                        'status': 'processed',
+                        'path': str(final_video_path),
+                        'was_truncated': was_truncated
+                    })
+                    st.rerun()
+
+    # --- Mensajes de estado ---
+    if st.session_state.video_uploaded and st.session_state.video_uploaded['status'] == 'processed':
+        st.success(f"Vídeo '{st.session_state.video_uploaded['original_name']}' pujat i processat correctament.")
+        if st.session_state.video_uploaded['was_truncated']:
+            st.warning(f"El vídeo s'ha truncat a {MAX_DURATION_S // 60} minuts.")
+
+    # --- 2. Detección de personajes ---
+    st.markdown("---")
+    col1, col2 = st.columns([1, 3])
+    with col1:
+        detect_button_disabled = st.session_state.video_uploaded is None
+        if st.button("Detectar Personatges", disabled=detect_button_disabled):
+            with st.spinner("Detectant personatges..."):
+                # Aquí iría la llamada a la API para detectar personajes
+                # Por ahora, usamos datos de ejemplo
+                st.session_state.characters_detected = [
+                    {"id": "char1", "image_path": "init_data/placeholder.png", "description": "Dona amb cabell ros i ulleres"},
+                    {"id": "char2", "image_path": "init_data/placeholder.png", "description": "Home amb barba i barret"},
+                ]
+                st.session_state.characters_saved = False # Resetear el estado de guardado
+
+    # --- 3. Formularios de personajes ---
+    if st.session_state.characters_detected:
+        st.subheader("Personatges detectats")
+        for char in st.session_state.characters_detected:
+            with st.form(key=f"form_{char['id']}"):
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.image(char['image_path'], width=150)
+                
+                with col2:
+                    st.caption(char['description'])
+                    st.text_input("Nom del personatge", key=f"name_{char['id']}")
+                    st.form_submit_button("Cercar")
+        
+        st.markdown("---_**")
+
+        # --- 4. Guardar y Generar ---
+        col1, col2, col3 = st.columns([1,1,2])
+        with col1:
+            if st.button("Desar", type="primary"):
+                # Aquí iría la lógica para guardar los nombres de los personajes
+                st.session_state.characters_saved = True
+                st.success("Personatges desats correctament.")
+
+        with col2:
+            if st.session_state.characters_saved:
+                st.button("Generar Audiodescripció")
+
+elif page == "Analitzar video-transcripcions":
+    require_login()
+    st.header("Analitzar video-transcripcions")
+    base_dir = Path("/tmp/data/videos")
+
+    if not base_dir.exists():
+        st.info("No s'ha trobat la carpeta **videos**. Crea-la i afegeix-hi subcarpetes amb els teus vídeos.")
+        st.stop()
+
+    carpetes = [p.name for p in sorted(base_dir.iterdir()) if p.is_dir() and p.name != 'completed']
+    if not carpetes:
+        st.info("No s'ha trobat la carpeta **videos**. Crea-la i afegeix-hi subcarpetes amb els teus vídeos.")
+        st.stop()
+
+    # --- Lógica de Estado y Selección ---
+
+    # Detectar si el vídeo principal ha cambiado para resetear el estado secundario
+    if 'current_video' not in st.session_state:
+        st.session_state.current_video = None
+
+    # Widget de selección de vídeo
+    seleccio = st.selectbox("Selecciona un vídeo (carpeta):", carpetes, index=None, placeholder="Tria una carpeta…")
+
+    if seleccio != st.session_state.current_video:
+        st.session_state.current_video = seleccio
+        # Forzar reseteo de los widgets dependientes
+        st.session_state.version_selector = None
+        st.session_state.add_ad_checkbox = False
+        st.rerun()
+
+    if not seleccio:
+        st.stop()
+
+    vid_dir = base_dir / seleccio
+    mp4s = sorted(vid_dir.glob("*.mp4"))
+
+    # --- Dibujado de la Interfaz ---
+    col_video, col_txt = st.columns([2, 1], gap="large")
+
+    with col_video:
+        # Selección de versión
+        subcarpetas_ad = [p.name for p in sorted(vid_dir.iterdir()) if p.is_dir()]
+        default_index_sub = subcarpetas_ad.index("Salamandra") if "Salamandra" in subcarpetas_ad else 0
+        subcarpeta_seleccio = st.selectbox(
+            "Selecciona una versió d'audiodescripció:", subcarpetas_ad,
+            index=default_index_sub if subcarpetas_ad else None,
+            placeholder="Tria una versió…" if subcarpetas_ad else "No hi ha versions",
+            key="version_selector"
+        )
+
+        # Lógica de vídeo AD
+        video_ad_path = vid_dir / subcarpeta_seleccio / "une_ad.mp4" if subcarpeta_seleccio else None
+        is_ad_video_available = video_ad_path is not None and video_ad_path.exists()
+
+        # Checkbox
+        add_ad_video = st.checkbox("Afegir audiodescripció", disabled=not is_ad_video_available, key="add_ad_checkbox")
+
+        # Decidir qué vídeo mostrar
+        video_to_show = None
+        if add_ad_video and is_ad_video_available:
+            video_to_show = video_ad_path
+        elif mp4s:
+            video_to_show = mp4s[0]
+
+        if video_to_show:
+            st.video(str(video_to_show))
+        else:
+            st.warning("No s'ha trobat cap fitxer **.mp4** a la carpeta seleccionada.")
+
+        st.markdown("---")
+
+        # Sección de ACCIONES
+        st.markdown("#### Accions")
+        c1, c2 = st.columns(2)
+        with c1:
+            if st.button("Reconstruir àudio amb narració lliure", use_container_width=True, key="rebuild_free_ad"):
+                if subcarpeta_seleccio:
+                    free_ad_path = vid_dir / subcarpeta_seleccio / "free_ad.txt"
+                    if free_ad_path.exists():
+                        with st.spinner("Generant àudio de la narració lliure..."):
+                            text_content = free_ad_path.read_text(encoding="utf-8")
+                            voice = "central/grau" # Voz fijada
+                            response = api.tts_matxa(text=text_content, voice=voice)
+                            if "mp3_bytes" in response:
+                                output_path = vid_dir / subcarpeta_seleccio / "free_ad.mp3"
+                                save_bytes(output_path, response["mp3_bytes"])
+                                st.success(f"Àudio generat i desat a: {output_path}")
+                            else:
+                                st.error(f"Error en la generació de l'àudio: {response.get('error', 'Desconegut')}")
+                    else:
+                        st.warning("No s'ha trobat el fitxer 'free_ad.txt' en aquesta versió.")
+
+        with c2:
+            if st.button("Reconstruir vídeo amb audiodescripció", use_container_width=True, key="rebuild_video_ad"):
+                if subcarpeta_seleccio and mp4s:
+                    une_srt_path = vid_dir / subcarpeta_seleccio / "une_ad.srt"
+                    video_original_path = mp4s[0]
+                    if une_srt_path.exists():
+                        with st.spinner("Reconstruint el vídeo amb l'audiodescripció... Aquesta operació pot trigar una estona."):
+                            response = api.rebuild_video_with_ad(video_path=str(video_original_path), srt_path=str(une_srt_path))
+                            if "video_bytes" in response:
+                                output_path = vid_dir / subcarpeta_seleccio / "video_ad_rebuilt.mp4"
+                                save_bytes(output_path, response["video_bytes"])
+                                st.success(f"Vídeo reconstruït i desat a: {output_path}")
+                                st.info("Pots visualitzar-lo activant la casella 'Afegir audiodescripció' i seleccionant el nou fitxer si cal.")
+                            else:
+                                st.error(f"Error en la reconstrucció del vídeo: {response.get('error', 'Desconegut')}")
+                    else:
+                        st.warning("No s'ha trobat el fitxer 'une_ad.srt' en aquesta versió.")
+
+
+    # --- Columna Derecha (Editor de texto y guardado) ---
+    with col_txt:
+        tipus_ad_options = ["narració lliure", "UNE-153010"]
+        tipus_ad_seleccio = st.selectbox("Fitxer d'audiodescripció a editar:", tipus_ad_options)
+        
+        ad_filename = "free_ad.txt" if tipus_ad_seleccio == "narració lliure" else "une_ad.srt"
+        
+        # Cargar el contenido del fichero seleccionado
+        text_content = ""
+        ad_path = None
+        if subcarpeta_seleccio:
+            ad_path = vid_dir / subcarpeta_seleccio / ad_filename
+            if ad_path.exists():
+                try:
+                    text_content = ad_path.read_text(encoding="utf-8")
+                except Exception:
+                    text_content = ad_path.read_text(errors="ignore")
+            else:
+                st.info(f"No s'ha trobat el fitxer **{ad_filename}**.")
+        else:
+            st.warning("Selecciona una versió per veure els fitxers.")
+
+        # Área de texto para edición
+        new_text = st.text_area(f"Contingut de {tipus_ad_seleccio}", value=text_content, height=500, key=f"editor_{seleccio}_{subcarpeta_seleccio}_{ad_filename}")
+
+        # Controles de reproducción de narración (selector de voz eliminado)
+        if st.button("▶️ Reproduir narració", use_container_width=True, disabled=not new_text.strip(), key="play_button_editor"):
+            with st.spinner("Generant àudio..."):
+                # Lógica de TTS con el texto del área
+                pass # Implementación de la llamada a la API TTS
+
+        # Botón de guardado
+        if st.button("Desar canvis", use_container_width=True, type="primary"):
+            if ad_path:
+                try:
+                    ad_path.write_text(new_text, encoding="utf-8")
+                    st.success(f"Fitxer **{ad_filename}** desat correctament.")
+                    st.rerun()
+                except Exception as e:
+                    st.error(f"No s'ha pogut desar el fitxer: {e}")
+            else:
+                st.error("No s'ha seleccionat una ruta de fitxer vàlida per desar.")
+
+
+    st.markdown("---")
+    st.subheader("Avaluació de la qualitat de l'audiodescripció")
+
+    c1, c2, c3 = st.columns(3)
+    with c1:
+        transcripcio = st.slider("Transcripció", 1, 10, 7)
+        identificacio = st.slider("Identificació de personatges", 1, 10, 7)
+    with c2:
+        localitzacions = st.slider("Localitzacions", 1, 10, 7)
+        activitats = st.slider("Activitats", 1, 10, 7)
+    with c3:
+        narracions = st.slider("Narracions", 1, 10, 7)
+        expressivitat = st.slider("Expressivitat", 1, 10, 7)
+
+    comments = st.text_area("Comentaris (opcional)", placeholder="Escriu els teus comentaris lliures…", height=120)
+
+    role = st.session_state.user["role"]
+    can_rate = role in ("verd", "groc", "blau")
+
+    if not can_rate:
+        st.info("El teu rol no permet enviar valoracions.")
+    else:
+        if st.button("Enviar valoració", type="primary", use_container_width=True):
+            try:
+                from database import add_feedback_ad
+                add_feedback_ad(
+                    video_name=seleccio,
+                    user_id=st.session_state.user["id"],
+                    transcripcio=transcripcio,
+                    identificacio=identificacio,
+                    localitzacions=localitzacions,
+                    activitats=activitats,
+                    narracions=narracions,
+                    expressivitat=expressivitat,
+                    comments=comments or None
+                )
+                st.success("Gràcies! La teva valoració s'ha desat correctament.")
+            except Exception as e:
+                st.error(f"S'ha produït un error en desar la valoració: {e}")
+
+
+elif page == "Estadístiques":
+    require_login()
+    st.header("Estadístiques")
+
+    from database import get_feedback_ad_stats
+    stats = get_feedback_ad_stats()  # medias por vídeo + avg_global
+    if not stats:
+        st.caption("Encara no hi ha valoracions.")
+        st.stop()
+
+    import pandas as pd
+    df = pd.DataFrame(stats, columns=stats[0].keys())
+    ordre = st.radio("Ordre de rànquing", ["Descendent (millors primer)", "Ascendent (pitjors primer)"], horizontal=True)
+    if ordre.startswith("Asc"):
+        df = df.sort_values("avg_global", ascending=True)
+    else:
+        df = df.sort_values("avg_global", ascending=False)
+
+    st.subheader("Rànquing de vídeos")
+    st.dataframe(
+        df[["video_name","n","avg_global","avg_transcripcio","avg_identificacio","avg_localitzacions","avg_activitats","avg_narracions", "avg_expressivitat"]],
+        use_container_width=True
+    )
+