Update app.py

app.py

@@ -190,14 +190,18 @@
 #)
 
 import logging
+import queue
+from pathlib import Path
+from typing import List, NamedTuple
+
+import av
 import cv2
 import numpy as np
-import mediapipe as mp
 import streamlit as st
-from streamlit_webrtc import webrtc_streamer
-import av
-import queue
-from typing import List, NamedTuple
+from streamlit_webrtc import WebRtcMode, webrtc_streamer
+
+from sample_utils.download import download_file
+from sample_utils.turn import get_ice_servers
 
 # Logging setup
 logger = logging.getLogger(__name__)
@@ -216,6 +220,13 @@ class Detection(NamedTuple):
     score: float
     box: np.ndarray
 
+@st.cache_resource  # type: ignore
+def generate_label_colors():
+    return np.random.uniform(0, 255, size=(len(CLASSES), 3))
+
+
+COLORS = generate_label_colors()
+
 # Angle calculation function
 def calculate_angle(a, b, c):
     a = np.array(a)
@@ -229,10 +240,11 @@ def calculate_angle(a, b, c):
 
 
 # Detection Queue
-result_queue: queue.Queue[List[Detection]] = queue.Queue()
+result_queue: "queue.Queue[List[Detection]]" = queue.Queue()
 
 def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
     image = frame.to_ndarray(format="bgr24")
+    h, w = image.shape[:2]
     image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
     with mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5) as pose:
@@ -294,11 +306,14 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
 
 
 # WebRTC streamer configuration
+
 webrtc_streamer(
     key="squat-detection",
-    video_frame_callback=video_frame_callback,
+    mode=WebRtcMode.SENDRECV,
+    rtc_configuration={"iceServers": get_ice_servers(), "iceTransportPolicy": "relay"},
     media_stream_constraints={"video": True, "audio": False},
-    async_processing=True
+    video_frame_callback=video_frame_callback,
+    async_processing=True,
 )