o 0j@sPddlmZddlmZmZddlmZmZddlm Z m Z Gddde Z d S) )Dict)Image ImageDraw) PINGFANG_FONT create_font) BaseCVResult JsonMixinc@sTeZdZdZdeeejffddZdeeeffddZdeeeffddZ d S) DocPreprocessorResultzdoc preprocessor resultreturncCs|ddddddddf}|ddddddddf}|d}|ddddddddf}|dd}|dd }|jd d \}}|jd d \} } |jd d \} } tt|| | } td || | | d fd}|t|d d ||f|t||d || | f|t||| d || | | ft|}gd}d|d|d|d<d|d|d <|| | g}d ||| g}tt |D]"}||}t |||dft j }|j d||| d g|d|dqd|i}|S)z Generate an image combining the original, rotated, and unwarping images. Returns: Dict[Image.Image]: A new image combining the original, rotated, and unwarping images Z input_imgNrot_imgangle output_imgmodel_settingsuse_doc_orientation_classifyuse_doc_unwarpingrRGB)rr)zOriginal Imagez Rotated ImagezUnwarping ImagezRotated Image (z, )zUnwarping Image ( )rrr)fillfontZpreprocessed_img)shapemaxrnewZpasteZ fromarrayrZDrawrangelenrrpathtext)selfimagerrrrrZh1Zw1h2Zw2Zh3Zw3hZimg_showZ draw_textZtxt_listZ region_w_listZ beg_w_listZtnotxtrZimgsr*t/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/doc_preprocessor/result.py_to_imgs<    &  zDocPreprocessorResult._to_imgcOLi}|d|d<|d|d<|d|d<|d|d<tj|g|Ri|S)axConverts the instance's attributes to a dictionary and then to a string. Args: *args: Additional positional arguments passed to the base class method. **kwargs: Additional keyword arguments passed to the base class method. Returns: Dict[str, str]: A dictionary with the instance's attributes converted to strings. input_path page_indexrr)r _to_strr%argskwargsdatar*r*r+r0As     zDocPreprocessorResult._to_strcOr-)a[ Converts the object's data to a JSON dictionary. Args: *args: Positional arguments passed to the JsonMixin._to_json method. **kwargs: Keyword arguments passed to the JsonMixin._to_json method. Returns: Dict[str, str]: A dictionary containing the object's data in JSON format. r.r/rr)r _to_jsonr1r*r*r+r5Rs     zDocPreprocessorResult._to_jsonN) __name__ __module__ __qualname____doc__rstrrr,r0r5r*r*r*r+r s 'r N) typingrZPILrrZ utils.fontsrrZ common.resultr r r r*r*r*r+s