Upload 13 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+lib/libopyidrecog.so filter=lfs diff=lfs merge=lfs -text
+lib/libopyocr.so filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.8-slim
+WORKDIR /home/opyidrecog
+COPY ./opyidrecog.py .
+COPY ./app.py .
+COPY ./demo.py .
+COPY ./license.txt .
+COPY ./requirements.txt .
+COPY ./run.sh .
+COPY ./arial.ttf .
+COPY ./examples ./examples
+COPY ./model ./model
+COPY ./lib /usr/lib
+COPY ./lib/opyidrecog.so ./lib/opyidrecog.so
+RUN pip3 install -r requirements.txt
+RUN chmod a+x run.sh
+CMD ["./run.sh"]
+EXPOSE 9000 7860

app.py

@@ -0,0 +1,105 @@
+import sys
+sys.path.append('.')
+
+import os
+import numpy as np
+import base64
+import json
+import io
+
+from PIL import Image, ExifTags
+from flask import Flask, request, jsonify
+from opyidrecog import getHWID
+from opyidrecog import setLicenseKey
+from opyidrecog import initSDK
+from opyidrecog import processImage
+
+licenseKeyPath = "license.txt"
+license = ""
+
+try:
+    with open(licenseKeyPath, 'r') as file:
+        license = file.read().strip()
+except IOError as exc:
+    print("failed to open license.txt: ", exc.errno)
+print("License Key: ", license)
+
+hwid = getHWID()
+print("HWID: ", hwid.decode('utf-8'))
+
+ret = setLicenseKey(license.encode('utf-8'))
+print("Set License: ", ret)
+
+ret = initSDK("model".encode('utf-8'))
+print("Init: ", ret)
+
+app = Flask(__name__) 
+
+
+@app.route('/process_image', methods=['POST'])
+def process_image():
+    base64_image1 = ""
+    base64_image2 = ""
+    try:
+        file1 = request.files['image1']
+        base64_image1 = base64.b64encode(file1.read()).decode('utf-8')
+    except:
+        result = "Failed to open image1"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    try:
+        file2 = request.files['image2']
+        base64_image2 = base64.b64encode(file2.read()).decode('utf-8')
+    except:
+        base64_image2 = ""
+
+    result = processImage(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+
+    result_dict = json.loads(result.decode('utf-8'))
+    response = jsonify({"resultCode": "Ok", "result": result_dict})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+@app.route('/process_image_base64', methods=['POST'])
+def process_image_base64():
+    base64_image1 = ""
+    base64_image2 = ""
+
+    try:
+        content = request.get_json()
+        base64_image1 = content['image1_base64']
+    except:
+        result = "Failed to parse base64"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    try:
+        content = request.get_json()
+        base64_image2 = content['image2_base64']
+    except:
+        base64_image2 = ""
+
+    image_np = np.asarray(image)
+    result = processImage(image_np, image_np.shape[1], image_np.shape[0])
+
+    result = processImage(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+
+    result_dict = json.loads(result.decode('utf-8'))
+    response = jsonify({"resultCode": "Ok", "result": result_dict})
+
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 9000))
+    app.run(host='0.0.0.0', port=port)

demo.py

@@ -0,0 +1,250 @@
+import os
+import gradio as gr
+import requests
+import json
+import io
+from gradio.components import Image
+from PIL import Image as PILImage, ImageDraw, ImageFont  # This import may be needed if you're processing images
+
+from PIL import Image
+
+def process_image(image1, image2):
+    try:
+        img_bytes1 = io.BytesIO()
+        image1.save(img_bytes1, format="JPEG")
+        img_bytes1.seek(0)
+    except:
+        return ["Failed to open image", {"resultCode": "Error", "result": "Failed to open image"}]
+
+    try:
+        img_bytes2 = io.BytesIO()
+        image2.save(img_bytes2, format="JPEG")
+        img_bytes2.seek(0)
+    except:
+        pass   
+    
+    url = "http://127.0.0.1:9000/process_image"
+    files = {'image1': img_bytes1, 'image2': img_bytes2}
+    result = requests.post(url=url, files=files)
+    if result.ok:
+        json_result = result.json()
+        if json_result.get("resultCode") == "Error":
+            return ["Failed to process image", {"resultCode": "Error", "result": "Failed to process image"}]
+
+        html = None    
+        images = None
+        mrz = None
+        barcode = None
+
+        r = json_result.get("result", {})
+        table_value = ""
+
+        if r.get('MRZ') is not None:
+            mrz = r.get('MRZ')
+
+        if r.get('Barcode') is not None:
+            barcode = r.get('Barcode')
+
+        for key, value in r.items():
+            if key == 'Images' or key == 'MRZ' or key == 'Barcode':
+                continue
+
+            mrz_value = ''
+            if mrz is not None and mrz.get(key) is not None:
+                mrz_value = mrz[key]
+                del mrz[key]
+
+            barcode_value = ''
+            if barcode is not None and barcode.get(key) is not None:
+                barcode_value = barcode[key]
+                del barcode[key]
+
+            if mrz is not None and barcode is not None:
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                                "<td>{mrz_value}</td>"
+                                "<td>{barcode_value}</td>"
+                            "</tr>".format(key=key, value=value, mrz_value=mrz_value, barcode_value=barcode_value))
+            elif mrz is not None:
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                                "<td>{mrz_value}</td>"
+                            "</tr>".format(key=key, value=value, mrz_value=mrz_value))
+            elif barcode is not None:    
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                                "<td>{barcode_value}</td>"
+                            "</tr>".format(key=key, value=value, barcode_value=barcode_value))
+            else:
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                            "</tr>".format(key=key, value=value))
+
+            table_value = table_value + row_value
+
+        if mrz is not None and barcode 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>"
+                                "<td></td>"
+                            "</tr>".format(key=key, value='', mrz_value=value))
+                table_value = table_value + row_value
+
+            for key, value in barcode.items():
+                if key == 'Barcode':
+                    value = value.replace('<', '&lt;')
+                    value = value.replace(',', '<p>')
+
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                                "<td></td>"
+                                "<td>{barcode_value}</td>"
+                            "</tr>".format(key=key, value='', barcode_value=value))
+                table_value = table_value + row_value
+        elif 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                
+        elif barcode is not None:
+            for key, value in barcode.items():
+                if key == 'Barcode':
+                    value = value.replace('<', '&lt;')
+                    value = value.replace(',', '<p>')
+
+                row_value = ("<tr>"
+                                "<td>{key}</td>"
+                                "<td>{value}</td>"
+                                "<td>{barcode_value}</td>"
+                            "</tr>".format(key=key, value='', barcode_value=value))
+                table_value = table_value + row_value
+
+        if mrz is not None and barcode is not None:
+            html = ("<table>"
+                        "<tr>"
+                            "<th>Field</th>"
+                            "<th>OCR</th>"
+                            "<th>MRZ</th>"
+                            "<th>Barcode</th>"
+                        "</tr>"
+                        "{table_value}"
+                        "</table>".format(table_value=table_value))
+        elif mrz is not None:
+            html = ("<table>"
+                        "<tr>"
+                            "<th>Field</th>"
+                            "<th>OCR</th>"
+                            "<th>MRZ</th>"
+                        "</tr>"
+                        "{table_value}"
+                        "</table>".format(table_value=table_value))
+        elif barcode is not None:
+            html = ("<table>"
+                        "<tr>"
+                            "<th>Field</th>"
+                            "<th>OCR</th>"
+                            "<th>Barcode</th>"
+                        "</tr>"
+                        "{table_value}"
+                        "</table>".format(table_value=table_value))
+        else:
+            html = ("<table>"
+                        "<tr>"
+                            "<th>Field</th>"
+                            "<th>OCR</th>"
+                        "</tr>"
+                        "{table_value}"
+                        "</table>".format(table_value=table_value))
+
+        
+        table_value = ""
+        for key, value in r.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 + "<br/><br/>" +  images, json_result]
+    else:
+        return [result.text, {"resultCode": "Error", "result": result.text}]
+    
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        <div style="display: flex;align-items: center;">
+            <img alt="Opulentyn Logo" src="https://github.com/user-attachments/assets/5fc78032-bff2-4f7e-a174-7d64b22f506d" width="350"/>
+            <div>
+                <h1>ID Document Recognition</h1>
+                <p>We offer <b>on-premises</b> OCR and liveness check solutions available with a <b>perpetual license</b>.</p>
+            </div>
+        </div>
+        
+        ## 🤝 Talk to us
+        
+        <div style="display: flex; align-items: center;">
+            <a href="https://opulentyn.com" target="_blank">
+                <img src="https://img.shields.io/badge/Website-https%3A%2F%2Fopulentyn.com-blue?style=flat&logo=google-chrome&logoColor=white" alt="Website">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="mailto:support@opulentyn.com">
+                <img src="https://img.shields.io/badge/Email-support%40opulentyn.com-blue?style=flat&logo=gmail&logoColor=white" alt="Email">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://wa.me/13435013587" target="_blank">
+                <img src="https://img.shields.io/badge/WhatsApp-%2B13435013587-blue?logo=whatsapp&logoColor=green" alt="WhatsApp">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://join.slack.com/t/opulentyn/shared_invite/zt-2s230jtbq-dWBs8XUZcrYim~nUqiimSA" target="_blank">
+                <img src="https://img.shields.io/badge/Slack-support--sdk-blueviolet?style=flat&logo=slack&logoColor=white" alt="Slack">
+            </a>
+        </div>
+        """
+    )
+
+    with gr.Row():
+        with gr.Column():
+            image_input1 = gr.Image(type='pil', label='Front')
+            image_input2 = gr.Image(type='pil', label='Back')
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg', 'examples/3.jpg'], 
+                        inputs=image_input1)
+            process_button = gr.Button("Process")
+        with gr.Column():
+            with gr.TabItem("Output"):
+                html_output = gr.HTML()
+            with gr.TabItem("JSON"):
+                json_output = gr.JSON()
+    
+    process_button.click(process_image, inputs=[image_input1, image_input2], outputs=[html_output, json_output])
+
+    gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fopulentyn%2FMrzOCR"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fopulentyn%2FMrzOCR&countColor=%23263759" /></a>')
+
+demo.launch(server_name="0.0.0.0", server_port=7860)

lib/libopyidrecog.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56ed0414ebea3695aa232234a63343398c29c5f4a548dbfe33485e2999064cba
+size 2910123

lib/libopyocr.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25832bf9dfc72bdf374f23df29c03077b8831707acf5636cb9b93667602bb372
+size 97975464

model/data1.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a58c75df0d7958367bc1b7e1a8a2e9b27a0e49bd761b20a1b95e748c56068c3b
+size 243984360

model/data2.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:480bb5f9aa41f1a6944b75f17523753c73b76dfdf829f20b2f0ec2e2b9a5bbd3
+size 292893244

opyidrecog.py

@@ -0,0 +1,24 @@
+import os
+
+from ctypes import *
+from numpy.ctypeslib import ndpointer
+
+libPath = os.path.abspath(os.path.dirname(__file__)) + '/lib/libopyidrecog.so'
+libopyidrecog = cdll.LoadLibrary(libPath)
+
+getHWID = libopyidrecog.getHWID
+getHWID.argtypes = []
+getHWID.restype = c_char_p
+
+setLicenseKey = libopyidrecog.setLicenseKey
+setLicenseKey.argtypes = [c_char_p]
+setLicenseKey.restype = c_int32
+
+initSDK = libopyidrecog.initSDK
+initSDK.argtypes = [c_char_p]
+initSDK.restype = c_int32
+
+processImage = libopyidrecog.processImage
+processImage.argtypes = [c_char_p, c_char_p]
+processImage.restype = c_char_p
+

requirements.txt

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

run.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+cd /home/opyidrecog
+exec python3 demo.py &
+exec python3 app.py