Create app.py

app.py

@@ -0,0 +1,141 @@
+import os
+import requests
+import gradio as gr
+
+
+HTML_HEADER = """
+<header style="text-align: center; padding: 20px; border-bottom: 2px solid #cc3300;">
+    <h1>Demo of Face Liveness Detection</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_IQA_FAILED = """
+<h3 style="color: #cc3300;">Liveness Checks Not Performed</h3>
+<p>
+        The liveness checks could not be conducted because the quality of the submitted image is not acceptable. 
+        Please ensure that your selfie meets the following criteria for optimal quality:
+    </p>
+    <ul>
+        <li>The face should be clearly visible and not obscured.</li>
+        <li>The lighting should be adequate, avoiding extreme shadows or bright spots.</li>
+        <li>The image should be in focus, without any blurriness.</li>
+        <li>Avoid using filters or effects that alter the appearance of your face.</li>
+    </ul>
+    <p>
+        Please re-upload a clearer selfie to proceed with the liveness verification.
+    </p>
+"""
+
+HTML_HEAD_POSITION_FAILED = """
+<h4 style="color: #cc3300;">Liveness Checks Not Performed</h4>
+<p>
+    The liveness checks could not be conducted because the position of the head in the submitted image does not meet the required criteria. 
+    Please ensure that your selfie adheres to the following guidelines for optimal head positioning:
+</p>
+<ul>
+    <li>The entire head should be visible in the frame without any cropping.</li>
+    <li>The head should not be tilted more than 25 degrees in any direction.</li>
+    <li>The face should be facing forward, with both eyes clearly visible.</li>
+    <li>Avoid extreme angles or positions that obscure facial features.</li>
+</ul>
+<p>
+    Please re-upload a selfie that meets these criteria to proceed with the liveness verification.
+</p>
+"""
+
+HTML_SUCCESSFUL_PRECHECKS = """
+<h3>Explanation of Results</h3>
+<p>Below is an explanation of the checks we run on the uploaded selfie to determine its quality, head position, and whether the selfie is genuine or not.</p>
+
+<ul>
+    <li><strong>Overall Status</strong>: Indicates whether all checks were successful. The selfie must meet the requirements for image quality, head position, and pass the liveness checks for the status to be <em>success</em>.</li>
+
+    <li><strong>Ran Checks</strong>: A list of checks performed on the selfie. These include prechecks for image quality (e.g., ensuring the image is not blurred, the head is fully visible and not tilted beyond acceptable angles). If these prechecks pass, liveness checks are performed to assess if the selfie is genuine.</li>
+
+    <li><strong>Face Box</strong>: The coordinates of the detected face, represented as (x1, y1, x2, y2), which define a rectangle surrounding the face within the image.</li>
+
+    <li><strong>Head Position Metrics</strong>:
+        <ul>
+            <li><strong>Pitch</strong>: The up or down tilt of the head. A positive value indicates a downward tilt, while a negative value indicates an upward tilt.</li>
+            <li><strong>Yaw</strong>: The left or right rotation of the head. Positive values indicate a turn to the right, and negative values indicate a turn to the left.</li>
+            <li><strong>Roll</strong>: The sideways tilt of the head. Positive values show a tilt to the right, and negative values show a tilt to the left.</li>
+        </ul>
+    </li>
+
+    <li><strong>Eye Aspect Ratio (EAR)</strong>: A measure of eye openness. A low EAR value indicates the eyes may be closed or blinking. This is an important metric for determining whether the person is actively looking at the camera.</li>
+
+    <li><strong>Liveness Score</strong>: A score ranging from 0 to 1 that estimates the likelihood that the selfie is of a real person. A higher score means a higher probability of authenticity.</li>
+    </ul>
+"""
+
+
+def main(file_path):
+    if file_path:
+        url = "https://api.dataspike.io/api/v4/upload/liveness-demo"
+        headers = {"ds-api-token": os.getenv("API_KEY")}
+        files = {"file": open(file_path, "rb")}
+        response = requests.post(url, headers=headers, files=files)
+        liveness_result = response.json()
+
+        if (
+            "ran_checks" not in liveness_result.keys()
+            or len(liveness_result["ran_checks"]) == 0
+        ):
+            msg = ""
+        elif liveness_result["ran_checks"] == ["ImageQuality"]:
+            msg = HTML_IQA_FAILED
+        elif liveness_result["ran_checks"] == ["ImageQuality", "HeadPosition"]:
+            msg = HTML_HEAD_POSITION_FAILED
+        else:
+            msg = HTML_SUCCESSFUL_PRECHECKS
+
+        return liveness_result, msg, True
+    else:
+        liveness_result = {
+            "overallStatus": "Failure",
+            "errors": [
+                "The image is not found. Please, submit an image first and then click the button 'Check Liveness!'"
+            ],
+        }
+        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.jpeg",
+                    "images/attack_deepfake_2.png",
+                    "images/attack_deepfake_3.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(
+            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()