from supervision.assets import download_assets, VideoAssets
import supervision as sv
import cv2
import os
from inference import get_model
from ultralytics import YOLO

print(sv.__version__)
HOME = os.getcwd()

download_assets(VideoAssets.VEHICLES)
VIDEO_PATH = VideoAssets.VEHICLES.value

sv.VideoInfo.from_video_path(video_path=VIDEO_PATH)
frame_generator = sv.get_video_frames_generator(source_path=VIDEO_PATH)
frame = next(iter(frame_generator))
sv.plot_image(image=frame, size=(4, 4))

RESULT_VIDEO_PATH = f"{HOME}/videos/vehicle-counting-result.mp4"