Update app.py

app.py

@@ -20,12 +20,14 @@ def midpoint(x1, y1, x2, y2):
     y_mid = int((y1 + y2)/2)
     return (x_mid, y_mid)
 
-def draw_boxes(image, bounds, color='yellow', width=2):
-    draw = ImageDraw.Draw(image)
+def draw_boxes(img, bounds, color='yellow', width=2):
+    
+    mask = np.zeros(img.shape[:2], dtype="uint8")
+    draw = ImageDraw.Draw(mask)
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
-        draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
-    return (image)
+        draw.polygon((*p0, *p1, *p2, *p3), fill=255)
+    return (mask)
     
 def draw_mask(img, bounds):
     mask = np.zeros(img.shape[:2], dtype="uint8")