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])