import gradio as gr
import face_recognition
import numpy as np
from PIL import Image

def match_faces(image1, image2):
    # 이미지에서 얼굴 인식 및 특징 추출
    face_encoding1 = face_recognition.face_encodings(image1)[0]
    face_encoding2 = face_recognition.face_encodings(image2)[0]
    
    # 두 얼굴 간의 유사도 측정 (유클리디안 거리)
    distance = np.linalg.norm(face_encoding1 - face_encoding2)
    
    # 유사도에 따라 결과 반환
    if distance < 0.6:
        return "얼굴이 매우 비슷합니다!"
    else:
        return "얼굴이 다릅니다."
    
# Gradio 인터페이스 생성
iface = gr.Interface(fn=match_faces,
                     inputs=[gr.inputs.Image(shape=(224, 224)), gr.inputs.Image(shape=(224, 224))],
                     outputs="text")

iface.launch()