Upload 5 files

page_modules/analyze_transcriptions.py

@@ -2,14 +2,72 @@
 
 from __future__ import annotations
 
+import csv
 from pathlib import Path
-from typing import Dict
+from typing import Dict, Optional
 
 import streamlit as st
 
 from utils import save_bytes
 
 
+def load_eval_values(vid_dir: Path, version: str) -> Optional[Dict[str, int]]:
+    """Carga los valores de evaluación desde eval.csv si existe.
+    
+    Args:
+        vid_dir: Directorio del vídeo
+        version: Versión seleccionada (MoE/Salamandra)
+        
+    Returns:
+        Diccionario con los valores de evaluación o None si no existe el CSV
+    """
+    csv_path = vid_dir / version / "eval.csv"
+    
+    if not csv_path.exists():
+        return None
+    
+    try:
+        with open(csv_path, 'r', encoding='utf-8') as f:
+            reader = csv.DictReader(f)
+            row = next(reader, None)
+            
+            if not row:
+                return None
+            
+            # Mapeo de nombres de columnas CSV a claves internas
+            # Usamos los nombres nuevos del CSV
+            mappings = {
+                'transcripcio': ['Precisió Descriptiva', 'Precisió Descriptiva'],
+                'identificacio': ['Sincronització Temporal', 'Sincronització Temporal'],
+                'localitzacions': ['Claredat i Concisió', 'Claredat i Concisió'],
+                'activitats': ['Inclusió de Diàleg/So', 'Inclusió de Diàleg', 'Inclusió de Diàleg/So'],
+                'narracions': ['Contextualització', 'Contextualització'],
+                'expressivitat': ['Flux i Ritme de la Narració', 'Flux i Ritme de la Narració']
+            }
+            
+            values = {}
+            for key, possible_names in mappings.items():
+                for name in possible_names:
+                    if name in row and row[name]:
+                        try:
+                            # Convertir a int y asegurar que está en rango 0-7
+                            val = int(float(row[name]))
+                            values[key] = max(0, min(7, val))
+                            break
+                        except (ValueError, TypeError):
+                            continue
+                
+                # Si no se encontró valor, usar 7 por defecto
+                if key not in values:
+                    values[key] = 7
+            
+            return values
+            
+    except Exception as e:
+        # Si hay cualquier error, simplemente ignorar y devolver None
+        return None
+
+
 def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> None:
     st.header("Analitzar video-transcripcions")
     base_dir = Path("/tmp/data/videos")
@@ -33,6 +91,9 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
         if "version_selector" in st.session_state:
             del st.session_state["version_selector"]
         st.session_state.add_ad_checkbox = False
+        # Limpiar valores de evaluación al cambiar de vídeo
+        if "eval_values" in st.session_state:
+            del st.session_state["eval_values"]
         st.rerun()
 
     if not seleccio:
@@ -53,6 +114,17 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
             placeholder="Tria una versió…" if subcarpetas_ad else "No hi ha versions",
             key="version_selector",
         )
+        
+        # Cargar valores de evaluación cuando cambia la versión
+        if subcarpeta_seleccio:
+            current_version_key = f"{seleccio}_{subcarpeta_seleccio}"
+            if "last_version_key" not in st.session_state or st.session_state.last_version_key != current_version_key:
+                st.session_state.last_version_key = current_version_key
+                eval_values = load_eval_values(vid_dir, subcarpeta_seleccio)
+                if eval_values:
+                    st.session_state.eval_values = eval_values
+                elif "eval_values" in st.session_state:
+                    del st.session_state["eval_values"]
 
         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()
@@ -147,14 +219,30 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
             key=f"editor_{seleccio}_{subcarpeta_seleccio}_{ad_filename}",
         )
 
-        if st.button(
-            "▶️ Reproduir narració",
-            use_container_width=True,
-            disabled=not new_text.strip(),
-            key="play_button_editor",
-        ):
-            with st.spinner("Generant àudio..."):
-                pass
+        # Solo mostrar botón de reproducción para narración libre
+        if tipus_ad_seleccio == "narració lliure":
+            if st.button(
+                "▶️ Reproduir narració",
+                use_container_width=True,
+                disabled=not new_text.strip(),
+                key="play_button_editor",
+            ):
+                with st.spinner("Generant àudio..."):
+                    try:
+                        # Generar audio a partir del texto editado
+                        voice = "central/grau"
+                        response = api.tts_matxa(text=new_text.strip(), voice=voice)
+                        
+                        if "mp3_bytes" in response:
+                            # Mostrar el reproductor de audio
+                            st.audio(response["mp3_bytes"], format="audio/mp3")
+                            st.success("✅ Àudio generat correctament")
+                        else:
+                            st.error(f"Error en la generació de l'àudio: {response.get('error', 'Desconegut')}")
+                    except Exception as e:
+                        st.error(f"S'ha produït un error: {e}")
+        else:
+            st.info("📝 L'audiodescripció UNE-153010 està en format SRT. Per reproduir-la, genera el vídeo amb el botó de reconstrucció.")
 
         if st.button("Desar canvis", use_container_width=True, type="primary"):
             if ad_path:
@@ -172,17 +260,50 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
 
     can_rate = permissions.get("valorar", False)
     controls_disabled = not can_rate
+    
+    # Obtener valores por defecto desde el CSV si existen
+    eval_values = st.session_state.get("eval_values", {})
 
     c1, c2, c3 = st.columns(3)
     with c1:
-        transcripcio = st.slider("Transcripció", 1, 10, 7, disabled=controls_disabled)
-        identificacio = st.slider("Identificació de personatges", 1, 10, 7, disabled=controls_disabled)
+        transcripcio = st.slider(
+            "Precisió Descriptiva", 
+            0, 7, 
+            eval_values.get('transcripcio', 7), 
+            disabled=controls_disabled
+        )
+        identificacio = st.slider(
+            "Sincronització Temporal", 
+            0, 7, 
+            eval_values.get('identificacio', 7), 
+            disabled=controls_disabled
+        )
     with c2:
-        localitzacions = st.slider("Localitzacions", 1, 10, 7, disabled=controls_disabled)
-        activitats = st.slider("Activitats", 1, 10, 7, disabled=controls_disabled)
+        localitzacions = st.slider(
+            "Claredat i Concisió", 
+            0, 7, 
+            eval_values.get('localitzacions', 7), 
+            disabled=controls_disabled
+        )
+        activitats = st.slider(
+            "Inclusió de Diàleg", 
+            0, 7, 
+            eval_values.get('activitats', 7), 
+            disabled=controls_disabled
+        )
     with c3:
-        narracions = st.slider("Narracions", 1, 10, 7, disabled=controls_disabled)
-        expressivitat = st.slider("Expressivitat", 1, 10, 7, disabled=controls_disabled)
+        narracions = st.slider(
+            "Contextualització", 
+            0, 7, 
+            eval_values.get('narracions', 7), 
+            disabled=controls_disabled
+        )
+        expressivitat = st.slider(
+            "Flux i Ritme de la Narració", 
+            0, 7, 
+            eval_values.get('expressivitat', 7), 
+            disabled=controls_disabled
+        )
 
     comments = st.text_area(
         "Comentaris (opcional)",
@@ -197,7 +318,8 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
         if st.button("Enviar valoració", type="primary", use_container_width=True):
             try:
                 from database import add_feedback_ad
-
+                
+                # Guardar en la base de datos
                 add_feedback_ad(
                     video_name=seleccio,
                     user_id=st.session_state.user["id"],
@@ -209,6 +331,27 @@ def render_analyze_transcriptions_page(api, permissions: Dict[str, bool]) -> Non
                     expressivitat=expressivitat,
                     comments=comments or None,
                 )
+                
+                # También guardar en CSV
+                video_dir = Path("demo/videos") / seleccio
+                version = st.session_state.get("selected_version", "MoE")
+                csv_path = video_dir / version / "eval.csv"
+                
+                csv_data = [
+                    {"Caracteristica": "Precisió Descriptiva", "Valoracio (0-7)": str(transcripcio), "Justificacio": comments or ""},
+                    {"Caracteristica": "Sincronització Temporal", "Valoracio (0-7)": str(identificacio), "Justificacio": comments or ""},
+                    {"Caracteristica": "Claredat i Concisió", "Valoracio (0-7)": str(localitzacions), "Justificacio": comments or ""},
+                    {"Caracteristica": "Inclusió de Diàleg/So", "Valoracio (0-7)": str(activitats), "Justificacio": comments or ""},
+                    {"Caracteristica": "Contextualització", "Valoracio (0-7)": str(narracions), "Justificacio": comments or ""},
+                    {"Caracteristica": "Flux i Ritme de la Narració", "Valoracio (0-7)": str(expressivitat), "Justificacio": comments or ""}
+                ]
+                
+                csv_path.parent.mkdir(parents=True, exist_ok=True)
+                with open(csv_path, 'w', newline='', encoding='utf-8') as f:
+                    writer = csv.DictWriter(f, fieldnames=["Caracteristica", "Valoracio (0-7)", "Justificacio"])
+                    writer.writeheader()
+                    writer.writerows(csv_data)
+                
                 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}")