fix

app.py

@@ -70,34 +70,24 @@ with block:
     input_image.upload(on_input_image_upload, [input_image], [raw_image, point_coords, point_labels], queue=False)
 
     # event - set coords
-    def on_input_image_select(raw_image, input_image, image_edit_trigger, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
-        
-        if image_edit_trigger:
-            unedited_image = input_image.copy()
-            image_edit_trigger = False
-        else:
-            unedited_image = raw_image
-
+    def on_prompt_image_select(input_image, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
         x, y = evt.index
         color = red if point_label_radio == 0 else blue
-        img = np.array(input_image)
+        img = input_image.copy()
+        img = np.array(img)
         cv2.circle(img, (x, y), 5, color, -1)
         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[generated_mask] = (255.0, 0.0, 0.0)
         img = Image.fromarray(img)
-        
         return  [ img,
-                  unedited_image,
-                  img,
                   point_coords,
-                  point_labels,
-                  image_edit_trigger]
+                  point_labels ]
     
-    prompt_image.select(on_input_image_select,
-                       [raw_image, input_image, image_edit_trigger, point_coords, point_labels, point_label_radio],
-                       [prompt_image, raw_image, input_image, point_coords, point_labels, image_edit_trigger], queue=False)
+    prompt_image.select(on_prompt_image_select,
+                       [input_image, point_coords, point_labels, point_label_radio],
+                       [prompt_image, point_coords, point_labels], queue=False)
 
     def on_click_sam_encode_btn(inputs):
         print("encoding")