Update pages/7_record.py

pages/7_record.py

@@ -11,11 +11,11 @@ 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")
-
-    # perform edge detection
     img = cv2.cvtColor(cv2.Canny(img, 100, 200), cv2.COLOR_GRAY2BGR)
-
     return av.VideoFrame.from_ndarray(img, format="bgr24")
 
 
@@ -23,21 +23,28 @@ RECORD_DIR = Path("./records")
 RECORD_DIR.mkdir(exist_ok=True)
 
 
-def app():
+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"
-        )  # HLS does not work. See https://github.com/aiortc/aiortc/issues/331
+        )
 
     def out_recorder_factory() -> MediaRecorder:
         return MediaRecorder(str(out_file), format="flv")
 
+    # Initialize WebRTC streamer
     webrtc_streamer(
         key="record",
         mode=WebRtcMode.SENDRECV,
@@ -51,6 +58,7 @@ def app():
         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(
@@ -64,4 +72,4 @@ def app():
 
 
 if __name__ == "__main__":
-    app()
+    virtual_keyboard_app()