Made project river-agnostic

src/clip_annotator/video_loader.py

@@ -0,0 +1,56 @@
+import io
+import os
+import tempfile
+import zipfile
+
+import cv2
+
+
+def load_frames(
+    zip_path,
+    max_frames: int,
+    display_max: int,
+    fps_fallback: int,
+    video_in_zip: str = "left.mp4",
+    video_tmp_suffix: str = ".mp4",
+    fs=None,
+):
+    if fs is None:
+        video_bytes = zipfile.ZipFile(zip_path).read(video_in_zip)
+    else:
+        with fs.open(str(zip_path), "rb") as f:
+            video_bytes = zipfile.ZipFile(io.BytesIO(f.read())).read(video_in_zip)
+
+    with tempfile.NamedTemporaryFile(suffix=video_tmp_suffix, delete=False) as f:
+        f.write(video_bytes)
+        tmp_path = f.name
+
+    cap = cv2.VideoCapture(tmp_path)
+    fps = cap.get(cv2.CAP_PROP_FPS) or fps_fallback
+
+    total = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    step = max(1, total // max_frames)
+
+    frames = []
+    i = 0
+    while len(frames) < max_frames:
+        cap.set(cv2.CAP_PROP_POS_FRAMES, i)
+        ok, frame = cap.read()
+        if not ok:
+            break
+        frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+        i += step
+
+    cap.release()
+    os.unlink(tmp_path)
+
+    if not frames:
+        raise RuntimeError(f"No frames found in {zip_path}")
+
+    h, w = frames[0].shape[:2]
+    scale = display_max / max(h, w)
+    dh, dw = int(h * scale), int(w * scale)
+
+    frames = [cv2.resize(f, (dw, dh)) for f in frames]
+
+    return frames, fps, dh, dw, h, w