#!/usr/bin/python3

# Import libraries.
from PIL import Image
import numpy as np
import cv2
import os

# Clear screen.
os.system("clear")

# Print info.
print("[INFO] OpenCV running on v." + cv2.__version__)
print("[INFO] Classifier the face...")

# Main code.
def trainClassifier(dataPath):
	path = [os.path.join(dataPath,f) for f in os.listdir(dataPath)]
	faces = []
	ids = []

	for image in path:
		img = Image.open(image).convert("L")
		imageNp = np.array(img, 'uint8')
		id = int(os.path.split(image)[1].split(".")[1])
		faces.append(imageNp)
		ids.append(id)

	ids = np.array(ids)

	clf = cv2.face.LBPHFaceRecognizer_create()
	clf.train(faces, ids)
	clf.write("classifierCache.xml")

trainClassifier("dataset")

print("\n\n[INFO] Closing All Process")