Update app.py

app.py

@@ -16,44 +16,19 @@ st.set_page_config(
 )
 
 # --- IDENTIDAD CORPORATIVA, ESTILOS Y SCRIPTS ---
-
-# 1. Logo de la empresa en Base64
-# Asegúrate de que esta variable contiene tu logo en Base64
 logo_base64 = "..." # Pega aquí el código base64 de tu logo
-
-# 2. Estilos y el Script de copiado
 st.markdown(f"""
     <style>
-    :root {{
-        --gunmetal: #14323C;
-        --cambridge-blue: #94CBB1;
-        --charcoal: #2F4952;
-        --silver: #C1C7CA;
-        --white: #FFFFFF;
-    }}
-    .main {{ background-color: #F0F2F6; color: var(--charcoal); }}
-    .stApp {{ max-width: 800px; margin: auto; }}
-    .stButton>button {{
-        width: 100%; background-color: var(--gunmetal); color: var(--white);
-        border: none; padding: 12px; border-radius: 8px; font-size: 16px; font-weight: bold;
-    }}
+    /* ... (tu CSS aquí, no es necesario cambiarlo) ... */
+    :root {{ --gunmetal: #14323C; --cambridge-blue: #94CBB1; --charcoal: #2F4952; --silver: #C1C7CA; --white: #FFFFFF; }}
+    .main {{ background-color: #F0F2F6; color: var(--charcoal); }} .stApp {{ max-width: 800px; margin: auto; }}
+    .stButton>button {{ width: 100%; background-color: var(--gunmetal); color: var(--white); border: none; padding: 12px; border-radius: 8px; font-size: 16px; font-weight: bold; }}
     .stButton>button:hover {{ background-color: var(--charcoal); color: var(--white); }}
-    .logo-container {{
-        display: flex; justify-content: center; align-items: center; padding: 10px;
-        background-color: var(--white); border-radius: 10px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);
-        margin-bottom: 2rem;
-    }}
-    .logo-container img {{ width: 200px; }}
-    .stProgress > div > div > div > div {{ background-color: var(--cambridge-blue); }}
-    
-    .copy-btn {{
-        background-color: var(--silver); color: var(--charcoal); border: none;
-        padding: 5px 10px; border-radius: 5px; font-size: 14px; font-weight: bold;
-        cursor: pointer; transition: background-color 0.2s; float: right;
-    }}
+    .logo-container {{ display: flex; justify-content: center; align-items: center; padding: 10px; background-color: var(--white); border-radius: 10px; box-shadow: 0 4px 8px rgba(0,0,0,0.1); margin-bottom: 2rem; }}
+    .logo-container img {{ width: 200px; }} .stProgress > div > div > div > div {{ background-color: var(--cambridge-blue); }}
+    .copy-btn {{ background-color: var(--silver); color: var(--charcoal); border: none; padding: 5px 10px; border-radius: 5px; font-size: 14px; font-weight: bold; cursor: pointer; transition: background-color 0.2s; float: right; }}
     .copy-btn:hover {{ background-color: var(--cambridge-blue); }}
     </style>
-
     <script>
     function copyToClipboard(text, buttonId) {{
         navigator.clipboard.writeText(text).then(function() {{
@@ -66,43 +41,25 @@ st.markdown(f"""
     </script>
 """, unsafe_allow_html=True)
 
-# 3. Mostrar el logo y título
 if logo_base64 != "...":
-    # Muestra el logo solo si se ha pegado el código base64
-    st.markdown(f"""
-        <div class="logo-container">
-            <img src="data:image/png;base64,{logo_base64}" alt="Logo de la empresa">
-        </div>
-    """, unsafe_allow_html=True)
+    st.markdown(f'<div class="logo-container"><img src="data:image/png;base64,{logo_base64}" alt="Logo de la empresa"></div>', unsafe_allow_html=True)
 
 st.title("Transcriptor Profesional de Vídeos")
 st.markdown("Pega las URLs de los vídeos de Vimeo (públicos o privados) que quieres transcribir. **Límite de 30 minutos por vídeo.**")
 
-# --- Funciones Principales ---
 @st.cache_resource(show_spinner="Cargando modelo de IA por primera vez...")
 def load_whisper_model(model_name):
-    model = whisper.load_model(model_name, device="cpu")
-    return model
+    return whisper.load_model(model_name, device="cpu")
 
-# --- Interfaz de Usuario ---
-urls_input = st.text_area(
-    "📝 Pega aquí las URLs de Vimeo (una por línea):",
-    height=100,
-    placeholder="https://vimeo.com/..."
-)
+urls_input = st.text_area("📝 Pega aquí las URLs de Vimeo (una por línea):", height=100, placeholder="https://vimeo.com/...")
 col1, col2 = st.columns(2)
 with col1:
     whisper_model = st.selectbox("🤖 Modelo de IA:", ["tiny", "base", "small"], index=1, help="El modelo 'base' ofrece un excelente equilibrio entre velocidad y precisión.")
 with col2:
     force_language = st.checkbox("🇪🇸 Forzar idioma español", value=True, help="Márcalo si todos los vídeos están en español.")
 
-# --- Lógica de Procesamiento ---
 if st.button("🚀 Iniciar Transcripción"):
     urls = [line.strip() for line in urls_input.splitlines() if line.strip()]
-    total_urls = len(urls)
-    
-    MAX_DURATION_SECONDS = 1800
-
     if not urls:
         st.error("❌ Por favor, introduce al menos una URL de Vimeo.")
     else:
@@ -114,50 +71,44 @@ if st.button("🚀 Iniciar Transcripción"):
                 st.markdown("---")
                 progress_bar = st.progress(0, text="Progreso general")
                 
-                for i, url in enumerate(urls):
-                    st.markdown(f"### 🎬 Procesando Vídeo {i+1} de {total_urls}")
+                for i, raw_url in enumerate(urls):
+                    st.markdown(f"### 🎬 Procesando Vídeo {i+1} de {len(urls)}")
                     status_placeholder = st.empty()
                     audio_file_path = None
                     video_title = "Título desconocido"
                     
                     try:
-                        # --- INICIO DE LA MODIFICACIÓN CLAVE ---
-                        # Las opciones de ydl se definen DENTRO del bucle
-                        # para usar la 'url' específica como 'Referer'.
+                        # Limpiar la URL de parámetros innecesarios
+                        url = raw_url.split('?')[0]
+
+                        # --- LA SOLUCIÓN DEFINITIVA ---
+                        # Se añade el parámetro 'impersonate' para simular un navegador real
                         ydl_opts = {
                             'format': 'bestaudio/best',
                             'outtmpl': os.path.join(temp_dir, '%(title)s.%(ext)s'),
                             'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3', 'preferredquality': '64'}],
-                            'quiet': True,
-                            'no_warnings': True,
-                            'retries': 2,
-                            'socket_timeout': 30,
-                            'http_headers': {
-                                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-                                'Authorization': 'bearer 27d734898683a4b1f863c3453b3b4331',
-                                'Referer': url # El Referer ahora es la URL del vídeo
-                            },
+                            'quiet': True, 'no_warnings': True, 'retries': 2, 'socket_timeout': 30,
+                            'impersonate': 'chrome120', # <-- ¡ESTA ES LA LÍNEA CLAVE!
                             'postprocessor_args': ['-ar', '16000', '-ac', '1']
                         }
-                        # --- FIN DE LA MODIFICACIÓN CLAVE ---
                         
                         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                            status_placeholder.info("1/5 - Obteniendo información del vídeo...")
+                            status_placeholder.info("1/5 - Obteniendo información del vídeo (modo suplantación)...")
                             info_dict = ydl.extract_info(url, download=False)
                             video_title = info_dict.get('title', 'Título desconocido')
                             duration = info_dict.get('duration')
 
+                            MAX_DURATION_SECONDS = 1800
                             if not duration or duration > MAX_DURATION_SECONDS:
-                                status_placeholder.warning(f"⚠️ **'{video_title}' omitido:** El vídeo es demasiado largo (límite {int(MAX_DURATION_SECONDS/60)} min) o no se pudo obtener su duración.")
+                                status_placeholder.warning(f"⚠️ **'{video_title}' omitido:** El vídeo es demasiado largo (límite {int(MAX_DURATION_SECONDS/60)} min).")
                                 continue
 
                             status_placeholder.info(f"2/5 - Descargando audio de '{video_title}'...")
                             ydl.download([url])
                             
                             audio_file_path = next((os.path.join(temp_dir, f) for f in os.listdir(temp_dir) if f.endswith('.mp3')), None)
-
                             if not audio_file_path or os.path.getsize(audio_file_path) < 1024:
-                                status_placeholder.error(f"❌ **Error con '{video_title}':** La descarga falló o el archivo de audio está vacío. Se omite.")
+                                status_placeholder.error(f"❌ **Error con '{video_title}':** La descarga falló. Se omite.")
                                 continue
 
                             status_placeholder.info(f"3/5 - Verificando integridad del audio...")
@@ -168,37 +119,31 @@ if st.button("🚀 Iniciar Transcripción"):
                                 status_placeholder.error(f"❌ **Error con '{video_title}':** No se pudo decodificar el audio. Se omite.")
                                 continue
 
-                        status_placeholder.info(f"4/5 - Transcribiendo '{video_title}'...")
-                        start_time = time.time()
-                        
-                        result = model.transcribe(audio_file_path, fp16=False, language="es" if force_language else None, task="transcribe")
-                        transcription_text = result['text']
-                        
-                        status_placeholder.success(f"5/5 - ✅ Transcripción de '{video_title}' completada en {time.time() - start_time:.2f} segundos.")
-                        
-                        with st.expander(f"Ver Transcripción de '{video_title}'", expanded=True):
-                            js_text = transcription_text.replace("`", "\\`").replace("'", "\\'").replace("\n", "\\n")
-                            button_id = f"copy-btn-{i}"
-                            button_html = f'<button id="{button_id}" class="copy-btn" onclick="copyToClipboard(`{js_text}`, \'{button_id}\')">Copiar</button>'
-                            st.markdown(button_html, unsafe_allow_html=True)
+                            status_placeholder.info(f"4/5 - Transcribiendo '{video_title}'...")
+                            result = model.transcribe(audio_file_path, fp16=False, language="es" if force_language else None, task="transcribe")
+                            transcription_text = result['text']
                             
-                            st.text_area("Transcripción:", value=transcription_text, height=200, key=f"text_{i}", label_visibility="collapsed")
-                        
-                        clean_title = re.sub(r'[\\/*?:"<>|]', "", video_title)
-                        with open(os.path.join(transcriptions_dir, f"{clean_title}.txt"), "w", encoding="utf-8") as f: f.write(transcription_text)
+                            status_placeholder.success(f"5/5 - ✅ Transcripción de '{video_title}' completada.")
+                            
+                            with st.expander(f"Ver Transcripción de '{video_title}'", expanded=True):
+                                # ... (código para mostrar y copiar texto, sin cambios)
+                                js_text = transcription_text.replace("`", "\\`").replace("'", "\\'").replace("\n", "\\n")
+                                button_id = f"copy-btn-{i}"
+                                st.markdown(f'<button id="{button_id}" class="copy-btn" onclick="copyToClipboard(`{js_text}`, \'{button_id}\')">Copiar</button>', unsafe_allow_html=True)
+                                st.text_area("Transcripción:", value=transcription_text, height=200, key=f"text_{i}", label_visibility="collapsed")
+                            
+                            clean_title = re.sub(r'[\\/*?:"<>|]', "", video_title)
+                            with open(os.path.join(transcriptions_dir, f"{clean_title}.txt"), "w", encoding="utf-8") as f: f.write(transcription_text)
                         
                     except Exception as e:
-                        if "cannot reshape tensor of 0 elements" in str(e):
-                            status_placeholder.error(f"❌ **Error con '{video_title}':** La IA no pudo procesar el audio (posiblemente silencio). Se omite.")
-                        else:
-                            status_placeholder.error(f"❌ Error inesperado al procesar el vídeo '{video_title}'. Causa: {e}")
+                        status_placeholder.error(f"❌ Error al procesar '{video_title}'. Causa: {e}")
                     
                     finally:
-                        if audio_file_path and os.path.exists(audio_file_path):
-                            os.remove(audio_file_path)
+                        if audio_file_path and os.path.exists(audio_file_path): os.remove(audio_file_path)
                     
-                    progress_bar.progress((i + 1) / total_urls, text=f"Progreso: {i+1}/{total_urls} vídeos completados")
+                    progress_bar.progress((i + 1) / len(urls), text=f"Progreso: {i+1}/{len(urls)} vídeos completados")
 
+                # ... (resto del código para el zip, sin cambios)
                 st.markdown("---")
                 transcription_files = os.listdir(transcriptions_dir)
                 if transcription_files:
@@ -208,6 +153,6 @@ if st.button("🚀 Iniciar Transcripción"):
                     with open(f"{zip_path}.zip", "rb") as fp:
                         st.download_button(label=f"📥 Descargar {len(transcription_files)} transcripciones (.zip)", data=fp, file_name="transcripciones_vimeo.zip", mime="application/zip")
                 else:
-                    st.warning("⚠️ No se generó ninguna transcripción. Revisa los mensajes de error de cada vídeo.")
+                    st.warning("⚠️ No se generó ninguna transcripción.")
         except Exception as e:
             st.error(f"🚨 Ha ocurrido un error crítico durante la configuración: {e}")