Update app.py

app.py

@@ -20,6 +20,27 @@ CONFIDENCE_THRESHOLD = 0.5
 # Distance threshold to avoid duplicate counts
 DISTANCE_THRESHOLD = 50
 
+# Dictionary to define keyword-based time intervals
+TIME_INTERVALS = {
+    "fixed": 2,    # Fixed camera video
+    "moving": 5,    # Moving camera video
+    "drone": 7,     # Drone footage (slower interval)
+    "dashcam": 6,   # Dashcam video
+    "highway": 7,   # Highway footage
+    "city": 8,      # City road footage
+}
+
+def 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 count_unique_trucks(video_path):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
@@ -34,16 +55,12 @@ def count_unique_trucks(video_path):
     # Extract filename from the path
     video_filename = os.path.basename(video_path).lower()
 
-    # Dynamically adjust time interval based on filename
-    if "fixed" in video_filename:
-        time_interval = 2  # Process every 10 seconds for fixed camera
-    elif "moving" in video_filename:
-        time_interval = 7  # Process every 5 seconds for moving camera
-    else:
-        time_interval = 5  # Default to 7 seconds
+    # Extract filename from the path and convert to lowercase
+    video_filename = os.path.basename(video_path).lower()
 
+    # Determine the dynamic time interval based on filename keywords
+    time_interval = determine_time_interval(video_filename)
     frame_skip = fps * time_interval  # Convert time interval to frame count
-    #frame_skip = fps * 7  # Skip frames every 5 seconds
 
     frame_count = 0