Change how gr.State is used

app.py

@@ -1,5 +1,6 @@
 import json
 import os
+import uuid
 
 import cv2
 import gradio as gr
@@ -418,8 +419,8 @@ class ImageConductor:
 
 
 def reset_states(first_frame_path, tracking_points):
-    first_frame_path = gr.State()
-    tracking_points = gr.State([])
+    first_frame_path = None
+    tracking_points = []
     return {input_image: None, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
 
 
@@ -432,7 +433,7 @@ def preprocess_image(image, tracking_points):
     id = str(uuid.uuid4())[:4]
     first_frame_path = os.path.join(output_dir, f"first_frame_{id}.jpg")
     image_pil.save(first_frame_path, quality=95)
-    tracking_points = gr.State([])
+    tracking_points = []
     return {
         input_image: first_frame_path,
         first_frame_path_var: first_frame_path,
@@ -449,26 +450,16 @@ def add_tracking_points(
     elif drag_mode == "camera":
         color = (0, 0, 255, 255)
 
-    if not isinstance(tracking_points, list):
-        print(f"You selected {evt.value} at {evt.index} from {evt.target}")
-        tracking_points.value[-1].append(evt.index)
-        print(tracking_points.value)
-        tracking_points_values = tracking_points.value
-    else:
-        try:
-            tracking_points[-1].append(evt.index)
-        except Exception as e:
-            tracking_points.append([])
-            tracking_points[-1].append(evt.index)
-            print(f"Solved Error: {e}")
-
-        tracking_points_values = tracking_points
+    print(f"You selected {evt.value} at {evt.index} from {evt.target}")
+    if not tracking_points:
+        tracking_points = [[]]
+    tracking_points[-1].append(evt.index)
 
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     w, h = transparent_background.size
     transparent_layer = np.zeros((h, w, 4))
 
-    for track in tracking_points_values:
+    for track in tracking_points:
         if len(track) > 1:
             for i in range(len(track) - 1):
                 start_point = track[i]
@@ -497,12 +488,7 @@ def add_tracking_points(
 
 
 def add_drag(tracking_points):
-    if not isinstance(tracking_points, list):
-        # print("before", tracking_points.value)
-        tracking_points.value.append([])
-        # print(tracking_points.value)
-    else:
-        tracking_points.append([])
+    tracking_points.append([])
     return {tracking_points_var: tracking_points}
 
 
@@ -511,11 +497,11 @@ def delete_last_drag(tracking_points, first_frame_path, drag_mode):
         color = (255, 0, 0, 255)
     elif drag_mode == "camera":
         color = (0, 0, 255, 255)
-    tracking_points.value.pop()
+    tracking_points.pop()
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     w, h = transparent_background.size
     transparent_layer = np.zeros((h, w, 4))
-    for track in tracking_points.value:
+    for track in tracking_points:
         if len(track) > 1:
             for i in range(len(track) - 1):
                 start_point = track[i]
@@ -548,11 +534,11 @@ def delete_last_step(tracking_points, first_frame_path, drag_mode):
         color = (255, 0, 0, 255)
     elif drag_mode == "camera":
         color = (0, 0, 255, 255)
-    tracking_points.value[-1].pop()
+    tracking_points[-1].pop()
     transparent_background = Image.open(first_frame_path).convert("RGBA")
     w, h = transparent_background.size
     transparent_layer = np.zeros((h, w, 4))
-    for track in tracking_points.value:
+    for track in tracking_points:
         if len(track) > 1:
             for i in range(len(track) - 1):
                 start_point = track[i]
@@ -603,7 +589,7 @@ with block:
         with gr.Row(equal_height=True):
             gr.Markdown(instructions)
 
-    first_frame_path_var = gr.State(value=None)
+    first_frame_path_var = gr.State()
     tracking_points_var = gr.State([])
 
     with gr.Row():