colour according to mask

app.py

@@ -82,10 +82,12 @@ with block:
         color = red if point_label_radio == 0 else blue
         img = np.array(input_image)
         cv2.circle(img, (x, y), 5, color, -1)
-        img = Image.fromarray(img)
         point_coords.append([x,y])
         point_labels.append(point_label_radio)
         generated_mask, _, _ = sam.cond_pred(pts=np.array(point_coords), lbls=np.array(point_labels))
+        img[generated_mask] = (1.0, 0.0, 0.0)
+        img = Image.fromarray(img)
+        
         return  [ img,
                   unedited_image,
                   img,