Update app.py

app.py

@@ -282,22 +282,39 @@ def analyze_stream(prompt, chatbot):
     print("Desired start_frame =", start_frame)
     print("Video start, fps, base_start_time =", video_start, fps, base_start_time)
 
-    # --- Manual frame skipping ---
-    # Read and discard frames until we reach `start_frame`
-    if start_frame > 0:
-        print(f"Skipping {start_frame} frames...")
+    print("Local file detected. Attempting direct frame seek...")
+    
+    # Attempt to seek
+    success = cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
+
+    # Check if seeking was actually successful by reading a frame
+    ret, test_frame = cap.read()
+    
+    if not success or not ret:
+        # If seeking failed, fall back to manual skipping
+        print(f"Direct seek to frame {start_frame} failed. Falling back to manual skipping.")
+        # Reset capture to start
+        cap.release()
+        cap = cv2.VideoCapture(stream)
+        # Skip frames manually
         for _ in range(start_frame):
             ret, _ = cap.read()
             if not ret:
-                # If the video ends or fails before skipping all frames, break
-                print("Failed before reaching start_frame.")
+                print("Failed before reaching start_frame (manual skip).")
                 break
-
-    print("Starting capture from the current position now.")
+        # We'll use 'test_frame' from the final read below
+        ret, test_frame = cap.read()
 
     frames = []
     start_time = time.time()
     clip_id = 0
+    
+    print("Starting capture from the current position now.")
+    if ret and test_frame is not None:
+        # We already read one frame after seeking, so store it
+        frames.append(test_frame)
+
+
 
     while not stop_capture:
         ret, frame = cap.read()