import cv2
import inference
import supervision as sv

box_annotator = sv.BoxAnnotator(color=sv.Color(103, 6, 206))
trace_annotator = sv.TraceAnnotator(color=sv.Color(163, 81, 251), thickness=6)
byte_tracker = sv.ByteTrack()


def render(detections, image):
    detections = sv.Detections.from_roboflow(detections)
    detections = byte_tracker.update_with_detections(detections)
    if detections.tracker_id is not None and len(detections.tracker_id) > 0:
        image = trace_annotator.annotate(scene=image, detections=detections)
    image = box_annotator.annotate(scene=image, detections=detections)

    cv2.imshow("Prediction", image)
    cv2.waitKey(1)


inference.Stream(
    source="webcam",
    model="playing-cards-ow27d/1",
    # model="microsoft-coco/9",
    output_channel_order="BGR",
    use_main_thread=True,
    on_prediction=render,
)