import gradio as gr
from deepface import DeepFace

# Charger le modèle VGG-Face avec ses poids
model = DeepFace.build_model("VGG-Face")

# Fonction pour comparer deux images de visages
def compare_faces(image1, image2):
    # Utiliser DeepFace pour vérifier si les deux images correspondent
    result = DeepFace.verify(image1, image2, model_name="VGG-Face")
    return result

# Interface Gradio pour télécharger et comparer les images
iface = gr.Interface(
    fn=compare_faces, 
    inputs=[gr.Image(type="filepath"), gr.Image(type="filepath")], 
    outputs="json",  
    title="Reconnaissance Faciale avec VGG-Face",
    description="Téléchargez deux images de visages pour vérifier si elles sont similaires. Le modèle utilisé est VGG-Face."
)

# Lancer l'interface Gradio
iface.launch()