import gradio as gr import os import yaml from src.face_texture import GetFaceTexture from src.face_symmetry import GetFaceSymmetry from src.face_demographics import GetFaceDemographics from src.face_proportions import GetFaceProportions def combined_fn(input_image, input_image_2): demographics_dict = GetFaceDemographics().main(input_image) golden_ratios_dict, equal_ratios_dict, face_landmarks_image = GetFaceProportions().main(input_image) face_symmetry_image, symmetry_dict = GetFaceSymmetry().main(input_image) face_image, face_texture_image, texture_dict = GetFaceTexture().main(input_image) results = { "Demographic predictions": demographics_dict, "Face proportions (golden ratio)": golden_ratios_dict, "Face proportions (equal ratio)": equal_ratios_dict, "Face symmetry metrics": symmetry_dict, "Face texture metrics": texture_dict } with open("parameters.yml", 'r') as file: data = yaml.safe_load(file) results_interpretation = data["results_interpretation"] return (results, results_interpretation, face_image, face_landmarks_image, face_symmetry_image, face_texture_image) gigi_hadid = os.path.join(os.path.dirname(__file__), "data/gigi_hadid.webp") jay_z = os.path.join(os.path.dirname(__file__), "data/jay_z.jpg") iface = gr.Interface( fn=combined_fn, inputs=[ gr.Image(type="pil", label="Upload Face 1", value=jay_z), gr.Image(type="pil", label="Upload Face 2", value=gigi_hadid) ], outputs=[ gr.JSON(label="Results"), gr.JSON(label="Results explainer"), gr.Image(type="pil", label="Extracted face"), gr.Image(type="pil", label="Face landmarks"), gr.Image(type="pil", label="Face symmetry"), gr.Image(type="pil", label="Extracted face texture"), ], title="Advanced Facial Feature Detector", description= """ JSON Output in HTML

Description: This tool analyses a facial image to predict age and gender, assess symmetry, evaluate proportions, and examine texture.

Instructions: For optimal results, upload a clear front-facing image (see example image). To do so, either drag and drop your photo or click on "Upload Face Image", then press 'Submit'.

Interpreting the results:

Other information:

""", theme=gr.themes.Soft(), live=False, ) iface.launch()