Update app.py

app.py

@@ -372,17 +372,13 @@ def draw_mask(f, v, d, evt: gr.EventData):
     bg = cv2.cvtColor(d["background"], cv2.COLOR_RGB2GRAY)
     gd = cv2.cvtColor(gradient, cv2.COLOR_RGB2GRAY)
 
-    bg = cv2.convertScaleAbs(palette[bg]) # Converting image back to uint8
-    gd = cv2.convertScaleAbs(palette[gd])
-
     diff = (bg-gd).astype(np.uint8)
     mask = cv2.inRange(diff, -int(n/2), int(n/2))
-    d["background"][mask>0] = (0,0,0,255)
+    d["layers"][0][mask>0] = (0,0,0,255)
 
+    bg = cv2.convertScaleAbs(palette[bg]) # Converting image back to uint8
+    gd = cv2.convertScaleAbs(palette[gd])
     d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
-    d["composite"] = cv2.fillPoly(d["composite"], [pts], (0,0,0,255))
-    
-    print(d["composite"])
     
     return gr.ImageEditor(value=d)