Update app.py

app.py

@@ -38,29 +38,33 @@ def find_tip(points, convex_hull):
         if np.all(points[j] == points[indices[i - 1] - 2]):
             return tuple(points[j])
 
-
-
-
-
+def find_tail(points, convex_hull):
+    # Implement the logic to find the tail point
+    # You might need to adjust this based on the specific characteristics of your arrows
+    # This is just a simple example, and you may need to fine-tune it based on your requirements
+    tail_index = (convex_hull[0] + convex_hull[-1]) // 2
+    return tuple(points[tail_index])
 
 def infer(image_in):
     img = cv2.imread(image_in)
 
     contours, hierarchy = cv2.findContours(preprocess(img), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    
+
     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 6 > sides > 3 and sides + 2 == len(approx):
             arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
-            if arrow_tip:
+            arrow_tail = find_tail(approx[:,0,:], hull.squeeze())
+            
+            if arrow_tip and arrow_tail:
                 cv2.drawContours(img, [cnt], -1, (0, 255, 0), 3)
                 cv2.circle(img, arrow_tip, 3, (0, 0, 255), cv2.FILLED)
+                cv2.circle(img, arrow_tail, 3, (255, 0, 0), cv2.FILLED)
+
 
     cv2.imwrite("Image_result.png", img)