main.py

import serial
import time
import cv2
from facial_detection import FacialDetection

time.sleep(2)


def write_position(x, y):
    time.sleep(.02)
    ser.write(f"X{x}Y{y}\n".encode())
    return 0

stream = cv2.VideoCapture(0)

faces = FacialDetection()

if not stream.isOpened():
    print("no stream")
    exit()

while True:
    ret, frame = stream.read()  # Read a frame from the camera
    if not ret:
        print("no more stream")
        break
    faces.refresh(frame)
    cv2.rectangle(frame, (faces.faceCoords[0], faces.faceCoords[1]), (faces.faceCoords[2], faces.faceCoords[3]), (0, 255, 0), 2)
    #cv2.circle(frame, (faces.laserCoords[0], faces.laserCoords[1]), 4, (255, 0, 0))
    cv2.imshow("Webcam", frame)
    #write_position(xSet, ySet)

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

stream.release()
cv2.destroyAllWindows()