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 = ("" "{key}" "{value}" "{mrz_value}" "".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('<', '<') value = value.replace(',', '

') row_value = ("" "{key}" "{value}" "{mrz_value}" "".format(key=key, value='', mrz_value=value)) table_value = table_value + row_value html = ("" "" "" "" "" "" "" "" "" "" "" "{table_value}" "
FieldValueMRZ
Status{status}
".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 = ("" "{key}" "" "".format(key=image_key, base64_image=image_value)) table_value = table_value + row_value images = ("" "" "" "" "" "{table_value}" "
FieldImage
".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.
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=7860)