Update app.py

app.py

@@ -202,6 +202,13 @@ def calculate_and_display_angles(image1, return_angles=False):
     image = image1#cv2.imread(image_path)
     image_rgb = image1 #cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
+    # Decode image from memory
+    data = np.frombuffer(image1.getvalue(), dtype=np.uint8)
+    image = cv2.imdecode(data, cv2.IMREAD_COLOR)
+
+    # Get image dimensions
+    height, width, channels = image.shape
+
     # Get pose landmarks
     results = pose.process(image_rgb)
 
@@ -221,8 +228,8 @@ def calculate_and_display_angles(image1, return_angles=False):
         # Display the angle values at corresponding locations
         def overlay_angle(angle, location, color=(0, 255, 0)):
             # Coordinates are normalized, scale them to the image size
-            x = int(location[0] * image.shape[1])
-            y = int(location[1] * image.shape[0])
+            x = int(location[0] * width)#image.shape[1])
+            y = int(location[1] * height)#image.shape[0])
             cv2.putText(image, f'{angle:.2f}', (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.3, color, 1)
 
         # Overlay angles on the image