justin2341's picture
Update demo.py
8a9fbc7 verified
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 - ID Card Recognition
We offer SDKs for face recognition, liveness detection(anti-spoofing) and ID card recognition.
We also specialize in providing outsourcing services with a variety of technical stacks like AI(Computer Vision/Machine Learning), Mobile apps, and web apps.
##### KYC Verification Demo - https://github.com/kby-ai/KYC-Verification-Demo-Android
##### ID Capture Web Demo - https://id-document-recognition-react-alpha.vercel.app
##### Documentation - Help Center - https://docs.kby-ai.com
"""
)
with gr.TabItem("ID Card Recognition"):
gr.Markdown(
"""
##### Docker Hub - https://hub.docker.com/r/kbyai/idcard-recognition
```bash
sudo docker pull kbyai/idcard-recognition:latest
sudo docker run -e LICENSE="xxxxx" -p 8082:8080 -p 9002:9000 kbyai/idcard-recognition:latest
```
"""
)
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])
gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fkby-ai%2FIDCardRecognition"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fkby-ai%2FIDCardRecognition&countColor=%23263759" /></a>')
demo.launch(server_name="0.0.0.0", server_port=7860)