from yolo.yoloFace import YOLO_FACE
from vgg.vgg_face import MODEL_FACE
from database.retriever import BruteForceStore
import cv2


DB = BruteForceStore()

def pipeline(img):

    images = YOLO_FACE(img)
    for patch in images:
        embeddings = MODEL_FACE(patch)
        if DB(embeddings): return "Welcome!"
    
    return "Buzz off!!! petrichor me chor"

if __name__ == "__main__":
    
    import cv2
    
    img = cv2.imread('temp.jpg', cv2.IMREAD_UNCHANGED) 
    print(pipeline(img))