Show pipeline progress in Gradio UI

app.py

@@ -24,7 +24,8 @@ APP_DESCRIPTION = (
 def _prepare_config(config_file: str | None, work_dir: Path) -> Path:
     cfg = load_config(config_file)
     cfg.setdefault("output", {})
-    cfg["output"]["progress_enabled"] = False
+    cfg["output"]["progress_enabled"] = True
+    cfg["output"]["progress_step_percent"] = 2
 
     config_path = work_dir / "space_config.yaml"
     with config_path.open("w", encoding="utf-8") as handle:
@@ -50,7 +51,7 @@ def _copy_input_file(src: str, dst_dir: Path) -> Path:
     return dst_path
 
 
-def process_video(video_file: str | None, config_file: str | None):
+def process_video(video_file: str | None, config_file: str | None, progress=gr.Progress()):
     if not video_file:
         raise gr.Error("Debes subir un video de entrada.")
 
@@ -62,12 +63,19 @@ def process_video(video_file: str | None, config_file: str | None):
     work_dir.mkdir(parents=True, exist_ok=True)
 
     try:
+        progress(0, desc="Preparando ejecución")
         input_path = _copy_input_file(video_file, work_dir)
         config_path = _prepare_config(config_file, work_dir)
+
+        def on_pipeline_progress(pct: int, detail: str | None) -> None:
+            description = detail or "Procesando"
+            progress(min(max(pct / 100.0, 0.0), 1.0), desc=description)
+
         meta = run_pipeline(
             input_video=str(input_path),
             output_dir=str(output_dir),
             config_path=str(config_path),
+            progress_callback=on_pipeline_progress,
         )
 
         outputs = meta.get("outputs", {})

bat_tracker/pipeline.py

@@ -9,7 +9,7 @@ from math import ceil
 from math import hypot
 from pathlib import Path
 from statistics import mean
-from typing import Dict, List
+from typing import Callable, Dict, List
 
 import cv2
 import numpy as np
@@ -207,12 +207,19 @@ class TemporalBurstGate:
 
 
 class ProgressReporter:
-    def __init__(self, enabled: bool, step_percent: int, stages: list[tuple[str, float]]):
+    def __init__(
+        self,
+        enabled: bool,
+        step_percent: int,
+        stages: list[tuple[str, float]],
+        callback: Callable[[int, str | None], None] | None = None,
+    ):
         self.enabled = bool(enabled) and bool(stages)
         self.step_percent = max(1, min(100, int(step_percent)))
         self._next_threshold = self.step_percent
         self._last_reported = 0
         self._current_stage = ""
+        self._callback = callback
 
         total_weight = sum(max(0.0, float(weight)) for _, weight in stages)
         if total_weight <= 0:
@@ -252,6 +259,8 @@ class ProgressReporter:
         self._emit(pct, detail=detail)
 
     def _emit(self, pct: int, detail: str | None = None) -> None:
+        if self._callback is not None:
+            self._callback(pct, detail or self._current_stage or None)
         if pct < self._next_threshold:
             return
         self._last_reported = pct
@@ -644,7 +653,12 @@ def _export_track_clips(
     return clip_paths
 
 
-def run_pipeline(input_video: str, output_dir: str, config_path: str | None = None) -> Dict:
+def run_pipeline(
+    input_video: str,
+    output_dir: str,
+    config_path: str | None = None,
+    progress_callback: Callable[[int, str | None], None] | None = None,
+) -> Dict:
     cfg = load_config(config_path)
     execution_plan = build_execution_plan(cfg)
     execution_cfg = cfg.get("execution", {})
@@ -681,6 +695,7 @@ def run_pipeline(input_video: str, output_dir: str, config_path: str | None = No
             ("exports_core", 12.0),
             ("track_clips", 10.0 if export_track_clips_enabled else 0.0),
         ],
+        callback=progress_callback,
     )
 
     progress.start_stage("background")