Fix

app.py

@@ -424,7 +424,7 @@ class Drag:
         image_end = Image.open(last_frame_path).convert("RGB")
         image_end = image_end.resize((self.width, self.height))
 
-        input_all_points = tracking_points.constructor_args["value"]
+        input_all_points = tracking_points
 
         sift_track_update = False
         anchor_points_flag = None
@@ -545,10 +545,9 @@ class Drag:
 
 
 def reset_states(first_frame_path, last_frame_path, tracking_points):
-    first_frame_path = gr.State()
-    last_frame_path = gr.State()
-    tracking_points = gr.State([])
-
+    first_frame_path = None
+    last_frame_path = None
+    tracking_points = []
     return first_frame_path, last_frame_path, tracking_points
 
 
@@ -566,7 +565,7 @@ def preprocess_image(image):
 
     image_pil.save(first_frame_path)
 
-    return first_frame_path, first_frame_path, gr.State([])
+    return first_frame_path, first_frame_path, []
 
 
 def preprocess_image_end(image_end):
@@ -583,22 +582,22 @@ def preprocess_image_end(image_end):
 
     image_end_pil.save(last_frame_path)
 
-    return last_frame_path, last_frame_path, gr.State([])
+    return last_frame_path, last_frame_path, []
 
 
 def add_drag(tracking_points):
-    tracking_points.constructor_args["value"].append([])
+    tracking_points.append([])
     return tracking_points
 
 
 def delete_last_drag(tracking_points, first_frame_path, last_frame_path):
-    tracking_points.constructor_args["value"].pop()
+    tracking_points.pop()
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     transparent_background_end = Image.open(last_frame_path).convert("RGBA")
     w, h = transparent_background.size
     transparent_layer = np.zeros((h, w, 4))
 
-    for track in tracking_points.constructor_args["value"]:
+    for track in tracking_points:
         if len(track) > 1:
             for i in range(len(track) - 1):
                 start_point = track[i]
@@ -634,13 +633,13 @@ def delete_last_drag(tracking_points, first_frame_path, last_frame_path):
 
 
 def delete_last_step(tracking_points, first_frame_path, last_frame_path):
-    tracking_points.constructor_args["value"][-1].pop()
+    tracking_points[-1].pop()
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     transparent_background_end = Image.open(last_frame_path).convert("RGBA")
     w, h = transparent_background.size
     transparent_layer = np.zeros((h, w, 4))
 
-    for track in tracking_points.constructor_args["value"]:
+    for track in tracking_points:
         if len(track) > 1:
             for i in range(len(track) - 1):
                 start_point = track[i]
@@ -679,14 +678,14 @@ def add_tracking_points(
     tracking_points, first_frame_path, last_frame_path, evt: gr.SelectData
 ):  # SelectData is a subclass of EventData
     print(f"You selected {evt.value} at {evt.index} from {evt.target}")
-    tracking_points.constructor_args["value"][-1].append(evt.index)
+    tracking_points[-1].append(evt.index)
 
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     transparent_background_end = Image.open(last_frame_path).convert("RGBA")
 
     w, h = transparent_background.size
     transparent_layer = 0
-    for idx, track in enumerate(tracking_points.constructor_args["value"]):
+    for idx, track in enumerate(tracking_points):
         # mask = cv2.imread(
         #     os.path.join(args.output_dir, f"mask_{idx+1}.jpg")
         # )