Update app.py

app.py

@@ -46,6 +46,20 @@ for f in os.listdir("data"):
     # known_persons.append(os.path.splitext(os.path.basename(f))[0])
     known_persons.append(os.path.basename(f))
 
+import tempfile
+from face_symmetrizer import FaceSym
+
+def left_right_sim(img):
+    tmpf = tempfile.NamedTemporaryFile(delete=False)
+    img.save(tmpf.name, format='png')
+    f = FaceSym(tmpf.name)
+    _, left, _, _, right, _ = f.get_symmetrized_images(idx=0)
+    tmpf.close()
+    os.unlink(tmpf.name)
+    left_encoding = face_recognition.face_encodings(left)[0]
+    right_encoding = face_recognition.face_encodings(right)[0]
+    diff = face_recognition.face_distance([left_encoding], right_encoding)
+    return 100 * (1 - diff)
 
 import gradio as gr
 
@@ -63,6 +77,10 @@ def greet(image_to_test):
         os.path.splitext(filepath)[0],
         100 * (1 - face_distance))
     img = face_recognition.load_image_file(os.path.join("data", filepath))
+    ret += "\n\n \
+    The similarity (symmetry score) of \
+    left and right face = {:.3}".format(
+        left_right_sim(image_to_test))
     return img, ret
 
 iface = gr.Interface(fn=greet, inputs=gr.Image(source="webcam", streaming=True, type="numpy"), outputs=["image", "text"])