Update app.py

app.py

@@ -378,9 +378,6 @@ def draw_mask(l, c, t, v, d, evt: gr.EventData):
     pts = np.array(points, np.int32)
     pts = pts.reshape((-1,1,2))
 
-    x = points[len(points)-1][0]
-    y = points[len(points)-1][1]
-
     scale = 1
     delta = 0
     ddepth = cv2.CV_16S
@@ -419,12 +416,15 @@ def draw_mask(l, c, t, v, d, evt: gr.EventData):
       params["l"] = l
       params["c"] = c
 
-      #d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
+      d["layers"][0] = cv2.cvtColor(bg, cv2.COLOR_GRAY2RGBA)
       edge = bg.copy()
     else:
       bg = edge.copy()
+
+    x = points[len(points)-1][0]
+    y = points[len(points)-1][1]
     
-    mask = cv2.floodFill(bg, None, (x, y), 1, int(t*256/l), int(t*256/l))[2] #, (4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
+    mask = cv2.floodFill(bg, None, (x, y), 1, 0, 0)[2] #, (4 | cv2.FLOODFILL_FIXED_RANGE | cv2.FLOODFILL_MASK_ONLY | 255 << 8)
     # 255 << 8 tells to fill with the value 255)
     mask = mask[1:mask.shape[0]-1, 1:mask.shape[1]-1]
     
@@ -536,6 +536,7 @@ with gr.Blocks(css=css) as demo:
                   for (var i=0; i<256; i++) {
                     pts += i+\",0 \";
                   }
+                  alert(points);
                   document.getElementById(\"pl\").setAttribute(\"points\", pts.slice(0,-1)); 
                   var xold = 0;
                   var yold = 0;