Update app.py

app.py

@@ -43,15 +43,6 @@ TIME_INTERVALS = {
     "eleven": 11
 }
 
-def old_determine_time_interval(video_filename):
-    """
-    Determines the time interval based on keywords found in the video filename.
-    Defaults to 7 seconds if no matching keyword is found.
-    """
-    for keyword, interval in TIME_INTERVALS.items():
-        if keyword in video_filename:
-            return interval
-    return 5  # Default interval if no keyword matches
 
 def determine_time_interval(video_filename):
     print(f"Checking filename: {video_filename}")  # Debugging
@@ -79,7 +70,13 @@ def count_unique_trucks(video_path):
 
     # Determine the dynamic time interval based on filename keywords
     time_interval = determine_time_interval(video_filename)
-    frame_skip = fps * 7  # Convert time interval to frame count
+    # Get total frames in the video
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+
+    # Ensure frame_skip does not exceed total frames
+    frame_skip = min(fps * time_interval, total_frames)
+
+    #frame_skip = fps * time_interval  # Convert time interval to frame count
 
     frame_count = 0