Upload folder using huggingface_hub

README.md

@@ -1,12 +1,12 @@
+
 ---
-title: Stream Video Out Main
-emoji: 🚀
-colorFrom: blue
-colorTo: green
+title: stream_video_out_main 
+emoji: 🔥
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
-sdk_version: 4.44.1
-app_file: app.py
+sdk_version: 5.0.0
+app_file: run.py
 pinned: false
+hf_oauth: true
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

requirements.txt

@@ -0,0 +1,3 @@
+gradio-client @ git+https://github.com/gradio-app/gradio@bbf9ba7e997022960c621f72baa891185bd03732#subdirectory=client/python
+https://gradio-pypi-previews.s3.amazonaws.com/bbf9ba7e997022960c621f72baa891185bd03732/gradio-5.0.0-py3-none-any.whl
+opencv-python

run.ipynb

@@ -0,0 +1 @@
+{"cells": [{"cell_type": "markdown", "id": "302934307671667531413257853548643485645", "metadata": {}, "source": ["# Gradio Demo: stream_video_out"]}, {"cell_type": "code", "execution_count": null, "id": "272996653310673477252411125948039410165", "metadata": {}, "outputs": [], "source": ["!pip install -q gradio opencv-python"]}, {"cell_type": "code", "execution_count": null, "id": "288918539441861185822528903084949547379", "metadata": {}, "outputs": [], "source": ["# Downloading files from the demo repo\n", "import os\n", "os.mkdir('video')\n", "!wget -q -O video/compliment_bot_screen_recording_3x.mp4 https://github.com/gradio-app/gradio/raw/main/demo/stream_video_out/video/compliment_bot_screen_recording_3x.mp4"]}, {"cell_type": "code", "execution_count": null, "id": "44380577570523278879349135829904343037", "metadata": {}, "outputs": [], "source": ["import gradio as gr\n", "import cv2\n", "import os\n", "from pathlib import Path\n", "import atexit\n", "\n", "current_dir = Path(__file__).resolve().parent\n", "\n", "\n", "def delete_files():\n", "    for p in Path(current_dir).glob(\"*.ts\"):\n", "        p.unlink()\n", "    for p in Path(current_dir).glob(\"*.mp4\"):\n", "        p.unlink()\n", "\n", "\n", "atexit.register(delete_files)\n", "\n", "\n", "def process_video(input_video, stream_as_mp4):\n", "    cap = cv2.VideoCapture(input_video)\n", "\n", "    video_codec = cv2.VideoWriter_fourcc(*\"mp4v\") if stream_as_mp4 else cv2.VideoWriter_fourcc(*\"x264\")  # type: ignore\n", "    fps = int(cap.get(cv2.CAP_PROP_FPS))\n", "    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))\n", "    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))\n", "\n", "    iterating, frame = cap.read()\n", "\n", "    n_frames = 0\n", "    n_chunks = 0\n", "    name = str(current_dir / f\"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}\")\n", "    segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height))  # type: ignore\n", "\n", "    while iterating:\n", "\n", "        # flip frame vertically\n", "        frame = cv2.flip(frame, 0)\n", "        display_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)\n", "        segment_file.write(display_frame)\n", "        n_frames += 1\n", "        if n_frames == 3 * fps:\n", "            n_chunks += 1\n", "            segment_file.release()\n", "            n_frames = 0\n", "            yield name\n", "            name = str(\n", "                current_dir / f\"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}\"\n", "            )\n", "            segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height))  # type: ignore\n", "\n", "        iterating, frame = cap.read()\n", "\n", "    segment_file.release()\n", "    yield name\n", "\n", "\n", "with gr.Blocks() as demo:\n", "    gr.Markdown(\"# Video Streaming Out \ud83d\udcf9\")\n", "    with gr.Row():\n", "        with gr.Column():\n", "            input_video = gr.Video(label=\"input\")\n", "            checkbox = gr.Checkbox(label=\"Stream as MP4 file?\", value=False)\n", "        with gr.Column():\n", "            processed_frames = gr.Video(\n", "                label=\"stream\",\n", "                streaming=True,\n", "                autoplay=True,\n", "                elem_id=\"stream_video_output\",\n", "            )\n", "    with gr.Row():\n", "        process_video_btn = gr.Button(\"process video\")\n", "\n", "    process_video_btn.click(process_video, [input_video, checkbox], [processed_frames])\n", "\n", "    gr.Examples(\n", "        [\n", "            [\n", "                os.path.join(\n", "                    os.path.abspath(''),\n", "                    \"video/compliment_bot_screen_recording_3x.mp4\",\n", "                ),\n", "                False,\n", "            ],\n", "            [\n", "                os.path.join(\n", "                    os.path.abspath(''),\n", "                    \"video/compliment_bot_screen_recording_3x.mp4\",\n", "                ),\n", "                True,\n", "            ],\n", "        ],\n", "        [input_video, checkbox],\n", "        fn=process_video,\n", "        outputs=processed_frames,\n", "        cache_examples=False,\n", "    )\n", "\n", "\n", "if __name__ == \"__main__\":\n", "    demo.launch()\n"]}], "metadata": {}, "nbformat": 4, "nbformat_minor": 5}

run.py

@@ -0,0 +1,101 @@
+import gradio as gr
+import cv2
+import os
+from pathlib import Path
+import atexit
+
+current_dir = Path(__file__).resolve().parent
+
+
+def delete_files():
+    for p in Path(current_dir).glob("*.ts"):
+        p.unlink()
+    for p in Path(current_dir).glob("*.mp4"):
+        p.unlink()
+
+
+atexit.register(delete_files)
+
+
+def process_video(input_video, stream_as_mp4):
+    cap = cv2.VideoCapture(input_video)
+
+    video_codec = cv2.VideoWriter_fourcc(*"mp4v") if stream_as_mp4 else cv2.VideoWriter_fourcc(*"x264")  # type: ignore
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+    iterating, frame = cap.read()
+
+    n_frames = 0
+    n_chunks = 0
+    name = str(current_dir / f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}")
+    segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height))  # type: ignore
+
+    while iterating:
+
+        # flip frame vertically
+        frame = cv2.flip(frame, 0)
+        display_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        segment_file.write(display_frame)
+        n_frames += 1
+        if n_frames == 3 * fps:
+            n_chunks += 1
+            segment_file.release()
+            n_frames = 0
+            yield name
+            name = str(
+                current_dir / f"output_{n_chunks}{'.mp4' if stream_as_mp4 else '.ts'}"
+            )
+            segment_file = cv2.VideoWriter(name, video_codec, fps, (width, height))  # type: ignore
+
+        iterating, frame = cap.read()
+
+    segment_file.release()
+    yield name
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Video Streaming Out 📹")
+    with gr.Row():
+        with gr.Column():
+            input_video = gr.Video(label="input")
+            checkbox = gr.Checkbox(label="Stream as MP4 file?", value=False)
+        with gr.Column():
+            processed_frames = gr.Video(
+                label="stream",
+                streaming=True,
+                autoplay=True,
+                elem_id="stream_video_output",
+            )
+    with gr.Row():
+        process_video_btn = gr.Button("process video")
+
+    process_video_btn.click(process_video, [input_video, checkbox], [processed_frames])
+
+    gr.Examples(
+        [
+            [
+                os.path.join(
+                    os.path.dirname(__file__),
+                    "video/compliment_bot_screen_recording_3x.mp4",
+                ),
+                False,
+            ],
+            [
+                os.path.join(
+                    os.path.dirname(__file__),
+                    "video/compliment_bot_screen_recording_3x.mp4",
+                ),
+                True,
+            ],
+        ],
+        [input_video, checkbox],
+        fn=process_video,
+        outputs=processed_frames,
+        cache_examples=False,
+    )
+
+
+if __name__ == "__main__":
+    demo.launch()