Update app.py

app.py

@@ -42,6 +42,9 @@ def get_length(p1, p2):
     line_length = ((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2) ** 0.5
     return line_length
 
+def get_angle(tip, tail):
+    return np.degrees(np.arctan2(tail[1] - tip[1], tail[0] - tip[0]))
+    
 def get_max_distance_point(cnt, tip):
     max_distance = 0
     max_point = None
@@ -68,9 +71,13 @@ def find_tail(points, tip):
 
     return None
 
-def draw_arrow(img_result, tip, tail):
-    # Draw arrow on the blank image
-    cv2.arrowedLine(img_result, tuple(tail), tuple(tip), (0, 255, 0), 2)
+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 = f"Length: {length:.2f}, Angle: {angle:.2f}"
+    cv2.putText(img_result, text, (tip[0] + 10, tip[1] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1, cv2.LINE_AA)
     
 def infer(image_in):
     img = cv2.imread(image_in)
@@ -86,7 +93,7 @@ def infer(image_in):
         hull = cv2.convexHull(approx, returnPoints=False)
         sides = len(hull)
 
-        if 24 >= sides > 3 and sides + 2 == len(approx):
+        if 8 >= sides > 3 and sides + 2 == len(approx):
             arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
             
             
@@ -96,9 +103,11 @@ def infer(image_in):
                 arrow_tail = find_tail(approx[:,0,:], arrow_tip)
                 if arrow_tail :
                     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_angle(arrow_tip, arrow_tail)
                     # Draw arrow on the same blank image
-                    draw_arrow(img_result, arrow_tip, arrow_tail)
+                    draw_arrow(img_result, arrow_tip, arrow_tail, arrow_length, arrow_angle)