envs

app.py

@@ -304,7 +304,7 @@ class ImageConductor:
         else:
             input_all_points = tracking_points.value
         
-        
+        print("input_all_points", input_all_points)
         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]
 
         dir, base, ext = split_filename(first_frame_path)
@@ -392,7 +392,7 @@ def reset_states(first_frame_path, tracking_points):
     return {input_image:None, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
 
 
-def preprocess_image(image):
+def preprocess_image(image, tracking_points):
     image_pil = image2pil(image.name)
     raw_w, raw_h = image_pil.size
     resize_ratio = max(384/raw_w, 256/raw_h)
@@ -401,7 +401,8 @@ def preprocess_image(image):
     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, first_frame_path_var: first_frame_path, tracking_points_var: gr.State([])}
+    tracking_points =  gr.State([])
+    return {input_image: first_frame_path, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
 
 
 def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.SelectData):  # SelectData is a subclass of EventData
@@ -438,7 +439,10 @@ def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.Se
 
 
 def add_drag(tracking_points):
-    tracking_points.value.append([])
+    print("before", tracking_points.value)
+
+    if tracking_points.value != []:
+        tracking_points.value.append([])
     print(tracking_points.value)
     return {tracking_points_var: tracking_points}
     
@@ -615,7 +619,6 @@ with block:
             inputs=[input_image, prompt, drag_mode, seed, personalized, examples_type],
             outputs=[input_image, prompt, drag_mode, seed, personalized, examples_type],
             fn=process_example,
-            run_on_click=True,
             examples_per_page=10,
             cache_examples=False,
         )
@@ -625,9 +628,9 @@ with block:
         gr.Markdown(citation)
 
     
-    image_upload_button.upload(preprocess_image, image_upload_button, [input_image, first_frame_path_var, tracking_points_var])
+    image_upload_button.upload(preprocess_image, [image_upload_button, tracking_points_var], [input_image, first_frame_path_var, tracking_points_var])
 
-    add_drag_button.click(add_drag, [tracking_points_var], tracking_points_var)
+    add_drag_button.click(add_drag, tracking_points_var, tracking_points_var)
 
     delete_last_drag_button.click(delete_last_drag, [tracking_points_var, first_frame_path_var, drag_mode], [tracking_points_var, input_image])