added face comparison

app.py

@@ -75,8 +75,29 @@ HTML_SUCCESSFUL_PRECHECKS = """
     </ul>
 """
 
+HTML_FACE_COMPARISON_HEADER = """
+<header style="text-align: center; padding: 20px; border-bottom: 2px solid #cc3300;">
+    <h1>Demo of Face Comparison</h1>
+    <p style="font-size: 18px;">
+        To learn more, visit our website: <a href="https://dataspike.io/" target="_blank" style="font-size: 20px; text-decoration: none;">
+        https://dataspike.io/ </a>
+    </p>
+</header>
+"""
+
+HTML_FACE_COMPARISON_EXPLANATION = """
+<h3 style="color:#2E86C1;">🔍 Face Comparison Results Explanation</h3>
+<p>The system compares two photos and provides one of four results:</p>
+<ul>
+    <li><strong style="color:#28A745;">Same Person</strong>: The same person appears in both photos.</li>
+    <li><strong style="color:#DC3545;">Different Person</strong>: The photos show different individuals.</li>
+    <li><strong style="color:#F39C12;">Same Photo</strong>: Identical faces detected, possibly indicating a duplicate or edited image.</li>
+    <li><strong style="color:#6C757D;">Failure</strong>: Face comparison failed due to undetected or unclear faces.</li>
+</ul>
+"""
 
-def main(file_path):
+
+def check_liveness(file_path):
     if file_path:
         url = "https://api.dataspike.io/api/v4/upload/liveness-demo"
         headers = {"ds-api-token": os.getenv("API_KEY")}
@@ -107,38 +128,118 @@ def main(file_path):
         return liveness_result, "", False
 
 
-with gr.Blocks() as LivenessDemo:
-    header_box = gr.HTML(HTML_HEADER)
-    with gr.Row():
-        with gr.Column(scale=1):
-            input_img_path = gr.Image(label="Input Image", type="filepath", height=300)
-            gr.Examples(
-                [
-                    "images/real_client054_android_SD_scene01.jpg",
-                    "images/attack_client006_android_SD_ipad_video_scene01.jpg",
-                    "images/attack_client055_android_SD_iphone_video_scene01.jpg",
-                    "images/attack_client026_android_SD_printed_photo_scene01.jpg",
-                    "images/attack_deepfake_1.jpg",
-                    "images/attack_deepfake_3.jpg",
-                    "images/attack_mask.jpg",
-                ],
+def map_fc_result(api_status: str):
+    mapping = {
+        "FaceComparisonVerified": {"label": "Same Person", "color": "green"},
+        "FaceComparisonUnknown": {"label": "Failure", "color": "gray"},
+        "FaceComparisonFailedDifferentFaces": {
+            "label": "Different Person",
+            "color": "red",
+        },
+        "FaceComparisonFailedSamePhotos": {"label": "Same Photo", "color": "#F39C12"},
+    }
+    status = api_status["status"] if "status" in api_status.keys() else ""
+    return mapping.get(status, {"label": "Unknown", "color": "gray"})
+
+
+def compare_faces(image1_path, image2_path):
+    if image1_path and image2_path:
+        url = "https://api.dataspike.io/api/v4/upload/face-comparison-demo"
+        headers = {"ds-api-token": os.getenv("API_KEY")}
+        files = {"left": open(image1_path, "rb"), "right": open(image2_path, "rb")}
+        response = requests.post(url, headers=headers, files=files)
+        comparison_result = response.json()
+        result = map_fc_result(comparison_result)
+        return gr.Label(value=result["label"], color=result["color"]), True
+        # return comparison_result, True
+    else:
+        comparison_result = {
+            "overallStatus": "Failure",
+            "errors": ["Please submit both images before comparing"],
+        }
+        return comparison_result, "", False
+
+
+with gr.Blocks() as Demo:
+    with gr.Tabs():
+        with gr.Tab("Liveness Detection"):
+            header_box = gr.HTML(HTML_HEADER)
+            with gr.Row():
+                with gr.Column(scale=1):
+                    input_img_path = gr.Image(
+                        label="Input Image", type="filepath", height=300
+                    )
+                    gr.Examples(
+                        [
+                            "images/real_client054_android_SD_scene01.jpg",
+                            "images/attack_client006_android_SD_ipad_video_scene01.jpg",
+                            "images/attack_client055_android_SD_iphone_video_scene01.jpg",
+                            "images/attack_client026_android_SD_printed_photo_scene01.jpg",
+                            "images/attack_deepfake_1.jpeg",
+                            "images/attack_deepfake_2.png",
+                            "images/attack_deepfake_3.jpeg",
+                            "images/tiktok_3d_mask.jpeg",
+                        ],
+                        inputs=input_img_path,
+                    )
+                    check_button = gr.Button("Check Liveness!", variant="primary")
+                with gr.Column():
+                    liveness_result = gr.JSON(label="Liveness Result")
+                    explanation_box = gr.HTML("", visible=False)
+
+            state = gr.State(False)
+            check_button.click(
+                check_liveness,
                 inputs=input_img_path,
+                outputs=[liveness_result, explanation_box, state],
+            )
+            state.change(
+                lambda show: gr.update(visible=show),
+                inputs=state,
+                outputs=explanation_box,
+            )
+
+        with gr.Tab("Face Comparison"):
+            gr.HTML(HTML_FACE_COMPARISON_HEADER)
+            with gr.Row():
+                with gr.Column():
+                    image1_input = gr.Image(
+                        label="First Image", type="filepath", height=300
+                    )
+                    gr.Examples(
+                        [
+                            "images/brad_pitt_1.jpeg",
+                            "images/brad_pitt_2.jpeg",
+                            "images/angelina-jolie-1.jpeg",
+                            "images/angelina-jolie-2.jpeg",
+                        ],
+                        inputs=image1_input,
+                    )
+                with gr.Column():
+                    image2_input = gr.Image(
+                        label="Second Image", type="filepath", height=300
+                    )
+                    gr.Examples(
+                        [
+                            "images/brad_pitt_1.jpeg",
+                            "images/brad_pitt_2.jpeg",
+                            "images/angelina-jolie-1.jpeg",
+                            "images/angelina-jolie-2.jpeg",
+                        ],
+                        inputs=image2_input,
+                    )
+
+            compare_button = gr.Button("Compare Faces!", variant="primary")
+            comparison_result = gr.Label(label="Comparison Result")
+            gr.HTML(HTML_FACE_COMPARISON_EXPLANATION)
+
+            comparison_state = gr.State(False)
+            compare_button.click(
+                compare_faces,
+                inputs=[image1_input, image2_input],
+                outputs=[comparison_result, comparison_state],
             )
-            check_button = gr.Button("Check Liveness!", variant="primary")
-        with gr.Column():
-            liveness_result = gr.JSON(label="Liveness Result")
-            explanation_box = gr.HTML("", visible=False)
-
-        state = gr.State(False)
-        check_button.click(
-            main,
-            inputs=input_img_path,
-            outputs=[liveness_result, explanation_box, state],
-        )
-        state.change(
-            lambda show: gr.update(visible=show), inputs=state, outputs=explanation_box
-        )
 
 
 if __name__ == "__main__":
-    LivenessDemo.launch()
+    Demo.launch()