Update main_code_script.py

main_code_script.py

@@ -13,39 +13,40 @@ import torch
 # --- 1. Pose Estimation (using Mediapipe) ---
 def estimate_pose(image_path):
     """Detects the pose of a person in an image using Mediapipe.
-
     Args:
         image_path: Path to the input image.
-
     Returns:
-        A list of landmarks (x, y, visibility) or None if no pose is detected.
+        A list of landmarks (x, y, visibility) 
+        or None if no pose is detected.
     """
     mp_drawing = mp.solutions.drawing_utils
     mp_pose = mp.solutions.pose
-
-    with mp_pose.Pose(
-        static_image_mode=True,
-        model_complexity=2,
-        enable_segmentation=True,
-        min_detection_confidence=0.5) as pose:
-
-        image = cv2.imread(image_path)
-        image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-        results = pose.process(image_rgb)
-
-        if results.pose_landmarks:
-            # Example: Draw the pose landmarks on the image (for visualization)
-            annotated_image = image.copy()
-            mp_drawing.draw_landmarks(
-                annotated_image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
-
-            #cv2.imwrite("pose_annotated.jpg", annotated_image)  # Save annotated image
-            #return results.pose_landmarks.landmark  # Return the landmarks
-            return results, image # Return the entire result
-
-        else:
-            return None, None # or raise an exception
-
+    try:
+        with mp_pose.Pose(
+            static_image_mode=True,
+            model_complexity=2,
+            enable_segmentation=True,
+            min_detection_confidence=0.5) as pose:
+            image = cv2.imread(image_path)
+            if image is None:
+                raise FileNotFoundError(f"Could not open image: {image_path}")
+            image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+            results = pose.process(image_rgb)
+
+            if results.pose_landmarks:
+                # Example: Draw the pose landmarks on the image (for visualization)
+                annotated_image = image.copy()
+                mp_drawing.draw_landmarks(
+                    annotated_image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
+                #cv2.imwrite("pose_annotated.jpg", annotated_image)  # Save annotated image
+                #return results.pose_landmarks.landmark  
+                # Return the landmarks
+                return results, image # Return the entire result
+            else:
+                raise ValueError("No pose detected in the image.")
+    except Exception as e:
+        raise RuntimeError(f"Error processing image: {e}")
+        
 # --- 2. Clothing Segmentation (Example - using a placeholder function) ---
 def segment_clothing(image, results): #Added result
     """Segments the clothing region in the image.