File size: 2,231 Bytes
df80275 79ac659 df80275 5f71f30 df80275 5f71f30 df80275 5f71f30 df80275 5f71f30 df80275 5f71f30 df80275 5f71f30 df80275 79ac659 df80275 5f71f30 df80275 5f71f30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import uuid
from pathlib import Path
import av
import cv2
import streamlit as st
from aiortc.contrib.media import MediaRecorder
from streamlit_webrtc import WebRtcMode, webrtc_streamer
from sample_utils.turn import get_ice_servers
def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
"""
Process video frames: Perform edge detection on the incoming frames.
"""
img = frame.to_ndarray(format="bgr24")
img = cv2.cvtColor(cv2.Canny(img, 100, 200), cv2.COLOR_GRAY2BGR)
return av.VideoFrame.from_ndarray(img, format="bgr24")
RECORD_DIR = Path("./records")
RECORD_DIR.mkdir(exist_ok=True)
def virtual_keyboard_app():
"""
Streamlit application for recording video streams with a filter applied.
"""
if "prefix" not in st.session_state:
st.session_state["prefix"] = str(uuid.uuid4())
prefix = st.session_state["prefix"]
# Define input and output file paths
in_file = RECORD_DIR / f"{prefix}_input.flv"
out_file = RECORD_DIR / f"{prefix}_output.flv"
# Factory for creating MediaRecorder objects
def in_recorder_factory() -> MediaRecorder:
return MediaRecorder(
str(in_file), format="flv"
)
def out_recorder_factory() -> MediaRecorder:
return MediaRecorder(str(out_file), format="flv")
# Initialize WebRTC streamer
webrtc_streamer(
key="record",
mode=WebRtcMode.SENDRECV,
rtc_configuration={"iceServers": get_ice_servers()},
media_stream_constraints={
"video": True,
"audio": True,
},
video_frame_callback=video_frame_callback,
in_recorder_factory=in_recorder_factory,
out_recorder_factory=out_recorder_factory,
)
# Display download buttons for recorded files
if in_file.exists():
with in_file.open("rb") as f:
st.download_button(
"Download the recorded video without video filter", f, "input.flv"
)
if out_file.exists():
with out_file.open("rb") as f:
st.download_button(
"Download the recorded video with video filter", f, "output.flv"
)
if __name__ == "__main__":
virtual_keyboard_app()
|