"""UI logic for the "Validació" page.""" from __future__ import annotations from datetime import datetime from pathlib import Path from typing import Dict import sys import shutil import os import tempfile import zipfile import io import requests import streamlit as st from databases import ( get_accessible_videos_with_sha1, get_audiodescription, get_audiodescription_history, update_audiodescription_text, update_audiodescription_info_ad, log_action, update_video_status, get_video_owner_by_sha1, get_videos_by_status, insert_action, ) from persistent_data_gate import _load_data_origin def _log(msg: str) -> None: """Helper de logging a stderr amb timestamp (coherent amb auth.py).""" ts = datetime.now().strftime("%Y-%m-%d %H:%M:%S") sys.stderr.write(f"[{ts}] {msg}\n") sys.stderr.flush() def render_validation_page( compliance_client, runtime_videos: Path, permissions: Dict[str, bool], username: str, ) -> None: if not permissions.get("validar", False): st.warning("⚠️ No tens permisos per accedir a aquesta secció de validació.") st.stop() st.header("🔍 Validació de Vídeos") tab_videos, tab_ads = st.tabs(["📹 Validar Vídeos", "🎬 Validar Audiodescripcions"]) base_dir = Path(__file__).resolve().parent.parent data_origin = _load_data_origin(base_dir) # Llegir config.yaml per saber si cal enviar SMS a l'usuari quan el vídeo és aprovat config_path = base_dir / "config.yaml" user_sms_enabled = False try: if config_path.exists(): import yaml # import local per no afegir-lo al top del fitxer with config_path.open("r", encoding="utf-8") as f: cfg = yaml.safe_load(f) or {} validation_cfg = cfg.get("validation", {}) or {} user_sms_enabled = bool(validation_cfg.get("user_sms_enabled", False)) except Exception: user_sms_enabled = False # Llista de vídeos accessibles (mode internal) o pendents al backend (mode external) session_id = st.session_state.get("session_id") accessible_rows = get_accessible_videos_with_sha1(session_id) if data_origin == "internal" else [] # Rutes base per a media i vídeos pendents base_media_dir = base_dir / "temp" / "media" pending_root = base_dir / "temp" / "pending_videos" with tab_videos: st.subheader("📹 Validar Vídeos Pujats") video_folders = [] # Botó per actualitzar manualment la llista de vídeos pendents des de l'engine col_refresh_list, _ = st.columns([1, 3]) with col_refresh_list: if st.button("🔄 Actualitzar llista de vídeos pendents", key="refresh_pending_videos_list"): st.rerun() if data_origin == "internal": # Mode intern: llistar carpetes de vídeos pendents des de temp/pending_videos if pending_root.exists() and pending_root.is_dir(): for folder in sorted(pending_root.iterdir()): if not folder.is_dir(): continue sha1 = folder.name video_files = list(folder.glob("*.mp4")) + list(folder.glob("*.avi")) + list(folder.glob("*.mov")) if not video_files: continue mod_time = folder.stat().st_mtime fecha = datetime.fromtimestamp(mod_time).strftime("%Y-%m-%d %H:%M") video_folders.append( { "sha1sum": sha1, "video_name": sha1, "path": str(folder), "created_at": fecha, "video_files": video_files, } ) else: # Mode external: llistar vídeos pendents des de l'engine api_client = st.session_state.get("api_client") if api_client is not None: try: resp = api_client.list_pending_videos() _log(f"[pending_videos] list_pending_videos raw resp type= {type(resp)}") _log(f"[pending_videos] list_pending_videos raw resp content= {repr(resp)}") except Exception as e_list: _log(f"[pending_videos] Error cridant list_pending_videos: {e_list}") resp = {"error": "exception"} pending_list = [] if isinstance(resp, dict) and not resp.get("error"): # Pot ser un dict amb clau "videos" o directament una llista if isinstance(resp.get("videos"), list): pending_list = resp["videos"] elif isinstance(resp.get("items"), list): pending_list = resp["items"] elif isinstance(resp.get("results"), list): pending_list = resp["results"] elif isinstance(resp, list): pending_list = resp elif isinstance(resp, list): pending_list = resp _log(f"[pending_videos] parsed pending_list length= {len(pending_list) if isinstance(pending_list, list) else 'N/A'}") if isinstance(pending_list, list) and pending_list: _log(f"[pending_videos] first items: {pending_list[:3]}") for item in pending_list: sha1 = item.get("sha1") or item.get("video_hash") or item.get("id") if not sha1: continue video_name = item.get("latest_video") or sha1 # Carpeta local on descarregarem el vídeo pendent si cal folder = pending_root / sha1 if folder.exists(): video_files = list(folder.glob("*.mp4")) else: video_files = [] created_at = item.get("created_at") or datetime.utcnow().strftime("%Y-%m-%d %H:%M") video_folders.append( { "sha1sum": sha1, "video_name": video_name, "path": str(folder), "created_at": created_at, "video_files": video_files, } ) if not video_folders: st.info("📝 No hi ha vídeos pujats pendents de validació.") else: opciones_video = [f"{video['video_name']} - {video['created_at']}" for video in video_folders] seleccion = st.selectbox( "Selecciona un vídeo per validar:", opciones_video, index=0 if opciones_video else None, ) if seleccion: indice = opciones_video.index(seleccion) video_seleccionat = video_folders[indice] col1, col2 = st.columns([2, 1]) with col1: st.markdown("### 📹 Informació del Vídeo") st.markdown(f"**Nom:** {video_seleccionat['video_name']}") st.markdown(f"**Data:** {video_seleccionat['created_at']}") st.markdown(f"**Arxius:** {len(video_seleccionat['video_files'])} vídeos trobats") # Assegurar que disposem del fitxer local en mode external if data_origin == "external" and not video_seleccionat["video_files"]: api_client = st.session_state.get("api_client") if api_client is not None: try: resp = api_client.download_pending_video(video_seleccionat["sha1sum"]) except Exception: resp = {"error": "exception"} video_bytes = ( resp.get("video_bytes") if isinstance(resp, dict) else None ) if video_bytes: local_folder = pending_root / video_seleccionat["sha1sum"] local_folder.mkdir(parents=True, exist_ok=True) local_path = local_folder / "video.mp4" with local_path.open("wb") as f: f.write(video_bytes) video_seleccionat["video_files"] = [local_path] if video_seleccionat["video_files"]: video_path = str(video_seleccionat["video_files"][0]) st.markdown("**Vídeo principal:**") st.video(video_path) else: st.warning("⚠️ No s'han trobat arxius de vídeo.") with col2: st.markdown("### 🔍 Accions de Validació") col_btn1, col_btn2 = st.columns(2) with col_btn1: if st.button("✅ Acceptar", type="primary", key=f"accept_video_{video_seleccionat['sha1sum']}"): # 1) Registrar decisió al servei de compliance success = compliance_client.record_validator_decision( document_id=f"video_{video_seleccionat['video_name']}", validator_email=f"{username}@veureu.local", decision="acceptat", comments=f"Vídeo validat per {username}", ) # 2) Registrar acció d'acceptació d'input (input-OK) a actions.db session_id = st.session_state.get("session_id") or "" phone = st.session_state.get("phone_number") or "" try: log_action( session=session_id, user=username or "", phone=phone, action="input-OK", sha1sum=video_seleccionat["sha1sum"] or "", ) except Exception: pass if success: st.success("✅ Vídeo acceptat, registrat al servei de compliance i marcat com input-OK") else: st.error("❌ Error registrant el veredicte al servei de compliance") # 3) Moure el vídeo de temp/pending_videos a temp/media (tant en mode internal com external) sha1 = video_seleccionat["sha1sum"] local_pending_dir = pending_root / sha1 local_media_dir = base_media_dir / sha1 try: local_media_dir.mkdir(parents=True, exist_ok=True) src = local_pending_dir / "video.mp4" if src.exists(): dst = local_media_dir / "video.mp4" shutil.copy2(src, dst) if local_pending_dir.exists(): shutil.rmtree(local_pending_dir) except Exception: pass # 4) Actualitzar status="input-OK" a videos.db per aquest sha1 try: update_video_status(sha1, "input-OK") except Exception: pass # 5) Si està habilitat user_sms_enabled, enviar SMS a l'usuari (owner) if user_sms_enabled: try: owner_phone = get_video_owner_by_sha1(sha1) except Exception: owner_phone = "" if owner_phone: try: # Text proporcionat (en castellà, però segons els requisits de l'SMS) msg = "Su vídeo ha sido aprobado. Puede entrar en la aplicación y subirlo de nuevo para generar la audiodescripción" compliance_client.notify_user_video_approved( phone=owner_phone, message=msg, sha1sum=sha1, ) except Exception as e_sms: _log(f"[VIDEO USER SMS] Error enviant SMS a l'usuari: {e_sms}") with col_btn2: if st.button("❌ Rebutjar", type="secondary", key=f"reject_video_{video_seleccionat['video_name']}"): success = compliance_client.record_validator_decision( document_id=f"video_{video_seleccionat['video_name']}", validator_email=f"{username}@veureu.local", decision="rebutjat", comments=f"Vídeo rebutjat per {username}", ) if success: st.success("✅ Vídeo rebutjat i registrat al servei de compliance") else: st.error("❌ Error registrant el veredicte") with tab_ads: st.subheader("🎬 Validar Audiodescripcions") # Llistar vídeos amb status="UNE-pending" a videos.db pending_videos = get_videos_by_status("UNE-pending") if not pending_videos: st.info("📝 No hi ha audiodescripcions pendents de validació UNE.") else: options = [f"{v['video_name']} ({v['sha1sum']})" for v in pending_videos] seleccion_ad = st.selectbox( "Selecciona un vídeo per validar la seva audiodescripció:", options, index=0 if options else None, ) if seleccion_ad: idx = options.index(seleccion_ad) sel = pending_videos[idx] sha1 = sel["sha1sum"] video_name = sel["video_name"] # Permetre escollir versió (Salamandra / MoE) segons el que existeixi a audiodescriptions.db available_versions = [] for v_name in ("Salamandra", "MoE"): rows = get_audiodescription_history(sha1, v_name) if rows: available_versions.append(v_name) if not available_versions: st.error("No s'ha trobat cap audiodescripció per a aquest vídeo a la base de dades.") else: selected_version = st.selectbox( "Selecciona la versió a validar:", available_versions, index=0, key=f"ad_version_{sha1}", ) rows = get_audiodescription_history(sha1, selected_version) or [] row_ad = rows[-1] current_une = row_ad.get("une_ad") or "" current_free = row_ad.get("free_ad") or "" col1, col2 = st.columns([2, 1]) with col1: st.markdown("### 🎬 Informació de l'Audiodescripció") st.markdown(f"**Vídeo:** {video_name}") st.markdown(f"**SHA1:** {sha1}") st.markdown(f"**Versió:** {selected_version}") video_dir = base_media_dir / sha1 video_file = None if video_dir.exists(): for cand in [video_dir / "video.mp4", video_dir / "video.avi", video_dir / "video.mov"]: if cand.exists(): video_file = cand break if video_file is not None: st.video(str(video_file)) else: st.warning("⚠️ No s'ha trobat el fitxer de vídeo original.") st.markdown("#### 📝 Audiodescripció UNE-153010 (SRT)") new_une = st.text_area( "Text SRT UNE-153010", value=current_une, height=220, key=f"une_editor_{sha1}_{selected_version}", ) st.markdown("#### 🗒️ Narració lliure") new_free = st.text_area( "Text narració lliure", value=current_free, height=220, key=f"free_editor_{sha1}_{selected_version}", ) with col2: st.markdown("### 🔍 Accions de Validació") if st.button("✅ Acceptar", type="primary", key=f"accept_ad_{sha1}_{selected_version}"): try: # 1) Registrar decisió al servei de compliance (opcional) try: success = compliance_client.record_validator_decision( document_id=f"ad_{video_name}", validator_email=f"{username}@veureu.local", decision="acceptat", comments=f"Audiodescripció UNE validada per {username}", ) if not success: st.warning("⚠️ No s'ha pogut registrar el veredicte al servei de compliance.") except Exception as e_comp: _log(f"[UNE validation] Error amb compliance: {e_comp}") # 2) Actualitzar camps OK per a aquest vídeo/versió (sense tocar TEST) update_audiodescription_text( sha1sum=sha1, version=selected_version, ok_une_ad=new_une, ok_free_ad=new_free, ) # 3) Actualitzar status a 'UNE-OK' a videos.db try: update_video_status(sha1, "UNE-OK") except Exception as e_stat: _log(f"[UNE validation] Error actualitzant status a UNE-OK: {e_stat}") # 4) Registrar acció 'UNE-OK' a actions.db try: session_id_actions = session_id or "" user_for_action = username or "" phone_for_action = st.session_state.get("phone_number") or "" insert_action( session=session_id_actions, user=user_for_action, phone=phone_for_action, action="UNE-OK", sha1sum=sha1, ) except Exception as e_act: _log(f"[UNE validation] Error registrant acció UNE-OK: {e_act}") # 5) Enviar SMS a l'usuari que va pujar el vídeo, si user_sms_enabled if user_sms_enabled: try: owner_phone = get_video_owner_by_sha1(sha1) except Exception: owner_phone = "" if owner_phone: try: msg = ( "La seva audiodescripció ha estat validada segons la norma UNE-153020. " "Pots tornar a l'aplicació per revisar-la i descarregar-la." ) compliance_client.notify_user_video_approved( phone=owner_phone, message=msg, sha1sum=sha1, ) except Exception as e_sms: _log(f"[UNE USER SMS] Error enviant SMS a l'usuari: {e_sms}") # 6) Generar assets TTS definitius a temp/media//HITL OK try: if new_une.strip(): base_media_dir = base_dir / "temp" / "media" video_dir = base_media_dir / sha1 video_path = None if video_dir.exists(): for cand in [video_dir / "video.mp4", video_dir / "video.avi", video_dir / "video.mov"]: if cand.exists(): video_path = cand break if video_path is not None: hitl_ok_dir = video_dir / "HITL OK" hitl_ok_dir.mkdir(parents=True, exist_ok=True) tts_url = os.getenv("API_TTS_URL", "").strip() if tts_url: with tempfile.TemporaryDirectory(prefix="tts_hitl_ok_") as td: td_path = Path(td) srt_tmp = td_path / "ad_ok.srt" srt_tmp.write_text(new_une, encoding="utf-8") files = { "srt": ("ad_ok.srt", srt_tmp.open("rb"), "text/plain"), "video": (video_path.name, video_path.open("rb"), "video/mp4"), } data = { "voice": "central/grau", "ad_format": "mp3", "include_final_mp4": "1", } resp = requests.post( f"{tts_url.rstrip('/')}/tts/srt", files=files, data=data, timeout=300, ) resp.raise_for_status() zip_bytes = resp.content with zipfile.ZipFile(io.BytesIO(zip_bytes)) as zf: for member in zf.infolist(): name = member.filename lower = name.lower() if lower.endswith("ad_master.mp3"): target = hitl_ok_dir / "free_ad.mp3" with zf.open(member) as src, target.open("wb") as dst: shutil.copyfileobj(src, dst) elif lower.endswith("video_con_ad.mp4"): target = hitl_ok_dir / "une_ad.mp4" with zf.open(member) as src, target.open("wb") as dst: shutil.copyfileobj(src, dst) else: _log("[UNE TTS] API_TTS_URL no configurada; s'omet la generació de free_ad.mp3/une_ad.mp4 (HITL OK)") except Exception as e_tts: _log(f"[UNE TTS] Error generant assets HITL OK: {e_tts}") # 7) Registrar acció de validació a events.db (per traçabilitat) try: log_action( session=session_id or "", user=username or "", phone=st.session_state.get("phone_number") or "", action="validate_ad_une_153020", sha1sum=sha1, ) except Exception: pass st.success("✅ Audiodescripció UNE-153010 validada i desada (HITL OK).") st.rerun() except Exception as e_val: st.error(f"❌ Error durant la validació de l'audiodescripció: {e_val}") st.markdown("---") st.markdown("### ℹ️ Informació del Procés de Validació") st.markdown( """ - **Tots els veredictes** es registren al servei de compliance per garantir la traçabilitat - **Cada validació** inclou veredicte, nom del vídeo i validador responsable - **Els registres** compleixen amb la normativa AI Act i GDPR """ )