fix

app.py

@@ -70,10 +70,14 @@ 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(input_image, image_edit_trigger, point_coords, point_labels, point_label_radio, evt: gr.SelectData):
+    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
+
         x, y = evt.index
         color = red if point_label_radio == 0 else blue
         img = np.array(input_image)
@@ -81,14 +85,16 @@ with block:
         img = Image.fromarray(img)
         point_coords.append([x,y])
         point_labels.append(point_label_radio)
-        
-        return {raw_image: unedited_image,
-                input_image: img,
-                point_coords: point_coords,
-                point_labels: point_labels,
-                image_edit_trigger: image_edit_trigger}
+
+        return  [unedited_image,
+                 img,
+                 point_coords,
+                 point_labels,
+                 image_edit_trigger]
     
-    input_image.select(on_input_image_select, [input_image, image_edit_trigger, point_coords, point_labels, point_label_radio], [input_image, raw_image, point_coords, point_labels, image_edit_trigger], queue=False)
+    input_image.select(on_input_image_select,
+                       [raw_image, input_image, image_edit_trigger, point_coords, point_labels, point_label_radio],
+                       [raw_image, input_image, point_coords, point_labels, image_edit_trigger], queue=False)
 
     def on_click_sam_encode_btn(inputs):
         print("encoding")
@@ -109,7 +115,7 @@ with block:
 
     def on_depth_reconstruction_btn_click(inputs):
         print("depth reconstruction")
-        image = inputs[input_image]
+        image = inputs[raw_image]
         path = dpt.generate_obj_masks(image=image, n_samples=inputs[n_samples], cube_size=inputs[cube_size], masks=inputs[masks])
         return {pcl_figure: path}
     depth_reconstruction_btn.click(on_depth_reconstruction_btn_click, components, [pcl_figure], queue=False)