import os

from .face_alignment import image_align
from .landmarks_detector import LandmarksDetector


cur_dir = os.path.split(os.path.realpath(__file__))[0]
model_path = os.path.join(cur_dir, 'shape_predictor_68_face_landmarks.dat')


class FaceAlign:
    def __init__(self, predictor_model_path=model_path):
        self.landmarks_detector = LandmarksDetector(predictor_model_path)

    def get_crop_image(self, image):
        lms = []
        for i, face_landmarks in enumerate(self.landmarks_detector.get_landmarks(image), start=1):
            lms.append(face_landmarks)
        if len(lms) < 1:
            return None
        out_image = image_align(image, lms[0])

        return out_image