Upload 15 files

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.8-slim
+RUN apt-get update -y
+RUN apt-get install -y libpcsclite-dev psmisc libcurl4-openssl-dev libssl-dev
+RUN mkdir -p /root/kby-ai-idcard
+WORKDIR /root/kby-ai-idcard
+COPY ./libidsdk.so .
+COPY ./libimutils.so_for_ubuntu22 /usr/lib/libimutils.so
+COPY ./idsdk.py .
+COPY ./app.py .
+COPY ./demo.py .
+COPY ./requirements.txt .
+COPY ./run.sh .
+COPY ./idcard_examples ./idcard_examples
+COPY ./data ./data
+RUN pip3 install -r requirements.txt
+CMD ["./run.sh"]
+EXPOSE 8080 9000

app.py

@@ -0,0 +1,132 @@
+import sys
+sys.path.append('.')
+
+import os
+import base64
+import json
+
+from flask import Flask, request, jsonify
+from idsdk import getMachineCode
+from idsdk import setActivation
+from idsdk import initSDK
+from idsdk import idcardRecognition
+
+licensePath = "license.txt"
+license = ""
+
+# Get a specific environment variable by name
+license = os.environ.get("LICENSE")
+
+# Check if the variable exists
+if license is not None:
+    print("Value of LICENSE:", license)
+else:
+    license = ""
+    try:
+        with open(licensePath, 'r') as file:
+            license = file.read().strip()
+    except IOError as exc:
+        print("failed to open license.txt: ", exc.errno)
+    print("license: ", license)
+
+machineCode = getMachineCode()
+print("machineCode: ", machineCode.decode('utf-8'))
+
+ret = setActivation(license.encode('utf-8'))
+print("activation: ", ret)
+
+ret = initSDK()
+print("init: ", ret)
+
+app = Flask(__name__) 
+
+@app.route('/idcard_recognition', methods=['POST'])
+def idcard_recognition():
+    try:
+        file = request.files['file']
+
+        base64_image = base64.b64encode(file.read()).decode('utf-8')
+        ret = idcardRecognition(base64_image.encode('utf-8'), "".encode('utf-8'))
+
+        if ret != None:
+            j = json.loads(ret)
+            j.update({"Status": "Ok"})
+            response = jsonify(j)
+        else:
+            response = jsonify({"Status": "Error"})
+    except:
+        response = jsonify({"Status": "Error"})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+@app.route('/idcard_recognition_multi_page', methods=['POST'])
+def idcard_recognition_multi_page():
+    try:
+        file1 = request.files['file1']
+        file2 = request.files['file2']
+
+        base64_image1 = base64.b64encode(file1.read()).decode('utf-8')
+        base64_image2 = base64.b64encode(file2.read()).decode('utf-8')
+        ret = idcardRecognition(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+
+        if ret != None:
+            j = json.loads(ret)
+            j.update({"Status": "Ok"})
+            response = jsonify(j)
+        else:
+            response = jsonify({"Status": "Error"})
+    except:
+        response = jsonify({"Status": "Error"})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+@app.route('/idcard_recognition_base64', methods=['POST'])
+def idcard_recognition_base64():
+    try:
+        content = request.get_json()
+        base64_image = content['base64']
+
+        ret = idcardRecognition(base64_image.encode('utf-8'), "".encode('utf-8'))
+
+        if ret != None:
+            j = json.loads(ret)
+            j.update({"Status": "Ok"})
+            response = jsonify(j)
+        else:
+            response = jsonify({"Status": "Error"})
+    except:
+        response = jsonify({"Status": "Error"})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+@app.route('/idcard_recognition_base64_multi_page', methods=['POST'])
+def idcard_recognition_base64_multi_page():
+    try:
+        content = request.get_json()
+        base64_image1 = content['base64_1']
+        base64_image2 = content['base64_2']
+
+        ret = idcardRecognition(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+
+        if ret != None:
+            j = json.loads(ret)
+            j.update({"Status": "Ok"})
+            response = jsonify(j)
+        else:
+            response = jsonify({"Status": "Error"})
+    except:
+        response = jsonify({"Status": "Error"})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 8080))
+    app.run(host='0.0.0.0', port=port)

demo.py

@@ -0,0 +1,119 @@
+import gradio as gr
+import requests
+import datadog_api_client
+from PIL import Image
+
+
+def idcard_recognition(frame):
+    url = "http://127.0.0.1:8080/idcard_recognition"
+    files = {'file': open(frame, 'rb')}
+
+    r = requests.post(url=url, files=files)
+
+    html = None
+    images = None
+    mrz = None
+
+    status = r.json().get('Status')
+    table_value = ""
+
+    if r.json().get('MRZ') is not None:
+        mrz = r.json().get('MRZ')
+
+    for key, value in r.json().items():
+        if key == 'Status' or key == 'Images' or key == 'MRZ' or key == 'Position':
+            continue
+
+        mrz_value = ''
+        if mrz is not None and mrz.get(key) is not None:
+            mrz_value = mrz[key]
+            del mrz[key]
+
+        row_value = ("<tr>"
+                        "<td>{key}</td>"
+                        "<td>{value}</td>"
+                        "<td>{mrz_value}</td>"
+                    "</tr>".format(key=key, value=value, mrz_value=mrz_value))
+        table_value = table_value + row_value
+
+
+    if mrz is not None:
+        for key, value in mrz.items():
+            if key == 'MRZ':
+                value = value.replace('<', '&lt;')
+                value = value.replace(',', '<p>')
+
+            row_value = ("<tr>"
+                            "<td>{key}</td>"
+                            "<td>{value}</td>"
+                            "<td>{mrz_value}</td>"
+                        "</tr>".format(key=key, value='', mrz_value=value))
+            table_value = table_value + row_value
+            
+
+    html = ("<table>"
+                "<tr>"
+                    "<th style=""width:20%"">Field</th>"
+                    "<th style=""width:40%"">Value</th>"
+                    "<th style=""width:40%"">MRZ</th>"
+                "</tr>"
+                "<tr>"
+                    "<td>Status</td>"
+                    "<td>{status}</td>"
+                    "<td></td>"
+                "</tr>"
+                "{table_value}"
+                "</table>".format(status=status, table_value=table_value))
+    
+    table_value = ""
+    for key, value in r.json().items():
+        if key == 'Images':
+            for image_key, image_value in value.items():
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td><img src=""data:image/png;base64,{base64_image} width = '200'  height= '100' /></td>"
+                            "</tr>".format(key=image_key, base64_image=image_value))
+                table_value = table_value + row_value
+
+    images = ("<table>"
+                "<tr>"
+                    "<th>Field</th>"
+                    "<th>Image</th>"
+                "</tr>"
+                "{table_value}"
+                "</table>".format(table_value=table_value))
+    
+    return [html, images]
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+    # KBY-AI
+    We offer SDKs for Face Recognition, Face Liveness Detection(Face Anti-Spoofing), and ID Card Recognition.<br/>
+    Besides that, we can provide several AI models and development services in machine learning.
+
+    ## Simple Installation & Simple API
+    ```
+    sudo docker pull kbyai/idcard-recognition:latest
+    sudo docker run -e LICENSE="xxxxx" -p 8082:8080 -p 9002:9000 kbyai/idcard-recognition:latest
+    ```      
+    ## KYC Verification Demo
+    https://github.com/kby-ai/KYC-Verification    
+    """
+    )
+    with gr.TabItem("ID Card Recognition"):
+        with gr.Row():
+            with gr.Column(scale=3):
+                id_image_input = gr.Image(type='filepath')
+                gr.Examples(['idcard_examples/1.jpg', 'idcard_examples/2.jpg', 'idcard_examples/3.jpg'], 
+                            inputs=id_image_input)
+                id_recognition_button = gr.Button("ID Card Recognition")
+            with gr.Column(scale=5):
+                id_result_output = gr.HTML()
+        
+            with gr.Column(scale=2):
+                image_result_output = gr.HTML()
+
+        id_recognition_button.click(idcard_recognition, inputs=id_image_input, outputs=[id_result_output, image_result_output])
+
+demo.launch(server_name="0.0.0.0", server_port=9000)

idsdk.py

@@ -0,0 +1,23 @@
+import os
+
+from ctypes import *
+
+libPath = os.path.abspath(os.path.dirname(__file__)) + '/libidsdk.so'
+idsdk = cdll.LoadLibrary(libPath)
+
+getMachineCode = idsdk.getMachineCode
+getMachineCode.argtypes = []
+getMachineCode.restype = c_char_p
+
+setActivation = idsdk.setActivation
+setActivation.argtypes = [c_char_p]
+setActivation.restype = c_int32
+
+initSDK = idsdk.initSDK
+initSDK.argtypes = []
+initSDK.restype = c_int32
+
+idcardRecognition = idsdk.idcardRecognition
+idcardRecognition.argtypes = [c_char_p, c_char_p]
+idcardRecognition.restype = c_char_p
+

requirements.txt

@@ -0,0 +1,4 @@
+flask
+flask-cors
+gradio==3.50.2
+datadog_api_client

run.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+cd /root/kby-ai-idcard
+exec python3 demo.py &
+exec python3 app.py