Update app.py

app.py

@@ -10,8 +10,8 @@ import gradio as gr
 MODEL_PATH = "setosys_yolov12x.pt"
 model = YOLO(MODEL_PATH)
 
-# COCO dataset class ID for truck
-TRUCK_CLASS_ID = 7  # "truck"
+# COCO dataset class ID for people
+PERSON_CLASS_ID = 0  # "people"
 
 # Initialize SORT tracker
 tracker = Sort()
@@ -38,15 +38,15 @@ def determine_time_interval(video_filename):
     print("No keyword match, using default interval: 5")  # Debugging
     return 5  # Default interval
 
-def count_unique_trucks(video_path):
-    """ Counts unique trucks in a video using YOLOv12x and SORT tracking. """
+def count_unique_people(video_path):
+    """ Counts unique people in a video using YOLOv12x and SORT tracking. """
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
         return {"Error": "Unable to open video file."}
 
     # Reset variables at the start of each analysis
-    unique_truck_ids = set()
-    truck_history = {}
+    unique_people_ids = set()
+    people_history = {}
 
     # Get FPS of the video
     fps = int(cap.get(cv2.CAP_PROP_FPS))
@@ -87,7 +87,7 @@ def count_unique_trucks(video_path):
                 confidence = float(box.conf.item())  # Get confidence score
 
                 # Track only trucks
-                if class_id == TRUCK_CLASS_ID and confidence > CONFIDENCE_THRESHOLD:
+                if class_id == PEOPLE_CLASS_ID and confidence > CONFIDENCE_THRESHOLD:
                     x1, y1, x2, y2 = map(int, box.xyxy[0])  # Get bounding box
                     detections.append([x1, y1, x2, y2, confidence])
 
@@ -104,34 +104,34 @@ def count_unique_trucks(video_path):
         print(f"Frame {frame_count}: Tracked Objects -> {tracked_objects}")
 
         for obj in tracked_objects:
-            truck_id = int(obj[4])  # Unique ID assigned by SORT
+            people_id = int(obj[4])  # Unique ID assigned by SORT
             x1, y1, x2, y2 = obj[:4]  # Get the bounding box coordinates
 
-            truck_center = (x1 + x2) / 2, (y1 + y2) / 2  # Calculate truck center
+            people_center = (x1 + x2) / 2, (y1 + y2) / 2  # Calculate people center
 
-            # If truck is already in history, check movement distance
-            if truck_id in truck_history:
-                last_position = truck_history[truck_id]["position"]
-                distance = np.linalg.norm(np.array(truck_center) - np.array(last_position))
+            # If people is already in history, check movement distance
+            if people_id in people_history:
+                last_position = people_history[people_id]["position"]
+                distance = np.linalg.norm(np.array(people_center) - np.array(last_position))
 
                 if distance > DISTANCE_THRESHOLD:
-                    unique_truck_ids.add(truck_id)  # Add only if moved significantly
+                    unique_people_ids.add(people_id)  # Add only if moved significantly
 
             else:
-                # If truck is not in history, add it
-                truck_history[truck_id] = {
+                # If people is not in history, add it
+                people_history[truck_id] = {
                     "frame_count": frame_count,
-                    "position": truck_center
+                    "position": people_center
                 }
-                unique_truck_ids.add(truck_id)
+                unique_people_ids.add(people_id)
 
     cap.release()
-    return {"Total Unique Trucks": len(unique_truck_ids)}
+    return {"Total Unique People": len(unique_people_ids)}
 
 
 # Gradio UI function
 def analyze_video(video_file):
-    result = count_unique_trucks(video_file)
+    result = count_unique_people(video_file)
     return "\n".join([f"{key}: {value}" for key, value in result.items()])
 
 # Define Gradio interface
@@ -139,8 +139,8 @@ iface = gr.Interface(
     fn=analyze_video,
     inputs=gr.Video(label="Upload Video"),
     outputs=gr.Textbox(label="Analysis Result"),
-    title="YOLOv12x Unique Truck Counter",
-    description="Upload a video to count unique trucks using YOLOv12x and SORT tracking."
+    title="YOLOv12x Unique People Counter",
+    description="Upload a video to count unique people using YOLOv12x and SORT tracking."
 )
 
 # Launch the Gradio app