Update app.py

app.py

@@ -99,22 +99,22 @@ def infer(image_in):
         hull = cv2.convexHull(approx, returnPoints=False)
         sides = len(hull)
 
-        #if 10 > sides > 3 and sides + 2 == len(approx):
-        arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
-        
-        
-        if arrow_tip :
-            cv2.drawContours(img, [cnt], -1, (0, 255, 0), 3)
-            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([list(arrow_tail), list(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_angle(arrow_tip, arrow_tail)
-                # Draw arrow on the same blank image
-                draw_arrow(img_result, arrow_tip, arrow_tail, arrow_length, arrow_angle)
+        if 18 > sides > 3 and sides + 2 == len(approx):
+            arrow_tip = find_tip(approx[:,0,:], hull.squeeze())
+            
+            
+            if arrow_tip :
+                cv2.drawContours(img, [cnt], -1, (0, 255, 0), 3)
+                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([list(arrow_tail), list(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_angle(arrow_tip, arrow_tail)
+                    # Draw arrow on the same blank image
+                    draw_arrow(img_result, arrow_tip, arrow_tail, arrow_length, arrow_angle)