Update app.py

app.py

@@ -42,7 +42,7 @@ def get_length(p1, p2):
     line_length = ((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2) ** 0.5
     return line_length
 
-def get_clockwise_angle(tip, tail):
+def get_angle(tip, tail):
     return (np.degrees(np.arctan2(tip[1] - tail[1], tip[0] - tail[0]))*-1)
     
 def get_max_distance_point(cnt, tip):
@@ -74,14 +74,14 @@ def find_tail(points, tip):
 def draw_arrow(img_result, tip, tail, length, angle):
     # Draw arrow on the blank image with inverted tip and tail
     cv2.arrowedLine(img_result, tuple(tail), tuple(tip), (0, 255, 0), 3)
-
+    """
     # Add length and angle as text next to the tip point
     text_length = f"Length: {length:.2f}"
     text_angle = f"Angle: {angle:.2f}"
     
     cv2.putText(img_result, text_length, (tip[0] + 10, tip[1] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1, cv2.LINE_AA)
     cv2.putText(img_result, text_angle, (tip[0] + 10, tip[1] - 30), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1, cv2.LINE_AA)
-    
+    """
 def infer(image_in):
     img = cv2.imread(image_in)
 
@@ -90,13 +90,15 @@ def infer(image_in):
     # Create a blank image to accumulate arrows
     img_result = np.zeros_like(img)
 
+    arrows_coordinates = []
+    
     for cnt in contours:
         peri = cv2.arcLength(cnt, True)
         approx = cv2.approxPolyDP(cnt, 0.025 * peri, True)
         hull = cv2.convexHull(approx, returnPoints=False)
         sides = len(hull)
 
-        if 8 >= sides > 3 and sides + 2 == len(approx):
+        if 6 > sides > 3 and sides + 2 == len(approx):
             arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
             
             
@@ -105,10 +107,11 @@ def infer(image_in):
                 cv2.circle(img, arrow_tip, 3, (0, 0, 255), cv2.FILLED)
                 arrow_tail = find_tail(approx[:,0,:], arrow_tip)
                 if arrow_tail :
+                    arrows_coordinates.append([arrow_tail, arrow_tip])
                     cv2.circle(img, arrow_tail, 3, (255, 0, 0), cv2.FILLED)
                     # Calculate length and angle
                     arrow_length = get_length(arrow_tip, arrow_tail)
-                    arrow_angle = get_clockwise_angle(arrow_tip, arrow_tail)
+                    arrow_angle = get_angle(arrow_tip, arrow_tail)
                     # Draw arrow on the same blank image
                     draw_arrow(img_result, arrow_tip, arrow_tail, arrow_length, arrow_angle)
 
@@ -117,6 +120,8 @@ def infer(image_in):
     cv2.imwrite("Image_result.png", img)
     cv2.imwrite("Arrows_on_same_blank.png", img_result)
 
+    print(f"arrows coordinates: {arrows_coordinates}")
+
     return "Image_result.png", "Arrows_on_same_blank.png"
 
 gr.Interface(