Update app.py

app.py

@@ -87,3 +87,22 @@ def count_unique_trucks(video_path):
     cap.release()
 
     return {"Total Unique Trucks": len(unique_truck_ids)}
+
+# Gradio UI function
+def analyze_video(video_file):
+    result = count_unique_trucks(video_file)
+    return "\n".join([f"{key}: {value}" for key, value in result.items()])
+
+# Define Gradio interface
+import gradio as gr
+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."
+)
+
+# Launch the Gradio app
+if __name__ == "__main__":
+    iface.launch()