Update app.py

app.py

@@ -97,6 +97,16 @@ def classify_pose(landmarks, output_image, display=False):
        abs(landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].x - landmarks[mp_pose.PoseLandmark.LEFT_ANKLE.value].x) < 0.05 and \
        abs(landmarks[mp_pose.PoseLandmark.RIGHT_WRIST.value].x - landmarks[mp_pose.PoseLandmark.RIGHT_ANKLE.value].x) < 0.05:
         label = "Hands Under Feet Pose"
+
+    # Check for Plank Pose
+    #----------------------------------------------------------------------------------------------------------------
+    # The body should be in a straight line from head to heels, 
+    # so the shoulder and hip angles should be close to 180 degrees
+    if left_shoulder_angle > 160 and left_shoulder_angle < 200 and \
+       right_shoulder_angle > 160 and right_shoulder_angle < 200 and \
+       left_knee_angle > 160 and left_knee_angle < 200 and \
+       right_knee_angle > 160 and right_knee_angle < 200:
+        label = "Plank Pose"
     
     return output_image, label