Restore probe

app.py

@@ -313,6 +313,21 @@ def conditioner():
     return Client(CONDITIONER_SPACE)
 
 
+def probe(path: str) -> tuple[float | None, float | None]:
+    """`(video seconds, audio seconds)` of a media file, either being `None` when the stream is absent."""
+    import av
+
+    def seconds(stream, container):
+        if stream.duration is not None and stream.time_base is not None:
+            return float(stream.duration * stream.time_base)
+        return None if container.duration is None else container.duration / av.time_base
+
+    with av.open(path) as container:
+        video = seconds(container.streams.video[0], container) if container.streams.video else None
+        audio = seconds(container.streams.audio[0], container) if container.streams.audio else None
+    return video, audio
+
+
 def collect(image_paths, audio_path, video_path) -> list[tuple[str, str]]:
     """The `(kind, path)` references of a request, **in the order the model reads them**.