import face_alignment
from face_alignment import *
from skimage import io
import cv2
import os
import json

fa = face_alignment.FaceAlignment(
    face_alignment.LandmarksType._2D, device='cuda', flip_input=False)

#preds = fa.get_landmarks_from_directory('./images/1/')
dir = './images/2/'


def createLandmarks(name):
    input = io.imread(dir + name)
    img = cv2.cvtColor(input, cv2.COLOR_BGR2RGB)
    preds = fa.get_landmarks(img)
    if preds == None:
        return
    print(len(preds[0]))
    m = [[int(preds[0][ii, 0]), int(preds[0][ii, 1])] for ii in range(68)]
    print(m)
    k = json.dumps(m).replace('"', '')
    with open('./ready/2/' + name.replace('.jpg', '.json'), 'w', encoding='utf-8') as f:
        f.write(k)

#dot = [255, 255, 255]

# for x in preds[0]:
# circle:
#cv2.circle(img, (int(x[0]), int(x[1])), 1, dot, -1)
# dot:
#img[int(x[1]), int(x[0])] = dot

#cv2.imwrite('./ready/2/' + name, img)


# createLandmarks('00000.jpg')

for file in os.listdir(dir):
    if file.endswith(".jpg"):
        createLandmarks(file)