Update app.py

app.py

@@ -343,11 +343,20 @@ def draw_mask(th, v, d, evt: gr.EventData):
     pts = pts.reshape((-1,1,2))
 
     bg = cv2.cvtColor(d["background"], cv2.COLOR_RGB2GRAY)
-    gd = (cv2.cvtColor(cv2.imread(masks[len(masks)-1]), cv2.COLOR_RGB2GRAY)-th).astype(np.uint8)
-    diff = abs(bg-gd).astype(np.uint8)
+    gr = ((cv2.cvtColor(cv2.imread(masks[len(masks)-1]), cv2.COLOR_RGB2GRAY)-th)*2).astype(np.uint8)
+    diff = abs(bg-gr).astype(np.uint8)
     mask = cv2.inRange(diff, 0, 5)
     d["background"][mask>0] = (0,0,0,255)
 
+    n = 16    # Number of levels of quantization
+    indices = np.arange(0,256)   # List of all colors 
+    divider = np.linspace(0,255,n+1)[1] # we get a divider
+    quantiz = np.int0(np.linspace(0,255,n)) # we get quantization colors
+    color_levels = np.clip(np.int0(indices/divider),0,n-1) # color levels 0,1,2..
+    palette = quantiz[color_levels]
+
+    d["background"] = cv2.convertScaleAbs(palette[d["background"]]) # Converting image back to uint8
+
     d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
     d["composite"] = cv2.fillPoly(d["composite"], [pts], (0,0,0,255))