Update app.py

app.py

@@ -21,7 +21,7 @@ def preprocess(img):
     cv2.imwrite("4_dilate.png", img_dilate)
 
     # Erode the dilated edges
-    img_erode = cv2.erode(img_dilate, kernel, iterations=1.5)
+    img_erode = cv2.erode(img_dilate, kernel, iterations=2)
     cv2.imwrite("5_erode.png", img_erode)
 
     return img_erode
@@ -45,12 +45,13 @@ def find_tip(points, convex_hull):
 def infer(image_in):
     img = cv2.imread(image_in)
 
-    contours, hierarchy = cv2.findContours(preprocess(img), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    
+    contours, hierarchy = cv2.findContours(preprocess(img), cv2.RETR_CCOMP, cv2.CHAIN_APPROX_SIMPLE)
+    # Filter out only the internal contours (contours with a parent)
+    internal_contours = [contour for contour, hier in zip(contours, hierarchy[0]) if hier[3] != -1]
 
-    for cnt in contours:
+    for cnt in internal_contours:
         peri = cv2.arcLength(cnt, True)
-        approx = cv2.approxPolyDP(cnt, 0.024 * peri, True)
+        approx = cv2.approxPolyDP(cnt, 0.025 * peri, True)
         hull = cv2.convexHull(approx, returnPoints=False)
         sides = len(hull)