Update app.py

app.py

@@ -53,7 +53,7 @@ COLOR_PALETTE = [
 ]
 
 random_seed = 42
-video_length = 201
+video_length = 9000
 W = 1024
 H = W
 device = "cpu"
@@ -216,6 +216,14 @@ def get_source_info(video_state):
     del vr
     return (fps if fps and fps > 0 else 15.0), (w, h)
 
+def on_video_change(video_path):
+    if not video_path:
+        return gr.update()
+    vr = VideoReader(video_path, ctx=cpu(0))
+    n = len(vr)
+    del vr
+    return gr.update(maximum=max(9000, n), value=n, label=f"Tracking Frames N (video has {n})")
+
 def preprocess_for_removal(images, masks):
     out_images = []
     out_masks = []
@@ -480,7 +488,7 @@ with gr.Blocks() as demo:
             clear_btn = gr.Button("Clear All Clicks")
 
         with gr.Row(elem_id="my-btn"):
-            n_frames_slider = gr.Slider(minimum=1, maximum=201, value=81, step=1, label="Tracking Frames N")
+            n_frames_slider = gr.Slider(minimum=1, maximum=9000, value=81, step=1, label="Tracking Frames N")
             track_btn = gr.Button("Tracking")
         video_output = gr.Video(label="Tracking Result", elem_id="my-video")