Update app.py

app.py

@@ -6,7 +6,30 @@ import numpy as np
 import matplotlib.pyplot as plt
 from scipy.cluster.vq import kmeans
 
-# ... (previous functions remain the same)
+def get_coordinates_from_mask(mask_in):
+    x_y = np.where(mask_in != [0,0,0,255])[:2]
+    x_y = np.column_stack((x_y[1], x_y[0]))
+    x_y = np.float32(x_y)
+    centroids,_ = kmeans(x_y,4)
+    centroids = np.int64(centroids)
+    return centroids
+
+def get_top_bottom_coordinates(coords):
+    top_coord = min(coords, key=lambda x : x[1])
+    bottom_coord = max(coords, key=lambda x : x[1])
+    return top_coord, bottom_coord
+
+def sort_centroids_clockwise(centroids: np.ndarray):    
+    c_list = centroids.tolist()
+    c_list.sort(key = lambda y : y[0])
+    
+    left_coords = c_list[:2]
+    right_coords = c_list[-2:]
+    
+    top_left, bottom_left = get_top_bottom_coordinates(left_coords)
+    top_right, bottom_right = get_top_bottom_coordinates(right_coords)
+    
+    return top_left, top_right, bottom_right, bottom_left
 
 def infer(image_input, dst_height: str, dst_width: str):
     image_in = image_input["image"]