"""
@file: run.py
@desc: expose the rest api using flask
"""

import sys
import cv2
import numpy as np
from feature_api import load_model, align, get_feature, match_feature

feature_extractor, device = load_model('Res50', 512, '0')

def match_image(src_numpy, dst_numpy):
    face_image = align(src_numpy, output_size=(112, 112))
    src_feat = get_feature(face_image, feature_extractor, device)

    face_image = align(dst_numpy, output_size=(112, 112))
    dst_feat = get_feature(face_image, feature_extractor, device)

    output = match_feature(src_feat, dst_feat)
    sim = 0 if output[0] < 0 else output[0]

    print("Matching Score: ", sim)

    if sim < 0.72:
        result = "Different Person"
    else:
        result = "Same Person"
    

    response = {
        "confidence": sim,
        "threshold": 0.72,
        "result": result
    }
    return response


if __name__ == '__main__':
    src_file = '../images/11.jpg'
    dst_file = '../images/4.jpg'

    src_img = cv2.imread(src_file)
    dst_img = cv2.imread(dst_file)

    print(match_image(src_img, dst_img))