import torch
import gradio as gr
import matplotlib.pyplot as plt
from sentence_transformers import SentenceTransformer
from sklearn.metrics.pairwise import cosine_similarity


model = SentenceTransformer('clip-ViT-B-32')



def predict(im1, im2):
  embeding = model.encode([im1, im2])
  sim = cosine_similarity(embeding)
  sim = sim[0][1]
  if sim > 0.78: # THRESHOLD HERE
    return sim, "SAME PERSON, UNLOCK PHONE"
  else:
    return sim, "DIFFERENT PEOPLE, DON'T UNLOCK"



import gradio as gr


title = 'Face ID'
description = 'This model detects the similarity between two images and passes a command!'


article = """
            Upload and Image from your Device or Make use of your webcam 
          """

img_upload = gr.Interface(
    fn=predict, 
    inputs= [gr.Image(type="pil", source="upload"), 
             gr.Image(type="pil", source="upload")], 
    outputs= [gr.Number(label="Similarity"),
              gr.Textbox(label="Message")],
    title=title,
    description=description,
    article=article
    )

webcam_upload = gr.Interface(
    fn=predict, 
    inputs= [gr.Image(type="pil", source="webcam"), 
            gr.Image(type="pil", source="webcam")], 
    outputs= [gr.Number(label="Similarity"),
              gr.Textbox(label="Message")],
    title=title,
    description=description,
    article=article,
    )

face_id = gr.TabbedInterface(
    [img_upload, webcam_upload], 
    ["Upload-Image", "Use Webcam"])

face_id.launch(debug=True)