import numpy as np
import cv2 as cv

from LicensePlateAuthorizer import LicensePlateAuthorizer

cap = cv.VideoCapture(0)
lpa = LicensePlateAuthorizer()

if not cap.isOpened():
    print("Cannot open camera")
    exit()

while True:
    ret, frame = cap.read()
    if not ret:
        print("Can't receive frame (stream end?). Exiting ...")
        break

    processed_frame = lpa.forward(frame)
    if processed_frame is None:
        print("Processing error or no license plate detected. Skipping frame.")
        continue

    cv.imshow('frame', processed_frame)
    if cv.waitKey(1) == ord('q'):
        break

cap.release()
cv.destroyAllWindows()