class View:
    def draw_faces(self, img, faces):
        for (x, y, w, h) in faces:
            cx, cy = x + w//2, y + h//2
            cv2.ellipse(img, (cx, cy), (w//2, h//2), 0, 0, 360, (255, 255, 0), 2)
        return img

    def draw_features(self, img, features):
        if not features:
            return img

        for key in features:
            pts = features[key]
            for (x, y) in pts:
                px = int(x * img.shape[1])
                py = int(y * img.shape[0])
                cv2.circle(img, (px, py), 1, (0, 255, 0), -1)

        return img

    def show(self, img):
        cv2.imshow("FaceApp", img)

class FaceApp:
    def __init__(self):
        import cv2
        from app.config_model import ConfigModel
        from app.face_detector import FaceDetector
        from app.face_features_detector import FaceFeaturesDetector
        from app.view import View

        self.cv2 = cv2
        self.view = View()

        cfg = ConfigModel()
        self.model = cfg.load_model("yolo-face", "model/yolov9t-face-lindevs.pt")

        self.face_detector = FaceDetector(self.model)
        self.features_detector = FaceFeaturesDetector()

        self.cap = cv2.VideoCapture(0)

    def run(self):
        while True:
            ret, frame = self.cap.read()
            if not ret:
                break

            faces = self.face_detector.detect(frame)
            features = self.features_detector.extract(frame)

            frame = self.view.draw_faces(frame, faces)
            frame = self.view.draw_features(frame, features)
            self.view.show(frame)

            if self.cv2.waitKey(1) & 0xFF == ord('q'):
                break

        self.cap.release()
        self.cv2.destroyAllWindows()

if __name__ == "__main__":
    app = FaceApp()
    app.run()

-f https://download.pytorch.org/whl/cpu/torch_stable.html
torch==2.2.0+cpu
torchvision==0.17.0+cpu
torchaudio==2.2.0+cpu
opencv-python>=4.7.0
numpy>=1.26.0
ultralytics>=8.0.167
mediapipe>=0.10.21