Upload 2 files

app.py

@@ -0,0 +1,105 @@
+{\rtf1\ansi\ansicpg1252\cocoartf2822
+\cocoatextscaling0\cocoaplatform0{\fonttbl\f0\fswiss\fcharset0 Helvetica;}
+{\colortbl;\red255\green255\blue255;}
+{\*\expandedcolortbl;;}
+\paperw11900\paperh16840\margl1440\margr1440\vieww11520\viewh8400\viewkind0
+\pard\tx720\tx1440\tx2160\tx2880\tx3600\tx4320\tx5040\tx5760\tx6480\tx7200\tx7920\tx8640\pardirnatural\partightenfactor0
+
+\f0\fs24 \cf0 import cv2\
+import zipfile\
+import os\
+import tempfile\
+import gradio as gr\
+\
+\
+def extract_frames_to_zip(video_path, num_frames=10):\
+    cap = cv2.VideoCapture(video_path)\
+    if not cap.isOpened():\
+        raise RuntimeError(f"Could not open video: \{video_path\}")\
+\
+    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))\
+    if frame_count <= 0:\
+        raise RuntimeError("Could not determine frame count.")\
+\
+    num_frames = min(num_frames, frame_count)\
+\
+    if num_frames == 1:\
+        frame_indices = [0]\
+    else:\
+        frame_indices = [\
+            round(i * (frame_count - 1) / (num_frames - 1))\
+            for i in range(num_frames)\
+        ]\
+\
+    # Create a temp directory for this run\
+    tmp_dir = tempfile.mkdtemp()\
+    zip_path = os.path.join(tmp_dir, "frames.zip")\
+\
+    with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_DEFLATED) as zf:\
+        for idx, frame_idx in enumerate(frame_indices):\
+            cap.set(cv2.CAP_PROP_POS_FRAMES, frame_idx)\
+            ret, frame = cap.read()\
+            if not ret:\
+                print(f"Warning: Could not read frame at index \{frame_idx\}")\
+                continue\
+\
+            success, buffer = cv2.imencode(".jpg", frame)\
+            if not success:\
+                print(f"Warning: Could not encode frame at index \{frame_idx\}")\
+                continue\
+\
+            filename_in_zip = f"frame_\{idx:02d\}.jpg"\
+            zf.writestr(filename_in_zip, buffer.tobytes())\
+\
+    cap.release()\
+    return zip_path\
+\
+\
+def gradio_fn(video_file, num_frames):\
+    """\
+    video_file: path to uploaded video (from Gradio Video input)\
+    num_frames: integer from slider\
+    """\
+    if video_file is None:\
+        raise gr.Error("Please upload a video first.")\
+\
+    # Gradio passes a dict for Video \{ 'name': ..., 'data': ... \} in some versions,\
+    # but in newer versions it passes a filepath string. Handle both.\
+    if isinstance(video_file, dict):\
+        video_path = video_file.get("name") or video_file.get("data")\
+    else:\
+        video_path = video_file\
+\
+    if not video_path or not os.path.exists(video_path):\
+        raise gr.Error("Uploaded video file not found on the server.")\
+\
+    zip_path = extract_frames_to_zip(video_path, int(num_frames))\
+    return zip_path\
+\
+\
+with gr.Blocks() as demo:\
+    gr.Markdown("# Video Frame Extractor\\nUpload a video and get N evenly spaced frames as a ZIP.")\
+\
+    with gr.Row():\
+        video_input = gr.Video(label="Upload video", sources=["upload"])\
+        num_frames_input = gr.Slider(\
+            minimum=2,\
+            maximum=30,\
+            value=10,\
+            step=1,\
+            label="Number of frames to extract",\
+        )\
+\
+    zip_output = gr.File(label="Download frames ZIP")\
+\
+    run_btn = gr.Button("Extract frames")\
+\
+    run_btn.click(\
+        fn=gradio_fn,\
+        inputs=[video_input, num_frames_input],\
+        outputs=[zip_output],\
+    )\
+\
+if __name__ == "__main__":\
+    demo.launch()\
+}

requirements.txt

@@ -0,0 +1,10 @@
+{\rtf1\ansi\ansicpg1252\cocoartf2822
+\cocoatextscaling0\cocoaplatform0{\fonttbl\f0\fswiss\fcharset0 Helvetica;}
+{\colortbl;\red255\green255\blue255;}
+{\*\expandedcolortbl;;}
+\paperw11900\paperh16840\margl1440\margr1440\vieww11520\viewh8400\viewkind0
+\pard\tx720\tx1440\tx2160\tx2880\tx3600\tx4320\tx5040\tx5760\tx6480\tx7200\tx7920\tx8640\pardirnatural\partightenfactor0
+
+\f0\fs24 \cf0 opencv-python\
+gradio\
+}