envs

app.py

@@ -302,7 +302,7 @@ class ImageConductor:
         if isinstance(tracking_points, list):
             input_all_points = tracking_points
         else:
-            input_all_points = tracking_points
+            input_all_points = tracking_points.value
         
         
         resized_all_points = [tuple([tuple([float(e1[0]*self.width/original_width), float(e1[1]*self.height/original_height)]) for e1 in e]) for e in input_all_points]
@@ -394,8 +394,6 @@ def reset_states(first_frame_path, tracking_points):
 
 
 def preprocess_image(image, tracking_points):
-    if len(tracking_points) != 0:
-        tracking_points = gr.State([])
     image_pil = image2pil(image.name)
     raw_w, raw_h = image_pil.size
     resize_ratio = max(384/raw_w, 256/raw_h)
@@ -404,7 +402,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)
-    return {input_image: first_frame_path, output_image: None, output_video:None, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
+    return {input_image: first_frame_path, first_frame_path_var: first_frame_path, tracking_points_var: gr.State([])}
 
 
 def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.SelectData):  # SelectData is a subclass of EventData
@@ -414,14 +412,13 @@ def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.Se
         color = (0, 0, 255, 255)
 
     print(f"You selected {evt.value} at {evt.index} from {evt.target}")
-    tracking_points[-1].append(evt.index)
-    print(tracking_points)
+    tracking_points.value[-1].append(evt.index)
+    print(tracking_points.value)
     
-    print("first_frame_path", first_frame_path)
     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:
+    for track in tracking_points.value:
         if len(track) > 1:
             for i in range(len(track)-1):
                 start_point = track[i]
@@ -442,8 +439,8 @@ def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.Se
 
 
 def add_drag(tracking_points):
-    tracking_points.append([])
-    print(tracking_points)
+    tracking_points.value.append([])
+    print(tracking_points.value)
     return {tracking_points_var: tracking_points}
     
 
@@ -452,11 +449,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.pop()
+    tracking_points.value.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:
+    for track in tracking_points.value:
         if len(track) > 1:
             for i in range(len(track)-1):
                 start_point = track[i]
@@ -481,11 +478,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[-1].pop()
+    tracking_points.value[-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:
+    for track in tracking_points.value:
         if len(track) > 1:
             for i in range(len(track)-1):
                 start_point = track[i]
@@ -622,6 +619,7 @@ if __name__=="__main__":
                 fn=process_example,
                 run_on_click=True,
                 examples_per_page=10,
+                cache_examples=False,
             )
             
             
@@ -629,7 +627,7 @@ if __name__=="__main__":
             gr.Markdown(citation)
 
         
-        image_upload_button.upload(preprocess_image, [image_upload_button, tracking_points_var], [input_image, output_image, output_video, first_frame_path_var, tracking_points_var])
+        image_upload_button.upload(preprocess_image, image_upload_button, [input_image, first_frame_path_var, tracking_points_var])
 
         add_drag_button.click(add_drag, [tracking_points_var], tracking_points_var)