rad / funciones.py
Moibe's picture
Procesa DNI Listo
e0b339d
raw
history blame
841 Bytes
import herramientas
import documentos
import time
from gradio_client import Client, handle_file
async def procesa_dni(image):
temp_image = await herramientas.imageToTemp(image)
client = Client("Moibe/api_rapicash")
dict_recibido = client.predict(
img=handle_file(temp_image),
lang="en",
api_name="/predict"
) #Ésto porque gradio envía un dict con headers y su key data es el que contiene nuestro result original que tenía en la API original.
#Aquí es donde personalizo el proceso:
textos_extraidos = herramientas.listaTextosExtraidos(dict_recibido)
#Campos DNI Panamá.
nombre, apellido, identificacion = documentos.dni(textos_extraidos)
print(f"Hola: {nombre}, {apellido} con identificación: {identificacion}")
return {
"nombre": nombre,
"apellido": apellido,
"identificacion": identificacion
}