Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import torch
 from ultralytics import YOLO
 from sort import Sort
+import gradio as gr
 
 # Load YOLOv12x model
 MODEL_PATH = "yolov12x.pt"
@@ -20,7 +21,7 @@ CONFIDENCE_THRESHOLD = 0.5
 # Distance threshold to avoid duplicate counts
 DISTANCE_THRESHOLD = 50
 
-def count_unique_trucks(video_path):
+def count_unique_trucks(video_path, time_interval=7):
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
         return "Error: Unable to open video file."
@@ -29,8 +30,14 @@ def count_unique_trucks(video_path):
     truck_history = {}
 
     # Get FPS of the video
-    fps = int(cap.get(cv2.CAP_PROP_FPS))
-    frame_skip = fps * 7  # Skip frames every 5 seconds
+    fps = cap.get(cv2.CAP_PROP_FPS)
+
+    # Ensure FPS is valid, fallback to default if needed
+    if fps is None or fps <= 0:
+        fps = 30  # Default fallback FPS
+
+    # Set frame skip dynamically based on FPS and time_interval
+    frame_skip = max(int(fps * time_interval), 1)  # Ensure frame_skip is at least 1
 
     frame_count = 0
 
@@ -41,7 +48,7 @@ def count_unique_trucks(video_path):
 
         frame_count += 1
         if frame_count % frame_skip != 0:
-            continue  # Skip frames to process only every 5 seconds
+            continue  # Skip frames dynamically
 
         # Run YOLOv12x inference
         results = model(frame, verbose=False)
@@ -89,21 +96,22 @@ def count_unique_trucks(video_path):
     return {"Total Unique Trucks": len(unique_truck_ids)}
 
 # Gradio UI function
-def analyze_video(video_file):
-    result = count_unique_trucks(video_file)
+def analyze_video(video_file, time_interval=7):
+    result = count_unique_trucks(video_file, time_interval)
     return "\n".join([f"{key}: {value}" for key, value in result.items()])
 
-# Define Gradio interface
-import gradio as gr
+# Define Gradio interface with time_interval as a dynamic input
 iface = gr.Interface(
     fn=analyze_video,
-    inputs=gr.Video(label="Upload Video"),
+    inputs=[
+        gr.Video(label="Upload Video"),
+        gr.Slider(minimum=1, maximum=20, step=1, value=7, label="Time Interval (Seconds)")
+    ],
     outputs=gr.Textbox(label="Analysis Result"),
     title="YOLOv12x Unique Truck Counter",
-    description="Upload a video to count unique trucks using YOLOv12x and SORT tracking."
+    description="Upload a video and adjust the time interval to count unique trucks using YOLOv12x and SORT tracking."
 )
 
 # Launch the Gradio app
 if __name__ == "__main__":
     iface.launch()
-