envs

app.py

@@ -304,10 +304,6 @@ class ImageConductor:
         id = base.split('_')[-1]
         
         
-        # with open(f'{output_dir}/points-{id}.json', 'w') as f:
-        #     json.dump(input_all_points, f)
-        
-        
         visualized_drag, _ = visualize_drag(first_frame_path, resized_all_points, drag_mode, self.width, self.height, self.model_length)
 
         ## image condition        
@@ -401,7 +397,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, first_frame_path_var: first_frame_path, tracking_points_var: tracking_points}
+    return {input_image: first_frame_path, output_image: None, output_video:None, 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
@@ -414,6 +410,7 @@ def add_tracking_points(tracking_points, first_frame_path, drag_mode, evt: gr.Se
     tracking_points[-1].append(evt.index)
     print(tracking_points)
     
+    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))
@@ -626,7 +623,7 @@ if __name__=="__main__":
             gr.Markdown(citation)
 
         
-        image_upload_button.upload(preprocess_image, [image_upload_button, tracking_points_var], [input_image, first_frame_path_var, tracking_points_var])
+        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])
 
         add_drag_button.click(add_drag, [tracking_points_var], tracking_points_var)