import sys
sys.path.append('.')
sys.path.append('./face_recognition')
import os
import io
import cv2
import base64
import json
import gradio as gr
import requests
import numpy as np
from io import BytesIO
import configparser
import numpy as np
from PIL import Image

from face_recognition.match import match_1_1
# from face_recognition1.run import match_image


def face_recognition_on_file(file1, file2):
    img1 = cv2.imread(file1)
    img2 = cv2.imread(file2)

    response = match_1_1(img1, img2)

    return response


with gr.Blocks() as demo:
    gr.Markdown(
        """
    # FacePlugin Online Demo

    """
    )

    with gr.TabItem("Face Recognition"):
        with gr.Row():
            with gr.Column():
                first_input = gr.Image(type='filepath')
                gr.Examples(['images/rec_5.jpg', 'images/rec_1.jpg', 'images/9.png', 'images/rec_3.jpg'],
                            inputs=first_input)
                start_button = gr.Button("Run")
            with gr.Column():
                second_input = gr.Image(type='filepath')
                gr.Examples(['images/rec_6.jpg', 'images/rec_2.jpg', 'images/10.jpg', 'images/rec_4.jpg'],
                            inputs=second_input)

            with gr.Column():
                app_output = [gr.JSON()]

        start_button.click(face_recognition_on_file, inputs=[first_input, second_input], outputs=app_output)

demo.queue().launch(share=True)