| import gradio as gr |
| import os |
| import requests |
|
|
| from PIL import Image |
|
|
| def face_compare(frame1, frame2): |
| url = "https://face.miniai.live/api/face_match" |
| files = {'image1': open(frame1, 'rb'), 'image2': open(frame2, 'rb')} |
| r = requests.post(url=url, files=files) |
| response = r.json() |
|
|
| detections = response.get("detections", []) |
| matches = response.get("match", []) |
| detection_rows = "" |
| match_rows = "" |
|
|
| |
| for detection in detections: |
| face_image = detection.get("face", "") |
| face_img_tag = f"<img src='data:image/png;base64,{face_image}' width='100' />" if face_image else "N/A" |
| first_face_index = detection.get("firstFaceIndex", "N/A") |
| second_face_index = detection.get("secondFaceIndex", "N/A") |
|
|
| detection_rows += f""" |
| <tr> |
| <td>{first_face_index}</td> |
| <td>{second_face_index}</td> |
| <td>{face_img_tag}</td> |
| </tr> |
| """ |
|
|
| |
| for match in matches: |
| first_face_index = match.get("firstFaceIndex", "N/A") |
| second_face_index = match.get("secondFaceIndex", "N/A") |
| similarity = match.get("similarity", "N/A") |
|
|
| match_rows += f""" |
| <tr> |
| <td>{first_face_index}</td> |
| <td>{second_face_index}</td> |
| <td>{similarity:.6f}</td> |
| </tr> |
| """ |
|
|
| |
| detections_table = f""" |
| <h3>Face Detection</h3> |
| <table border="1" style="border-collapse: collapse; width: 100%;"> |
| <tr> |
| <th>First Face Index</th> |
| <th>Second Face Index</th> |
| <th>Face Image</th> |
| </tr> |
| {detection_rows} |
| </table> |
| """ |
|
|
| matches_table = f""" |
| <h3>Matching Results</h3> |
| <table border="1" style="border-collapse: collapse; width: 100%;"> |
| <tr> |
| <th>First Face Index</th> |
| <th>Second Face Index</th> |
| <th>Similarity</th> |
| </tr> |
| {match_rows} |
| </table> |
| """ |
|
|
| return detections_table + matches_table |
|
|
| def check_liveness(frame): |
| url = "https://facelive.miniai.live/api/check_liveness" |
| files = {'image': open(frame, 'rb')} |
| r = requests.post(url=url, files=files) |
|
|
| html = None |
| table_value = "" |
|
|
| for key, value in r.json().items(): |
| row_value = ("<tr>" |
| "<td>{key}</td>" |
| "<td>{value}</td>" |
| "</tr>".format(key=key, value=value)) |
| table_value = table_value + row_value |
|
|
| html = ("<table>" |
| "<tr>" |
| "<th style=""width:30%"">Field</th>" |
| "<th style=""width:50%"">Value</th>" |
| "</tr>" |
| "{table_value}" |
| "</table>".format(table_value=table_value)) |
| |
| return html |
|
|
| |
| with gr.Blocks() as MiniAIdemo: |
| gr.Markdown( |
| """ |
| <a href="https://miniai.live" style="display: flex; align-items: center;"> |
| <img src="https://miniai.live/wp-content/uploads/2024/02/logo_name-1-768x426-1.png" style="width: 18%; margin-right: 15px;"/> |
| <div> |
| <p style="font-size: 40px; font-weight: bold; margin-right: 20px;">FaceRecognition-LivenessDetection SDK Demo</p> |
| <p style="font-size: 20px; margin-right: 0;">Experience our NIST FRVT Top Ranked FaceRecognition, iBeta 2 Certified Face Liveness Detection Engine</p> |
| </div> |
| </a> |
| <br/> |
| <div style="display: flex; justify-content: center; align-items: center;"> |
| <table style="text-align: center;"> |
| <tr> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://github.com/MiniAiLive"><img src="https://miniai.live/wp-content/uploads/2024/10/new_git-1-300x67.png" style="height: 50px; margin-right: 5px;" title="GITHUB"/></a></td> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://huggingface.co/MiniAiLive"><img src="https://miniai.live/wp-content/uploads/2024/10/new_hugging-1-300x67.png" style="height: 50px; margin-right: 5px;" title="HuggingFace"/></a></td> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://demo.miniai.live"><img src="https://miniai.live/wp-content/uploads/2024/10/new_gradio-300x67.png" style="height: 50px; margin-right: 5px;" title="Gradio"/></a></td> |
| </tr> |
| <tr> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://docs.miniai.live/"><img src="https://miniai.live/wp-content/uploads/2024/10/a-300x70.png" style="height: 50px; margin-right: 5px;" title="Documentation"/></a></td> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://www.youtube.com/@miniailive"><img src="https://miniai.live/wp-content/uploads/2024/10/Untitled-1-300x70.png" style="height: 50px; margin-right: 5px;" title="Youtube"/></a></td> |
| <td style="text-align: center; vertical-align: middle;"><a href="https://play.google.com/store/apps/dev?id=5831076207730531667"><img src="https://miniai.live/wp-content/uploads/2024/10/googleplay-300x62.png" style="height: 50px; margin-right: 5px;" title="Google Play"/></a></td> |
| </tr> |
| </table> |
| </div> |
| <br/> |
| """ |
| ) |
| with gr.Tabs(): |
| with gr.Tab("Face Recognition"): |
| with gr.Row(): |
| with gr.Column(): |
| im_match_in1 = gr.Image(type='filepath', height=300) |
| gr.Examples( |
| [ |
| "images/compare/demo-pic22.jpg", |
| "images/compare/demo-pic60.jpg", |
| "images/compare/demo-pic35.jpg", |
| "images/compare/demo-pic33.jpg", |
| "images/compare/demo-pic34.jpg", |
| ], |
| inputs=im_match_in1 |
| ) |
| with gr.Column(): |
| im_match_in2 = gr.Image(type='filepath', height=300) |
| gr.Examples( |
| [ |
| "images/compare/demo-pic41.jpg", |
| "images/compare/demo-pic32.jpg", |
| "images/compare/demo-pic39.jpg", |
| "images/compare/demo-pic61.jpg", |
| "images/compare/demo-pic40.jpg", |
| ], |
| inputs=im_match_in2 |
| ) |
| with gr.Column(): |
| txt_compare_out = gr.HTML() |
| btn_f_match = gr.Button("Check Comparing!", variant='primary') |
| btn_f_match.click(face_compare, inputs=[im_match_in1, im_match_in2], outputs=txt_compare_out) |
| with gr.Tab("Face Liveness Detection"): |
| with gr.Row(): |
| with gr.Column(): |
| im_liveness_in = gr.Image(type='filepath', height=300) |
| gr.Examples( |
| [ |
| "images/liveness/f_real_andr.jpg", |
| "images/liveness/f_fake_andr_mask3d.jpg", |
| "images/liveness/f_fake_andr_monitor.jpg", |
| "images/liveness/f_fake_andr_outline.jpg", |
| "images/liveness/f_fake_andr_outline3d.jpg", |
| "images/liveness/1.jpg", |
| "images/liveness/3.png", |
| "images/liveness/4.jpg", |
| ], |
| inputs=im_liveness_in |
| ) |
| btn_f_liveness = gr.Button("Check Liveness!", variant='primary') |
| with gr.Column(): |
| livness_result_output = gr.HTML() |
| btn_f_liveness.click(check_liveness, inputs=im_liveness_in, outputs=livness_result_output) |
| gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FMiniAiLive%2FFaceRecognition-LivenessDetection-Demo"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FMiniAiLive%2FFaceRecognition-LivenessDetection-Demo&label=VISITORS&labelColor=%2337d67a&countColor=%23ff8a65&style=plastic&labelStyle=none" /></a>') |
| if __name__ == "__main__": |
| MiniAIdemo.launch() |
