Update app.py

app.py

@@ -392,7 +392,9 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
 
     mask = cv2.inRange(abs_grad, 1, 255)
     d["layers"][0][mask>0] = (0,0,0,255)
-    d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
+    
+    d["layers"][0] = cv2.floodFill(d["layers"][0], d["layers"][0], pts[len(pts)-1], (0,0,0,255), 0, 0)
+    #d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
     
     return gr.ImageEditor(value=d)