#from PIL import Image
#from sys import argv, exit
import numpy as np
import cv2 as cv

from tensorflow.keras.utils import array_to_img as img

from matplotlib import pyplot as plt

#if len(argv) != 2:
#    exit(1)

#i = np.array(Image.open(argv[1]).convert('RGB').resize((300,300)))

from m import *
mod.load_weights("model3-val-acc-0.74.model.keras")

c = cv.VideoCapture(0)

while True:
    ret, frame = c.read()

    pi = np.array(img(cv.cvtColor(frame, cv.COLOR_BGR2RGB)).resize((300, 300)).convert("RGB"))
    #plt.imshow(pi)
    #plt.show()

    #print(pi.astype(float))

    r = mod.predict(np.array([pi.astype(float)]))

    cv.putText(frame, f"Y: {r[0][1]:.9f}", (10, 40), cv.FONT_HERSHEY_SIMPLEX, 1, (200,255,200), 2)
    cv.putText(frame, f"N: {r[0][0]:.9f}", (10, 70), cv.FONT_HERSHEY_SIMPLEX, 1, (200,200,255), 2)

    cv.imshow('me', frame)

    if cv.waitKey(1) == ord('q'):
        break

#print(frame)

c.release()
cv.destroyAllWindows()

#r = mod.predict(np.array([i]))
#print(r)