Update app.py

app.py

@@ -4,8 +4,10 @@ from typing import List, Tuple, Optional
 import spaces
 
 # Define the command to be executed
+command = ["python", "setup.py", "build_ext", "--inplace"]
 
 # Execute the command
+result = subprocess.run(command, capture_output=True, text=True)
 
 css="""
 div#component-18, div#component-25, div#component-35, div#component-41{
@@ -13,24 +15,15 @@ div#component-18, div#component-25, div#component-35, div#component-41{
 }
 """
 
-predictor = None
-
-
-
-
-@spaces.GPU(duration=120)
-def run_install(command):
-    result = subprocess.run(command, capture_output=True, text=True)
-    # Print the output and error (if any)
-    print("Output:\n", result.stdout)
-    print("Errors:\n", result.stderr)
-
-    # Check if the command was successful
-    if result.returncode == 0:
-        print("Command executed successfully.")
-    else:
-        print("Command failed with return code:", result.returncode)
+# Print the output and error (if any)
+print("Output:\n", result.stdout)
+print("Errors:\n", result.stderr)
 
+# Check if the command was successful
+if result.returncode == 0:
+    print("Command executed successfully.")
+else:
+    print("Command failed with return code:", result.returncode)
 
 import gradio as gr
 from datetime import datetime
@@ -68,11 +61,7 @@ def clear_points(image):
         #gr.State()     # stored_inference_state
     ]
 
-@spaces.GPU(duration=120) 
 def preprocess_video_in(video_path):
-    # command = ["python", "setup.py", "build_ext", "--inplace"]
-    command = ["pip", "install", "--no-build-isolation", "-e", "."]
-    run_install(command)
 
     # Generate a unique ID based on the current date and time
     unique_id = datetime.now().strftime('%Y%m%d%H%M%S')
@@ -250,7 +239,6 @@ def get_mask_sam_process(
     print("MODEL LOADED")
 
     # set predictor 
-    global predictor
     predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
     print("PREDICTOR READY")
 
@@ -325,13 +313,12 @@ def get_mask_sam_process(
         print(available_frames_to_check)
     
     # return gr.update(visible=True), "output_first_frame.jpg", frame_names, predictor, inference_state, gr.update(choices=available_frames_to_check, value=working_frame, visible=True)
-    return "output_first_frame.jpg", frame_names, inference_state, gr.update(choices=available_frames_to_check, value=working_frame, visible=False)
+    return "output_first_frame.jpg", frame_names, predictor, inference_state, gr.update(choices=available_frames_to_check, value=working_frame, visible=False)
 
 @spaces.GPU(duration=120)
 def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_names, video_frames_dir, vis_frame_type, available_frames_to_check, working_frame, progress=gr.Progress(track_tqdm=True)):   
     #### PROPAGATION ####
     sam2_checkpoint, model_cfg = load_model(checkpoint)
-    global predictor
     predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint)
     
     inference_state = stored_inference_state
@@ -432,7 +419,7 @@ def switch_working_frame(working_frame, scanned_frames, video_frames_dir):
 
 
 @spaces.GPU(duration=120) 
-def reset_propagation(first_frame_path, stored_inference_state):
+def reset_propagation(first_frame_path, predictor, stored_inference_state):
     
     predictor.reset_state(stored_inference_state)
     # print(f"RESET State: {stored_inference_state} ")
@@ -445,7 +432,7 @@ with gr.Blocks(css=css) as demo:
     trackings_input_label = gr.State([])
     video_frames_dir = gr.State()
     scanned_frames = gr.State()
-    # loaded_predictor = gr.State()
+    loaded_predictor = gr.State()
     stored_inference_state = gr.State()
     stored_frame_names = gr.State()
     available_frames_to_check = gr.State([])
@@ -601,7 +588,7 @@ with gr.Blocks(css=css) as demo:
         outputs = [
             output_result, 
             stored_frame_names, 
-            # loaded_predictor,
+            loaded_predictor,
             stored_inference_state,
             working_frame,
         ],
@@ -610,7 +597,7 @@ with gr.Blocks(css=css) as demo:
 
     reset_prpgt_brn.click(
         fn = reset_propagation,
-        inputs = [first_frame_path, stored_inference_state],
+        inputs = [first_frame_path, loaded_predictor, stored_inference_state],
         outputs = [points_map, tracking_points, trackings_input_label, output_propagated, stored_inference_state, output_result, available_frames_to_check, input_first_frame_image, working_frame, reset_prpgt_brn],
         queue=False
     )
@@ -626,6 +613,4 @@ with gr.Blocks(css=css) as demo:
         outputs = [output_propagated, output_video, working_frame, available_frames_to_check, reset_prpgt_brn]
     )
 
-
-
 demo.launch()