o 0j#A @sddlZddlZddlZddlZddlZddlZddlmZmZddl Z ddl Z ddl m Z m Z mZddlmZddlmZmZddlmZddlmZd d lmZmZed r\ddlZed rjddlZd d lmZGdddeZ de!de!fddZ"de!de!fddZ#de!de!ddfddZ$ d5de!de!de%dee%fddZ&ed d e j'deee(fd!d"Z)ed d d5d#e!d$e!d%e%fd&d'Z* d5d(e+d)e,de!de%fd*d+Z-de%fd,d-Z.ed d(e+d)e,d.e!d/e!fd0d1Z/d.e!d2e+d/e!dej0fd3d4Z1dS)6N)ListOptional)Image ImageDraw ImageFont)logging)function_requires_depsis_dep_available) custom_open) PINGFANG_FONT) BaseCVResult JsonMixinzopencv-contrib-python pypdfium2) pdfium_lockc@s,eZdZddZddZdejfddZdS) FormulaRecResultcOs8t|}|dtj|g|Ri|d}d|iS)N input_imgres)copydeepcopypopr_to_str)selfargskwargsdata_strrt/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/formula_recognition/result.pyr*s  zFormulaRecResult._to_strcOs,t|}|dtj|g|Ri|S)Nr)rrrr_to_json)rrrrrrrr 0s  zFormulaRecResult._to_jsonreturnc Cst|d}ztWntjy(}ztdd|iWYd}~Sd}~wwt|d}t | d}t |}|durS|\}}}}|||||||f}t|}|j \} } ddg| dg| | gd| gg} zNt |j | |dd } t| } | j \} }|}t|| | }|||ftj}|jt| d }td||fd }||d || |jd dfd|iWStjy}ztd d|iWYd}~Sd}~ww)a Draws a recognized formula on an image. This method processes an input image to recognize and render a LaTeX formula. It overlays the rendered formula onto the input image and returns the combined image. If the LaTeX rendering engine is not installed or a syntax error is detected, it logs a warning and returns the original image. Returns: Image.Image: An image with the recognized formula rendered alongside the original image. rzPlease refer to 2.3 Formula Recognition Pipeline Visualization in Formula Recognition Pipeline Tutorial to install the LaTeX rendering engine at first.rN rec_formulaRGBrF)is_debug r'r')rrz3Syntax error detected in formula, rendering failed.)rZ fromarray env_valid subprocessCalledProcessErrorrwarningstrnparrayconvertcrop_white_areasizedraw_formula_moduleintresizeZLANCZOSwidthnewZpaste)rimageer"xywhxywhZ image_widthZ image_heightboxZ img_formula render_width render_heightZ resize_heightZ resize_widthZnew_image_widthZ new_imagerrr_to_img5sP            zFormulaRecResult._to_imgN)__name__ __module__ __qualname__rr rrArrrrr(s requationr!cCsVd}t|d}ddg}|D] }||vrd}nq|s)d|dddd}|S) a Wraps an equation in LaTeX environment tags if not already aligned. This function checks if a given LaTeX equation contains any alignment tags (`align` or `align*`). If the equation does not contain these tags, it wraps the equation in `equation` and `nonumber` tags. Args: equation (str): The LaTeX equation to be checked and potentially modified. Returns: str: The modified equation with appropriate LaTeX tags for alignment. F z begin{align}z begin{align*}Tz\begin{equation} z \nonumberz\end{equation})r,strip)rEZis_align begin_dictZ begin_symrrrget_align_equationks0  rIformulacCs"td}dd}|||}|S)Nz([^\x00-\x7F]+)cSsd|ddS)Nz\text{})group)matchrrrreplacersz)add_text_for_zh_formula..replacer)recompilesub)rJpatternrOZreplaced_formularrradd_text_for_zh_formulas  rT tex_file_pathcCsft|d$}d}||t|}t|}||d}||WddS1s,wYdS)a Generates a LaTeX file containing a specific equation. This function creates a LaTeX file at the specified file path, writing the necessary LaTeX preamble and wrapping the provided equation in a document structure. The equation is processed to ensure it includes alignment tags if necessary. Args: tex_file_path (str): The file path where the LaTeX file will be saved. equation (str): The LaTeX equation to be written into the file. r<a \documentclass[varwidth]{standalone} \usepackage{cite} \usepackage{amsmath,amssymb,amsfonts,upgreek} \usepackage{graphicx} \usepackage{textcomp} \usepackage{xeCJK} \DeclareMathSizes{14}{14}{9.8}{7} \pagestyle{empty} \makeatletter \def\x@arrow{\DOTSB\Relbar} \def\xlongequalsignfill@{\arrowfill@\x@arrow\Relbar\x@arrow} \newcommand{\xlongequal}[2][]{\ext@arrow 0099\xlongequalsignfill@{#1}{#2}} \def\xLongleftrightarrowfill@{\arrowfill@\Longleftarrow\Relbar\Longrightarrow} \newcommand{\xLongleftrightarrow}[2][]{\ext@arrow 0099\xLongleftrightarrowfill@{#1}{#2}} \def\xlongleftrightarrowfill@{\arrowfill@\longleftarrow\relbar\longrightarrow} \newcommand{\xlongleftrightarrow}[2][]{\ext@arrow 0099\xlongleftrightarrowfill@{#1}{#2}} \def\xLeftrightarrowfill@{\arrowfill@\Leftarrow\Relbar\Rightarrow} \newcommand{\xLeftrightarrow}[2][]{\ext@arrow 0099\xLeftrightarrowfill@{#1}{#2}} \def\xleftrightarrowfill@{\arrowfill@\leftarrow\relbar\rightarrow} \newcommand{\xleftrightarrow}[2][]{\ext@arrow 0099\xleftrightarrowfill@{#1}{#2}} \def\xLongleftarrowfill@{\arrowfill@\Longleftarrow\Relbar\Relbar} \newcommand{\xLongleftarrow}[2][]{\ext@arrow 0099\xLongleftarrowfill@{#1}{#2}} \def\xLongrightarrowfill@{\arrowfill@\Relbar\Relbar\Longrightarrow} \newcommand{\xLongrightarrow}[2][]{\ext@arrow 0099\xLongrightarrowfill@{#1}{#2}} \def\xlongleftarrowfill@{\arrowfill@\longleftarrow\relbar\relbar} \newcommand{\xlongleftarrow}[2][]{\ext@arrow 0099\xlongleftarrowfill@{#1}{#2}} \def\xlongrightarrowfill@{\arrowfill@\relbar\relbar\longrightarrow} \newcommand{\xlongrightarrow}[2][]{\ext@arrow 0099\xlongrightarrowfill@{#1}{#2}} \makeatother \begin{document} \begin{large} z\end{large} \end{document} N)r writerTrI)rUrEfpZstart_templateZ end_templaterrrgenerate_tex_files  !  "rXFtex_pathpdf_dirr$cCsXtj|r*d||}|rtj|dddSttjdtj|tjtj dddSdS)a Generates a PDF file from a LaTeX file using pdflatex. This function checks if the specified LaTeX file exists, and then runs pdflatex to generate a PDF file in the specified directory. It can run in debug mode to show detailed output or in silent mode. Args: tex_path (str): The path to the LaTeX file. pdf_dir (str): The directory where the PDF file will be saved. is_debug (bool, optional): If True, runs pdflatex with detailed output. Defaults to False. Returns: Optional[bool]: Returns True if the PDF was generated successfully, False if the LaTeX file does not exist, and None if an error occurred during the pdflatex execution. zGxelatex -interaction=nonstopmode -halt-on-error -output-directory={} {}T)shellr<)stdoutstderrr[N) ospathexistsformatr) check_callr devnullPIPESTDOUT)rYrZr$commandrrrgenerate_pdf_files    rgr7c Cs~t|d}t|tj}t|ddtj\}}t|tj tj \}}t |dkr=t t |\}}}}||||gSdS)a Finds and returns the bounding box of the non-white area in an image. This function converts an image to grayscale and uses binary thresholding to find contours. It then calculates the bounding rectangle around the non-white areas of the image. Args: image (np.ndarray): The input image as a NumPy array. Returns: Optional[List[int]]: A list [x, y, w, h] representing the bounding box of the non-white area, or None if no such area is found. uint8r'rN)r-r.Zastypecv2ZcvtColorZCOLOR_BGR2GRAY thresholdZTHRESH_BINARY_INVZ findContoursZ RETR_EXTERNALZCHAIN_APPROX_SIMPLElenZ boundingRectZ concatenate) r7gray_ZthreshZcontoursr:r;r<r=rrrr0s  r0pdf_pathimg_path is_paddingc Cstt|}zlt|dkrW|WddS|D]P}td}d}|j||d}|t|}|durn|\} } } } || | | | | | f}|r^t j |ddddt j dd}|W|WdSqW|n|wWddS1swYdS) a Converts a single-page PDF to an image, optionally cropping white areas and adding padding. Args: pdf_path (str): The path to the PDF file. img_path (str): The path where the image will be saved. is_padding (bool): If True, adds a 30-pixel white padding around the image. Returns: np.ndarray: The resulting image as a NumPy array, or None if the PDF is not single-page. rKNr)scaleZrotationr&)value) rpdfiumZ PdfDocumentrlcloser3renderZto_numpyr0rjZcopyMakeBorderBORDER_CONSTANT) rorprqZpdfDocpagerotateZzoomZimgr9r:r;r<r=rrrpdf2img s<     r|img_sizer>c Cs|\}}tT}tj|d}tj|d}tj|d} t||tj|r/t|||d} tj|r>t|| dd} | durK| WdSt ||dt j} | WdS1s_wYdS)a Draw box formula for module. Args: img_size (tuple): The size of the image as (width, height). box (list): The coordinates for the bounding box. formula (str): The LaTeX formula to render. is_debug (bool): If True, retains intermediate files for debugging purposes. Returns: np.ndarray: The resulting image with the formula or an error message. temp.textemp.pdftemp.jpgNFrqzRendering Failed) tempfileTemporaryDirectoryr^r_joinrXr`rgr|draw_box_txt_finer ) r}r>rJr$ box_width box_heighttdrU pdf_file_path img_file_path formula_imgimg_right_textrrrr21s&      $r2cCstH}tj|d}tj|d}tj|d}d}d}t||tj|r/t|||tj|rDt||dd}WddSWddS1sOwYdS)z Validates if the environment is correctly set up to convert LaTeX formulas to images. Returns: bool: True if the environment is valid and the conversion is successful, False otherwise. r~rrza+b=cFrN) rrr^r_rrXr`rgr|)rrUrrrJr$rrrrr(Ts      "r(txt font_pathc Cstt|dd|ddd|dd|ddd}tt|dd|ddd|dd|ddd}|d|krv|dkrvtd||fd}t|}|rot|||f|}|jddg|d|d | tj }n#td||fd}t|}|rt|||f|}|jddg|d|d t ddg|dg||gd|gg} t j |t j d } t| | } t j |t jd }tj|| |tjtjdd } | S) a Draw box text. Args: img_size (tuple): Size of the image as (width, height). box (list): List of four points defining the box, each point is a tuple (x, y). txt (str): The text to draw inside the box. font_path (str): Path to the font file to be used for drawing text. Returns: np.ndarray: Image array with the text drawn and transformed to fit the box. rr rrrKrtr#r&)rrr)fillfont)Zdtype)flagsZ borderModeZ borderValue)r3mathsqrtrr6rZDraw create_fonttextZ transposeZ ROTATE_270r-Zfloat32r.rjZgetPerspectiveTransformrhZwarpPerspectiveZ INTER_NEARESTry) r}r>rrrrZimg_textZ draw_textrZpts1Zpts2MrrrrrhsB<<   rszcCst|dd}tj||dd}ttjdddkr$||d}n||}||dkrAt||d|}tj||dd}|S)a[ Creates a font object with a size that ensures the text fits within the specified dimensions. Args: txt (str): The text to fit. sz (tuple): The target size as (width, height). font_path (str): The path to the font file. Returns: ImageFont.FreeTypeFont: A PIL font object at the appropriate size. rKg?zutf-8)encoding.rr%)r3rZtruetypePIL __version__splitgetsizeZ getlength)rrrZ font_sizerlengthrrrrs   r)F)2rrr^rPr)rtypingrrnumpyr-rrrrutilsrZ utils.depsr r Zutils.file_interfacer Z utils.fontsr Z common.resultrrrjrrvZutils.pdfium_lockrrr,rIrTrXboolrgZndarrayr3r0r|tuplelistr2r(rZ FreeTypeFontrrrrrsh    C% 7 & # 4