o
    ,uh                     @   sR   d dl Z d dlZd dlmZ d dlZedZdej_dejd< dd Zd	d
 Z	dS )    N)YOLOzsrc/train/best.ptz,C:\Program Files\Tesseract-OCR\tesseract.exez C:\Program Files\Tesseract-OCR\\ZTESSDATA_PREFIXc                 C   s<   t | t j}t |dt jt jdd}d}tj||d}|S )z4
    Extrae texto de una imagen utilizando OCR.
             z--oem 3 --psm 6 -l spa)configN)cv2cvtColorCOLOR_BGR2GRAYadaptiveThresholdADAPTIVE_THRESH_MEAN_CTHRESH_BINARYpytesseractZimage_to_string)imagegraythreshZcustom_configtext r   dC:\Users\Equipo\Documents\GitHub\reconocimiento-facial-lambda\src\validators\validateidentitycard.pyextract_text	   s   r   c                 C   s   t | }t|}|d jr<g }|d jD ]!}tt|jd \}}}}|||||f }	t|	}
||
 qd|dS ddiS )uY   
    Procesa una imagen para detectar y validar la cédula de identidad ecuatoriana.
    r   u'   Cédula(s) detectada(s) y procesada(s).)messageextracted_textsr   u(   No se detectó una cédula en la imagen.N)	r   imreadmodelboxesmapintxyxyr   append)pathr   resultsr   boxx1y1x2y2roir   r   r   r   process_image   s   


r&   )
r   r   ultralyticsr   osr   Ztesseract_cmdenvironr   r&   r   r   r   r   <module>   s    
