Update app.py

app.py

@@ -99,7 +99,6 @@ def classify_pose(landmarks, output_image, display=False):
         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 \
@@ -108,9 +107,17 @@ def classify_pose(landmarks, output_image, display=False):
        right_knee_angle > 160 and right_knee_angle < 200:
         label = "Plank Pose"
     
+    # Update the color to green if pose is classified
+    if label != 'Unknown Pose':
+        color = (0, 255, 0)
+    
+    # Write the label on the output image
+    cv2.putText(output_image, label, (10, 30), cv2.FONT_HERSHEY_PLAIN, 2, color, 2)
+
     return output_image, label
 
 def detect_and_classify_pose(input_image):
+    input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)
     results = pose.process(input_image)
     pose_classification = "No pose detected"
     if results.pose_landmarks: