Update app.py

app.py

@@ -221,15 +221,26 @@ def calculate_angle(a, b, c):
     return angle
 
 # Detection Queue
-result_queue: "queue.Queue[list]" = 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")
     image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
     with mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5) as pose:
         results = pose.process(image_rgb)
         landmarks = results.pose_landmarks.landmark if results.pose_landmarks else []
 
+        # Corrected detection logic
+        detections = [
+            Detection(
+                class_id=0,  # Assuming a generic class_id for pose detections
+                label="Pose",
+                score=1.0,  # Full confidence as pose landmarks were detected
+                box=np.array([0, 0, image.shape[1], image.shape[0]])  # Full image as bounding box
+            )
+        ] if landmarks else []
+
         if landmarks:
             hip = [landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].x, 
                    landmarks[mp_pose.PoseLandmark.LEFT_HIP.value].y]
@@ -264,14 +275,15 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
                     cv2.putText(image, "Squat Too Deep!", (300, 250), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
                 elif knee_angle > 110:
                     cv2.putText(image, "Lower Your Hips!", (300, 300), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
-        
+
         mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS,
                                   mp_drawing.DrawingSpec(color=(255, 175, 0), thickness=2, circle_radius=2),
                                   mp_drawing.DrawingSpec(color=(0, 255, 200), thickness=2, circle_radius=2))
 
-    result_queue.put(landmarks)
+    result_queue.put(detections)
     return av.VideoFrame.from_ndarray(image, format="bgr24")
 
+
 # WebRTC streamer configuration
 webrtc_streamer(
     key="squat-detection",