Update app.py

app.py

@@ -273,24 +273,24 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
                     landmarks[mp_pose.PoseLandmark.LEFT_FOOT_INDEX.value].y]
 
             # Calculate angles
-            angleKneeL = calculate_angle(hip, knee, ankle)
-            angleHipL = calculate_angle(shoulder, hip, [hip[0], 0])
-            angleAnkleL = calculate_angle(foot, ankle, knee)
+            angleKneeL = calculate_angle(hipL, kneeL, ankleL)
+            angleHipL = calculate_angle(shoulderL, hipL, [hipL[0], 0])
+            angleAnkleL = calculate_angle(footIndexL, ankleL, kneeL)
             
             #Visualize of left leg
             cv2.putText(image, str(angleHipL),tuple(np.multiply(angleHipL, [640, 480]).astype(int)),cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2, cv2.LINE_AA)
             
             # Squat logic
-            if 80 < knee_angle < 110 and 29 < hip_angle < 40:
+            if 80 < angleKneeL < 110 and 29 < angleHipL < 40:
                 cv2.putText(image, "Squat Detected!", (300, 100), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 255, 0), 3)
             else:
-                if hip_angle < 29:
+                if angleHipL < 29:
                     cv2.putText(image, "Lean Forward!", (300, 200), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
-                elif hip_angle > 45:
+                elif angleHipL > 45:
                     cv2.putText(image, "Lean Backward!", (300, 200), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
-                if knee_angle < 80:
+                if angleKneeL < 80:
                     cv2.putText(image, "Squat Too Deep!", (300, 250), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
-                elif knee_angle > 110:
+                elif angleKneeL > 110:
                     cv2.putText(image, "Lower Your Hips!", (300, 300), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
             
     #                                 # 1. Bend Forward Warning
@@ -373,9 +373,7 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
 #                (230,60),cv2.FONT_HERSHEY_SIMPLEX,1,(255,255,255),1,cv2.LINE_AA)
             
 
-        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))
+        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(detections)
     return av.VideoFrame.from_ndarray(image, format="bgr24")