Spaces:
Running
Running
File size: 4,127 Bytes
5d2426e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 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('<', '<')
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) |