Update app.py

app.py

@@ -1,37 +1,20 @@
 import cv2
 import mediapipe as mp
+import os
 
-# Initialize MediaPipe Pose
+# Initialize MediaPipe utilities
 mp_drawing = mp.solutions.drawing_utils
 mp_pose = mp.solutions.pose
 
-# Set up webcam
-cap = cv2.VideoCapture(0)
-
-with mp_pose.Pose(
-    static_image_mode=False,
-    model_complexity=1,
-    smooth_landmarks=True,
-    enable_segmentation=True,
-    min_detection_confidence=0.5,
-    min_tracking_confidence=0.5
-) as pose:
-    while cap.isOpened():
-        ret, frame = cap.read()
-        if not ret:
-            print("❌ Failed to capture frame")
-            break
-
-        # Convert to RGB for MediaPipe
-        image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        image.flags.writeable = False
-        results = pose.process(image)
-
-        # Convert back to BGR for OpenCV display
-        image.flags.writeable = True
-        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
-
-        # Draw pose landmarks
+def process_image(image_path):
+    """Run pose estimation on a static image."""
+    image = cv2.imread(image_path)
+    if image is None:
+        print(f"❌ Could not read image: {image_path}")
+        return
+
+    with mp_pose.Pose(static_image_mode=True) as pose:
+        results = pose.process(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
         if results.pose_landmarks:
             mp_drawing.draw_landmarks(
                 image,
@@ -41,11 +24,67 @@ with mp_pose.Pose(
                 connection_drawing_spec=mp_drawing.DrawingSpec(color=(255, 255, 255), thickness=2, circle_radius=2)
             )
 
-        # Show
-        cv2.imshow("🧍 Pose Estimation (ESC to exit)", image)
+        cv2.imshow("🧍 Pose Estimation (Image)", image)
+        cv2.waitKey(0)
+        cv2.destroyAllWindows()
+
+def process_webcam():
+    """Run pose estimation using webcam if available."""
+    cap = cv2.VideoCapture(0)
+    if not cap.isOpened():
+        print("⚠️ Webcam not detected. Switching to image mode...")
+        return False
+
+    with mp_pose.Pose(
+        static_image_mode=False,
+        model_complexity=1,
+        smooth_landmarks=True,
+        enable_segmentation=True,
+        min_detection_confidence=0.5,
+        min_tracking_confidence=0.5
+    ) as pose:
+        print("🎥 Webcam started. Press ESC to exit.")
+        while True:
+            ret, frame = cap.read()
+            if not ret:
+                print("❌ Failed to capture frame.")
+                break
+
+            # Convert to RGB for MediaPipe
+            image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            image.flags.writeable = False
+            results = pose.process(image)
+
+            # Convert back to BGR for display
+            image.flags.writeable = True
+            image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+
+            # Draw pose landmarks
+            if results.pose_landmarks:
+                mp_drawing.draw_landmarks(
+                    image,
+                    results.pose_landmarks,
+                    mp_pose.POSE_CONNECTIONS,
+                    landmark_drawing_spec=mp_drawing.DrawingSpec(color=(0, 255, 0), thickness=2, circle_radius=2),
+                    connection_drawing_spec=mp_drawing.DrawingSpec(color=(255, 255, 255), thickness=2, circle_radius=2)
+                )
+
+            cv2.imshow("🧍 Pose Estimation (ESC to exit)", image)
+            if cv2.waitKey(5) & 0xFF == 27:  # ESC key
+                break
+
+    cap.release()
+    cv2.destroyAllWindows()
+    return True
 
-        if cv2.waitKey(5) & 0xFF == 27:
-            break
+if __name__ == "__main__":
+    print("🚀 Starting MediaPipe Pose App...")
 
-cap.release()
-cv2.destroyAllWindows()
+    # Try webcam first
+    if not process_webcam():
+        # If no webcam, use fallback image
+        test_image = "pose_test.jpg"
+        if not os.path.exists(test_image):
+            print(f"⚠️ No '{test_image}' found. Please place a test image in this folder.")
+        else:
+            process_image(test_image)