Update app.py

app.py

@@ -1,57 +1,90 @@
 import gradio as gr
 import os
 import subprocess
-from huggingface_hub import HfApi, hf_hub_download, list_repo_files
+from huggingface_hub import HfApi, hf_hub_url, list_repo_files, hf_hub_download
 import spaces
 
 api = HfApi()
 
 @spaces.GPU
-def process_batch_videos(token, source_dataset, target_dataset, folder_path):
+def process_full_migration(token, source_dataset, target_dataset, folder_path):
     if not token or not source_dataset or not target_dataset or not folder_path:
-        return "❌ Error: Faltan datos obligatorios."
+        yield "❌ Error: Faltan datos obligatorios."
+        return
         
+    log_output = ""
+    def log(msg):
+        nonlocal log_output
+        log_output += msg + "\n"
+        print(msg, flush=True)
+        yield log_output
+
     try:
         folder_path = folder_path.strip().rstrip('/')
+        yield from log(f"🔍 Escaneando archivos de video (.mp4 y .mkv) en toda la ruta: {folder_path}...")
         
-        # 1. Listar todos los archivos del repo origen
         all_files = list_repo_files(repo_id=source_dataset, repo_type="dataset", token=token)
         
-        # Filtrar los archivos mp4 dentro de la carpeta especificada
-        mp4_files = [
+        video_files = [
             f for f in all_files 
-            if f.startswith(folder_path) and f.endswith('.mp4')
+            if f.startswith(folder_path) and (f.endswith('.mp4') or f.endswith('.mkv'))
         ]
         
-        # Ordenar para consistencia y tomar estrictamente los primeros 3
-        mp4_files = sorted(mp4_files)[:3]
-        
-        if not mp4_files:
-            return f"⚠️ No se encontraron archivos .mp4 en la ruta: {folder_path}"
+        total_files = len(video_files)
+        if total_files == 0:
+            yield from log(f"⚠️ No se encontraron archivos de video en la ruta: {folder_path}")
+            return
             
-        results = []
-        results.append(f"📁 Encontrados {len(mp4_files)} archivos para procesar (Modo WIP - Primeros 3):")
+        yield from log(f"📁 ¡Se encontraron {total_files} videos en total para verificar y migrar!\n")
         
-        for idx, file_path in enumerate(mp4_files, 1):
-            results.append(f"\n--- Procesando [{idx}/3]: {file_path} ---")
+        for idx, file_path in enumerate(video_files, 1):
+            yield from log(f"==================================================")
+            yield from log(f"🎬 Procesando [{idx}/{total_files}]: {file_path}")
+            yield from log(f"==================================================")
+            
+            is_mkv = file_path.endswith('.mkv')
+            temp_local_path = "temp_download.mkv" if is_mkv else "temp_download.mp4"
+            output_ext = ".mkv" if is_mkv else ".mp4"
+            output_path = f"video_arreglado{output_ext}"
+            
+            file_url = hf_hub_url(repo_id=source_dataset, filename=file_path, repo_type="dataset")
             
-            # Descargar archivo original
-            local_path = hf_hub_download(
+            # Descarga estándar por caché de Hugging Face (se aloja temporalmente en el disco local de la VPS)
+            yield from log("📥 Descargando archivo al disco local de la VPS...")
+            temp_local_path = hf_hub_download(
                 repo_id=source_dataset, 
                 filename=file_path, 
                 repo_type="dataset", 
                 token=token
             )
             
-            # Procesar con FFmpeg (Remux y Faststart)
-            output_path = f"video_arreglado_{idx}.mp4"
-            command = [
-                "ffmpeg", "-y", "-i", local_path, 
-                "-c", "copy", "-movflags", "+faststart", output_path
-            ]
-            subprocess.run(command, check=True)
+            # Procesar con FFmpeg
+            if is_mkv:
+                yield from log("⚙️ Verificando y limpiando contenedor MKV con FFmpeg (-c copy)...")
+                command = [
+                    "ffmpeg", "-y", "-i", temp_local_path, 
+                    "-c", "copy", output_path
+                ]
+            else:
+                yield from log("⚙️ Aplicando fix moov atom a MP4/TS con FFmpeg (-c copy -movflags +faststart)...")
+                command = [
+                    "ffmpeg", "-y", "-i", temp_local_path, 
+                    "-c", "copy", "-movflags", "+faststart", output_path
+                ]
+                
+            try:
+                subprocess.run(command, check=True)
+            except subprocess.CalledProcessError:
+                yield from log("⚠️ ¡Aviso! El copiado directo falló debido a corrupción interna en el archivo. Intentando re-codificación de emergencia...")
+                if is_mkv:
+                    command_fallback = ["ffmpeg", "-y", "-i", temp_local_path, output_path]
+                else:
+                    command_fallback = ["ffmpeg", "-y", "-i", temp_local_path, "-movflags", "+faststart", output_path]
+                subprocess.run(command_fallback, check=True)
+                yield from log("✅ Re-codificación de emergencia completada con éxito.")
             
-            # Subir al dataset destino limpio
+            # Subir el archivo resultante limpio al dataset de destino
+            yield from log(f"☁️ Subiendo video optimizado a {target_dataset} ({file_path})...")
             api.upload_file(
                 path_or_fileobj=output_path,
                 path_in_repo=file_path,
@@ -60,22 +93,21 @@ def process_batch_videos(token, source_dataset, target_dataset, folder_path):
                 token=token
             )
             
-            # Limpieza local
+            # Limpieza local exhaustiva del disco temporal generado
             if os.path.exists(output_path):
                 os.remove(output_path)
                 
-            results.append(f"✅ ¡Reparado y subido con éxito a {target_dataset}!")
+            yield from log(f"🎉 ¡Video [{idx}/{total_files}] procesado y subido con éxito!\n")
             
-        results.append("\n🎉 ¡Lote de prueba de los primeros 3 completado exitosamente!")
-        return "\n".join(results)
+        yield from log("🏆 ¡Recorrido masivo total finalizado! Ningún archivo fue omitido y todos están limpios.")
         
     except Exception as e:
-        return f"❌ Algo falló en el proceso por lotes: {str(e)}"
+        yield from log(f"❌ Error crítico en el recorrido: {str(e)}")
 
-# Interfaz gráfica adaptada para procesamiento por lotes (WIP)
+# Interfaz gráfica de control total con botón de Cancelar/Detener
 with gr.Blocks() as demo:
-    gr.Markdown("# 🛠️ Migrador y Reparador Masivo (Modo WIP - Primeros 3)")
-    gr.Markdown("Escanea la carpeta, verifica y repara únicamente los primeros 3 archivos MP4 encontrados.")
+    gr.Markdown("# 🛠️ Migrador y Reparador Masivo Total (.MP4, .MKV y Fakes TS)")
+    gr.Markdown("Recorre de forma recursiva todas las temporadas, descarga al disco local, repara fallos y migra limpio a tu organización.")
     
     token_input = gr.Textbox(label="Tu HF_TOKEN (Con permisos de WRITE)", type="password")
     
@@ -84,13 +116,20 @@ with gr.Blocks() as demo:
         target_input = gr.Textbox(label="Dataset DESTINO", value="orglatamsrc/latamsrcfixed")
     
     folder_input = gr.Textbox(
-        label="Carpeta a escanear", 
-        value="SERIES/EN_EMISION/Rivales_2024/TEMPORADA1"
+        label="Carpeta principal a escanear", 
+        value="SERIES/EN_EMISION/Rivales_2024"
     )
     
-    btn = gr.Button("Verificar y Reparar Primeros 3 🚀", variant="primary")
-    output = gr.Textbox(label="Estado del proceso / Log", lines=15)
+    with gr.Row():
+        btn = gr.Button("Iniciar Migración Masiva Total 🔥", variant="primary")
+        stop_btn = gr.Button("Detener Proceso ⏹️", variant="stop")
+        
+    output = gr.Textbox(label="Consola de Logs en Tiempo Real", lines=22)
+    
+    # Evento de inicio del proceso
+    watcher = btn.click(fn=process_full_migration, inputs=[token_input, source_input, target_input, folder_input], outputs=output)
     
-    btn.click(fn=process_batch_videos, inputs=[token_input, source_input, target_input, folder_input], outputs=output)
+    # Evento del botón de detener para cancelar la ejecución al instante si lo deseas
+    stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[watcher])
 
-demo.launch(theme=gr.themes.Soft())
+demo.launch(theme=gr.themes.Soft(), server_name="0.0.0.0", server_port=7860)