Update app.py

app.py

@@ -10,12 +10,59 @@ mp_drawing = mp.solutions.drawing_utils
 
 # Define a function to classify yoga poses
 def classify_pose(landmarks):
-    # Example logic for classifying poses
-    if landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].visibility > 0.5 and \
-       landmarks[mp_pose.PoseLandmark.RIGHT_SHOULDER.value].visibility > 0.5:
-        return "T-Pose"
+    # Check if the both arms are straight.
+    if left_elbow_angle > 165 and left_elbow_angle < 195 and right_elbow_angle > 165 and right_elbow_angle < 195:
+
+        # Check if shoulders are at the required angle.
+        if left_shoulder_angle > 80 and left_shoulder_angle < 110 and right_shoulder_angle > 80 and right_shoulder_angle < 110:
+
+    # Check if it is the warrior II pose.
+    #----------------------------------------------------------------------------------------------------------------
+
+            # Check if one leg is straight.
+            if left_knee_angle > 165 and left_knee_angle < 195 or right_knee_angle > 165 and right_knee_angle < 195:
+
+                # Check if the other leg is bended at the required angle.
+                if left_knee_angle > 90 and left_knee_angle < 120 or right_knee_angle > 90 and right_knee_angle < 120:
+
+                    # Specify the label of the pose that is Warrior II pose.
+                    label = 'Warrior II Pose'
+
+    #----------------------------------------------------------------------------------------------------------------
+
+    # Check if it is the T pose.
+    #----------------------------------------------------------------------------------------------------------------
+
+            # Check if both legs are straight
+            if left_knee_angle > 160 and left_knee_angle < 195 and right_knee_angle > 160 and right_knee_angle < 195:
+
+                # Specify the label of the pose that is tree pose.
+                label = 'T Pose'
+    
+    #----------------------------------------------------------------------------------------------------------------
+
+    # Check if it is the tree pose.
+    #----------------------------------------------------------------------------------------------------------------
+
+    # Check if one leg is straight
+    if left_knee_angle > 165 and left_knee_angle < 195 or right_knee_angle > 165 and right_knee_angle < 195:
+
+        # Check if the other leg is bended at the required angle.
+        if left_knee_angle > 315 and left_knee_angle < 335 or right_knee_angle > 25 and right_knee_angle < 45:
+
+            # Specify the label of the pose that is tree pose.
+            label = 'Tree Pose'
+    
+    # Check for Upward Salute Pose
+    if abs(landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value][0] - landmarks[mp_pose.PoseLandmark.LEFT_HIP.value][0]) < 100 and \
+       abs(landmarks[mp_pose.PoseLandmark.RIGHT_WRIST.value][0] - landmarks[mp_pose.PoseLandmark.RIGHT_HIP.value][0]) < 100 and \
+       landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value][1] < landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value][1] and \
+       landmarks[mp_pose.PoseLandmark.RIGHT_WRIST.value][1] < landmarks[mp_pose.PoseLandmark.RIGHT_SHOULDER.value][1] and \
+       abs(landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value][1] - landmarks[mp_pose.PoseLandmark.RIGHT_SHOULDER.value][1]) < 50:
+        label = "Upward Salute Pose"
+
     # Add more classification rules here
-    return "Unknown Pose"
+            return "Unknown Pose"
 
 def detect_and_classify_pose(input_image):
     frame = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)