import sys
import os
import gradio as gr
import json
import requests
from PIL import Image
css = """
.example-image img{
display: flex; /* Use flexbox to align items */
justify-content: center; /* Center the image horizontally */
align-items: center; /* Center the image vertically */
height: 300px; /* Set the height of the container */
object-fit: contain; /* Preserve aspect ratio while fitting the image within the container */
}
.example-image img{
display: flex; /* Use flexbox to align items */
justify-content: center; /* Center the image horizontally */
align-items: center; /* Center the image vertically */
height: 300px; /* Set the height of the container */
object-fit: contain; /* Preserve aspect ratio while fitting the image within the container */
.block-background {
# background-color: #202020; /* Set your desired background color */
border-radius: 5px;
}
}
"""
def find_key_in_dict(d, target_key):
for key, value in d.items():
if key == target_key:
return value
elif isinstance(value, dict): # If the value is a dictionary, search recursively
result = find_key_in_dict(value, target_key)
if result is not None:
return result
return None
def json_to_html_table(data, image_keys):
html = ""
for key, value in data.items():
if isinstance(value, dict):
html += f"| {key} |
"
for sub_key, sub_value in value.items():
if sub_key in image_keys:
html += f"| {sub_key} |  |
"
else:
html += f"| {sub_key} | {sub_value} |
"
else:
if key in image_keys:
html += f"| {key} |  |
"
else:
html += f"| {key} | {value} |
"
html += "
"
return html
def idcard_recognition(frame1, frame2):
url = "https://recognito-iddocumentrecognition.p.rapidapi.com/api/read_idcard"
files = None
if frame1 is not None and frame2 is not None:
files = {'image': open(frame1, 'rb'), 'image2': open(frame2, 'rb')}
elif frame1 is not None and frame2 is None:
files = {'image': open(frame1, 'rb')}
elif frame1 is None and frame2 is not None:
files = {'image': open(frame2, 'rb')}
else:
return ['', None]
headers = {"X-RapidAPI-Key": os.environ.get("API_KEY")}
r = requests.post(url=url, files=files, headers=headers)
images = None
rawValues = {}
image_table_value = ""
result_table_dict = {
'portrait':'',
'documentName':'',
'score':'',
'countryName':'',
'name':'',
'sex':'',
'address':'',
'dateOfBirth':'',
'dateOfIssue':'',
'dateOfExpiry':'',
'documentNumber':'',
}
if 'data' in r.json():
for key, value in r.json()['data'].items():
if key == 'image':
for image_key, image_value in value.items():
row_value = (""
"| {key} | "
"![]() | "
"
".format(key=image_key, base64_image=image_value))
image_table_value = image_table_value + row_value
images = (""
""
"| Field | "
"Image | "
"
"
"{image_table_value}"
"
".format(image_table_value=image_table_value))
for key, value in r.json().items():
if key == 'data':
if 'image' in value:
del value['image']
rawValues[key] = value
else:
rawValues[key] = value
for result_key in result_table_dict.keys():
result_table_dict[result_key] = find_key_in_dict(r.json(), result_key)
result = json_to_html_table(result_table_dict, {'portrait'})
json_result = json.dumps(rawValues, indent=6)
return [result, json_result, images]
def launch_demo():
with gr.Blocks(css=css) as demo:
gr.Markdown(
f"""
Recognito
www.recognito.vision
📘 Product Documentation
🏠 Visit Recognito
🤝 Contact us for our on-premise ID Document Verification SDKs deployment
"""
)
with gr.Row():
with gr.Column(scale=6):
with gr.Row():
with gr.Column(scale=3):
id_image_input1 = gr.Image(type='filepath', label='Front', elem_classes="example-image")
with gr.Column(scale=3):
id_image_input2 = gr.Image(type='filepath', label='Back', elem_classes="example-image")
with gr.Row():
id_examples = gr.Examples(
examples=[['examples/1_f.png', 'examples/1_b.png'],
['examples/2_f.png', 'examples/2_b.png'],
['examples/3_f.png', 'examples/3_b.png'],
['examples/4.png', None]],
inputs=[id_image_input1, id_image_input2],
outputs=None,
fn=idcard_recognition
)
with gr.Blocks():
with gr.Column(scale=4, min_width=400, elem_classes="block-background"):
id_recognition_button = gr.Button("ID Card Recognition", variant="primary", size="lg")
with gr.Tab("Key Fields"):
id_result_output = gr.HTML()
with gr.Tab("Raw JSON"):
json_result_output = gr.JSON()
with gr.Tab("Images"):
image_result_output = gr.HTML()
id_recognition_button.click(idcard_recognition, inputs=[id_image_input1, id_image_input2], outputs=[id_result_output, json_result_output, image_result_output])
gr.HTML('
')
demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)
if __name__ == '__main__':
launch_demo() 
