o 0jv~@s`ddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlZddlZddlmZddlmZdd lmZmZmZmZmZmZmZmZm Z Gd d d Z!Gd d d Z"GdddZ#ddZ$GdddZ%GdddZ&GdddZ'GdddZ(GdddZ)GdddZ*GdddZ+Gd d!d!Z,Gd"d#d#Z-dS)$N)abstractmethod)Path)AnyCallableDictListOptionalUnion)Image)logging) AudioWriter CSVWriter HtmlWriter ImageWriter JsonWriterMarkdownWriter TextWriter VideoWriter XlsxWriterc@s`eZdZdZdedefddZedee e ffddZ e dee e ffd d Z dd d Zd S) WordMixinzB Mixin class for adding Word (.docx) export capabilities. argskwargscO|j|jdSN) _save_funcsappend save_to_wordselfrrr!f/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/common/result/mixin.py__init__.zWordMixin.__init__returncCtz Convert the result to a Word-compatible format. Returns: Dict[str, Any]: A dictionary containing Word-compatible blocks and image data. NotImplementedErrorr r!r!r"_to_word1zWordMixin._to_wordcC|Sr')r+r*r!r!r"word;zWordMixin.wordNc Osddlm}m}t|}t|}||jd}|}||d|} |j|d| |dd|dd d } | | d S) z Save the result to a .docx file.) WordConverter save_imagesz.docximagesZ word_blocksoriginal_image_widthioriginal_image_heightr)abs_image_pathsr4r5N) converterr1r2r _get_input_fnstemr+convertgetsaveas_posix) r save_pathrrr1r2fn save_fileZ word_datar6docr!r!r"rEs   zWordMixin.save_to_wordr%N)__name__ __module__ __qualname____doc__listdictr#rrstrrr+propertyr.rr!r!r!r"r)s  rc@s\eZdZdedefddZedeee ffddZ e deee ffdd Z d d d Z d S) LatexMixinrrcOrr)rr save_to_latexrr!r!r"r#Yr$zLatexMixin.__init__r%cCr&)z Convert the result to a LaTeX-compatible format. Returns: Dict[str, Any]: A dictionary containing LaTeX-compatible blocks and image data. r(r*r!r!r" _to_latex\r,zLatexMixin._to_latexcCr-)zProperty to access the LaTeX-compatible data. Returns: Dict[str, Any]: A dictionary containing LaTeX-compatible blocks and image data. )rMr*r!r!r"latexfzLatexMixin.latexNc Osddlm}m}t|}t|}||jd}|}||d|} |j|d| d} t| ddd } | | Wd d S1sHwYd S) zSave the result to a .tex file.r0)LatexConverterr2z.texr3Z latex_blocks)r6wzutf-8)encodingN) r7rPr2rr8r9rMr:openr=write) r r>rrrPr2r?r@Z latex_datar6rNfr!r!r"rLos  "zLatexMixin.save_to_latexrB)rCrDrErGrHr#rrrIrrMrJrNrLr!r!r!r"rKXs rKc@s<eZdZdZedeeeffddZddZd dd ZdS) StrMixinz6Mixin class for adding string conversion capabilities.r%cCr-)zProperty to get the string representation of the result. Returns: Dict[str, str]: The string representation of the result. )_to_strr*r!r!r"rIr/z StrMixin.strcCsd|iS)aConvert the given result data to a string representation. Args: json_format (bool): If True, return a JSON formatted string. Default is False. indent (int): Number of spaces to indent for JSON formatting. Default is 4. ensure_ascii (bool): If True, ensure all characters are ASCII. Default is False. Returns: Dict[str, str]: The string representation of the result. resr!r*r!r!r"rWs zStrMixin._to_strNcCst|dS)z.Print the string representation of the result.N)r inforWr*r!r!r"printszStrMixin.printrB) rCrDrErFrJrrIrWrZr!r!r!r"rVs  rVcCst|tjtjfr t|St|tjtjfrt|St|tjr)dd| DSt|t j r9t |jdddSt|trB|St|trRtdd|DSt|ttfr`dd|DS|S) zHelper function to format data into a JSON-serializable format. Args: obj: The object to be formatted. Returns: Any: The formatted object. cSg|]}t|qSr! _format_data).0itemr!r!r" z _format_data..recordsF)ZorientZ force_asciicSsi|] \}}|t|qSr!r\)r^kvr!r!r" sz _format_data..cSr[r!r\)r^ir!r!r"r`ra) isinstancenpZfloat32Zfloat64floatZint32Zint64intZndarraytolistpd DataFramejsonloadsto_jsonrr=rHitemsrGtuple)objr!r!r"r]s     r]c @seZdZdZdddZdeeeeefffddZe deeeeefffdd Z dd ed e de de deddf ddZ dde d e de fddZ dde d e de ddfddZdS) JsonMixinz7Mixin class for adding JSON serialization capabilities.r%NcCst|_|j|jdSr)r _json_writerrr save_to_jsonr*r!r!r"r#szJsonMixin.__init__cCsdtt|iS)zConvert the object to a JSON-serializable format. Returns: Dict[str, Dict[str, Any]]: A dictionary representation of the object that is JSON-serializable. rX)r]copydeepcopyr*r!r!r"_to_jsonszJsonMixin._to_jsoncCr-)zProperty to get the JSON representation of the result. Returns: Dict[str, Dict[str, Any]]: The dict type JSON representation of the result. )ryr*r!r!r"rnr/zJsonMixin.jsonr Fr>indent ensure_asciirrc Osdd}|}||s?t|}|j} t|} |D]!} | | d| d}|jj||| g|R||d|qd St|dkrNt d|d|jj||t | d g|R||d|d S) aeSave the JSON representation of the object to a file. Args: save_path (str): The path to save the JSON file. If the save path does not end with '.json', it appends the base name and suffix of the input path. indent (int): The number of spaces to indent for pretty printing. Default is 4. ensure_ascii (bool): If False, non-ASCII characters will be included in the output. Default is False. *args: Additional positional arguments to pass to the underlying writer. **kwargs: Additional keyword arguments to pass to the underlying writer. cSt|\}}|duo|dkS)Nzapplication/json mimetypes guess_type file_path mime_type_r!r!r" _is_json_filez-JsonMixin.save_to_json.._is_json_filerz.jsonrzr{r0z`The result has multiple json files need to be saved. But the `save_path` has been specified as ``!rN) ryrr8r9rurTr=lenr warningrGkeys) r r>rzr{rrrZ json_datar?r9base_save_pathkeyr!r!r"rvsF    zJsonMixin.save_to_json json_formatcCs$|rtjtd|i||dSd|iS)aConvert the given result data to a string representation. Args: data (dict): The data would be converted to str. json_format (bool): If True, return a JSON formatted string. Default is False. indent (int): Number of spaces to indent for JSON formatting. Default is 4. ensure_ascii (bool): If True, ensure all characters are ASCII. Default is False. Returns: Dict[str, str]: The string representation of the result. rXr)rndumpsr])r rrzr{r!r!r"rW s zJsonMixin._to_strcCs|j|||d}t|dS)aUPrint the string representation of the result. Args: json_format (bool): If True, print a JSON formatted string. Default is False. indent (int): Number of spaces to indent for JSON formatting. Default is 4. ensure_ascii (bool): If True, ensure all characters are ASCII. Default is False. )rrzr{N)rWr rY)r rrzr{Zstr_r!r!r"rZ#s zJsonMixin.printrB)r F)Fr F)rCrDrErFr#rrIrryrJrnrjboolrrvrWrZr!r!r!r"rtsR    7 rtc@teZdZdZdededdfddZedeeeffdd Z e deeeffd d Z d edededdfd dZ dS) Base64Mixinz4Mixin class for adding Base64 encoding capabilities.rrr%NcO"t|i||_|j|jdS)zInitializes the Base64Mixin. Args: *args: Positional arguments to pass to the TextWriter. **kwargs: Keyword arguments to pass to the TextWriter. N)r_base64_writerrrsave_to_base64rr!r!r"r#6zBase64Mixin.__init__cCr&)zAbstract method to convert the result to Base64. Returns: Dict[str, str]: The str type Base64 representation result. r(r*r!r!r" _to_base64@zBase64Mixin._to_base64cCr-)z Property that returns the Base64 encoded content. Returns: Dict[str, str]: The base64 representation of the result. )rr*r!r!r"base64Ir/zBase64Mixin.base64r>c Os|}t|ds=t|}|j}t|}|D]}||d|d}|jj| ||g|Ri|qdSt |dkrLt d|d|jj||t |dg|Ri|dS)aSaves the Base64 encoded content to the specified path. Args: save_path (str): The path to save the base64 representation result. If the save path does not end with '.b64', it appends the base name and suffix of the input path. *args: Additional positional arguments that will be passed to the base64 writer. **kwargs: Additional keyword arguments that will be passed to the base64 writer. z.b64rr0zbThe result has multiple base64 files need to be saved. But the `save_path` has been specified as `rrN)rrIlowerendswithrr8r9rrTr=rr rrGr) r r>rrrr?r9rrr!r!r"rSs4      zBase64Mixin.save_to_base64) rCrDrErFrrr#rrIrrJrrr!r!r!r"r3s  rc @s~eZdZdZddedededdfdd Zedeee j ffd d Z e deee j ffd d Z dedededdfddZ dS)ImgMixinz3Mixin class for adding image handling capabilities.pillowbackendrrr%NcOs&t|d|i||_|j|jdS)a%Initializes ImgMixin. Args: backend (str): The backend to use for image processing. Defaults to "pillow". *args: Additional positional arguments to pass to the ImageWriter. **kwargs: Additional keyword arguments to pass to the ImageWriter. rN)r _img_writerrr save_to_imgr rrrr!r!r"r#sszImgMixin.__init__cCr&)zAbstract method to convert the result to an image. Returns: Dict[str, Image.Image]: The image representation result. r(r*r!r!r"_to_img~rzImgMixin._to_imgcCr-)zProperty to get the image representation of the result. Returns: Dict[str, Image.Image]: The image representation of the result. )rr*r!r!r"imgrOz ImgMixin.imgr>c Osdd}|}||sFt|}||r|jnd}|j}t|} |D]} | |d| |}|jj||| g|Ri|q$d St|dkrUt d|d|jj||t | dg|Ri|d S) aSaves the image representation of the result to the specified path. Args: save_path (str): The path to save the image. If the save path does not end with .jpg or .png, it appends the input path's stem and suffix to the save path. *args: Additional positional arguments that will be passed to the image writer. **kwargs: Additional keyword arguments that will be passed to the image writer. cS t|\}}|duo|dS)Nzimage/r~r startswithrr!r!r"_is_image_filez,ImgMixin.save_to_img.._is_image_filez.pngrr0z_The result has multiple img files need to be saved. But the `save_path` has been specified as `rrN) rrr8suffixr9rrTr=rr rrGr) r r>rrrrr?rr9rrr!r!r"rs   &  0zImgMixin.save_to_img)r)rCrDrErFrIrrr#rr rrJrrr!r!r!r"rp rc @~eZdZdZddedededdfdd Zedeee j ffd d Z e deee j ffd d Z dedededdfddZdS)CSVMixinz1Mixin class for adding CSV handling capabilities.pandasrrrr%NcOs6t|d|i||_t|dsg|_|j|jdS)aInitializes the CSVMixin. Args: backend (str): The backend to use for CSV operations (default is "pandas"). *args: Optional positional arguments to pass to the CSVWriter. **kwargs: Optional keyword arguments to pass to the CSVWriter. rrN)r _csv_writerhasattrrr save_to_csvrr!r!r"r#s zCSVMixin.__init__cCr-)zProperty to get the pandas Dataframe representation of the result. Returns: Dict[str, pd.DataFrame]: The pandas.DataFrame representation of the result. )_to_csvr*r!r!r"csvrOz CSVMixin.csvcCr&)zAbstract method to convert the result to pandas.DataFrame. Returns: Dict[str, pd.DataFrame]: The pandas.DataFrame representation result. r(r*r!r!r"rrzCSVMixin._to_csvr>c Odd}|}||s._is_csv_filerz.csvr0z_The result has multiple csv files need to be saved. But the `save_path` has been specified as `rrN) rrr8r9rrTr=rr rrGr) r r>rrrrr?r9rrr!r!r"rs  &  0zCSVMixin.save_to_csv)r)rCrDrErFrIrrr#rJrlrmrrrrr!r!r!r"rs rc@r) HtmlMixinz2Mixin class for adding HTML handling capabilities.rrr%NcOr)a Initializes the HTML writer and appends the save_to_html method to the save functions list. Args: *args: Positional arguments passed to the HtmlWriter. **kwargs: Keyword arguments passed to the HtmlWriter. N)r _html_writerrr save_to_htmlrr!r!r"r#szHtmlMixin.__init__cCr-)zProperty to get the HTML representation of the result. Returns: str: The str type HTML representation of the result. )_to_htmlr*r!r!r"htmlrOzHtmlMixin.htmlcCr&)zAbstract method to convert the result to str type HTML representation. Returns: Dict[str, str]: The str type HTML representation result. r(r*r!r!r"rrzHtmlMixin._to_htmlr>c Or) zSaves the HTML representation of the object to the specified path. Args: save_path (str): The path to save the HTML file. *args: Additional positional arguments. **kwargs: Additional keyword arguments. cSr|)Nz text/htmlr}rr!r!r" _is_html_filerz-HtmlMixin.save_to_html.._is_html_filerz.htmlr0z`The result has multiple html files need to be saved. But the `save_path` has been specified as `rrN) rrr8r9rrTr=rr rrGr) r r>rrrrr?r9rrr!r!r"rs6      zHtmlMixin.save_to_html) rCrDrErFrrr#rJrIrrrrr!r!r!r"rs rc@r) XlsxMixinz2Mixin class for adding XLSX handling capabilities.rrr%NcOr)aInitializes the XLSX writer and appends the save_to_xlsx method to the save functions. Args: *args: Positional arguments to be passed to the XlsxWriter constructor. **kwargs: Keyword arguments to be passed to the XlsxWriter constructor. N)r _xlsx_writerrr save_to_xlsxrr!r!r"r#2rzXlsxMixin.__init__cCr-)zProperty to get the XLSX representation of the result. Returns: Dict[str, str]: The str type XLSX representation of the result. )_to_xlsxr*r!r!r"xlsx<rOzXlsxMixin.xlsxcCr&)zAbstract method to convert the result to str type XLSX representation. Returns: Dict[str, str]: The str type HTML representation result. r(r*r!r!r"rErzXlsxMixin._to_xlsxr>c Or) aSaves the HTML representation to an XLSX file. Args: save_path (str): The path to save the XLSX file. If the path does not end with ".xlsx", the filename will be set to the stem of the input path with ".xlsx" extension. *args: Additional positional arguments to pass to the XLSX writer. **kwargs: Additional keyword arguments to pass to the XLSX writer. cSr|)NzAapplication/vnd.openxmlformats-officedocument.spreadsheetml.sheetr}rr!r!r" _is_xlsx_fileXs z-XlsxMixin.save_to_xlsx.._is_xlsx_filerz.xlsxr0z`The result has multiple xlsx files need to be saved. But the `save_path` has been specified as `rrN) rrr8r9rrTr=rr rrGr) r r>rrrrr?r9rrr!r!r"rNs6      zXlsxMixin.save_to_xlsx) rCrDrErFrrr#rJrIrrrrr!r!r!r"r/s rc @r) VideoMixinz3Mixin class for adding Video handling capabilities.opencvrrrr%NcOs||_|j|jdS)a'Initializes VideoMixin. Args: backend (str): The backend to use for video processing. Defaults to "opencv". *args: Additional positional arguments to pass to the VideoWriter. **kwargs: Additional keyword arguments to pass to the VideoWriter. N)_backendrr save_to_videorr!r!r"r#wszVideoMixin.__init__cCr&)zAbstract method to convert the result to a video. Returns: Dict[str, np.array]: The video representation result. r(r*r!r!r" _to_videorzVideoMixin._to_videocCr-)zProperty to get the video representation of the result. Returns: Dict[str, np.array]: The video representation of the result. )rr*r!r!r"videorOzVideoMixin.videor>c Osdd}t|d|ji|}|}||sOt|}|j}||r&|jnd} t|} |D]} | |d| | }|j||| g|Ri|q.d St |dkr^t d|d|j||t | d g|Ri|d S) aSaves the video representation of the result to the specified path. Args: save_path (str): The path to save the video. If the save path does not end with .mp4 or .avi, it appends the input path's stem and suffix to the save path. *args: Additional positional arguments that will be passed to the video writer. **kwargs: Additional keyword arguments that will be passed to the video writer. cSr)Nzvideo/rrr!r!r"_is_video_filerz0VideoMixin.save_to_video.._is_video_filerz.mp4rr0zaThe result has multiple video files need to be saved. But the `save_path` has been specified as `rrN)rrrrr8r9rrTr=rr rrGr) r r>rrrZ video_writerrr?r9rrrr!r!r"rs"  $  .zVideoMixin.save_to_video)r)rCrDrErFrIrrr#rrharrayrrJrrr!r!r!r"rtrrc@sxeZdZdZdededdfddZedeee j ffdd Z e deee j ffd d Z d edededdfd dZdS) AudioMixinz3Mixin class for adding Audio handling capabilities.rrr%NcOs.||_|j|jt|d|ji||_dS)zInitializes AudioMixin. Args: *args: Additional positional arguments to pass to the AudioWriter. **kwargs: Additional keyword arguments to pass to the AudioWriter. rN)rrr save_to_audior _audio_writerrr!r!r"r#szAudioMixin.__init__cCr&)zAbstract method to convert the result to a audio. Returns: Dict[str, np.array]: The audio representation result. r(r*r!r!r" _to_audiorzAudioMixin._to_audiocCr-)zProperty to get the audio representation of the result. Returns: Dict[str, np.array]: The audio representation of the result. )rr*r!r!r"audiorOzAudioMixin.audior>c Osdd}|}||sFt|}|j}||r|jnd}t|} |D]} | |d| |}|jj||| g|Ri|q$d St|dkrUt d|d|jj||t | dg|Ri|d S) aSaves the audio representation of the result to the specified path. Args: save_path (str): The path to save the audio. If the save path does not end with .mp4 or .avi, it appends the input path's stem and suffix to the save path. *args: Additional positional arguments that will be passed to the audio writer. **kwargs: Additional keyword arguments that will be passed to the audio writer. cSr)Nzaudio/rrr!r!r"_is_audio_filerz0AudioMixin.save_to_audio.._is_audio_filez.wavrr0zaThe result has multiple audio files need to be saved. But the `save_path` has been specified as `rrN) rrr8r9rrrTr=rr rrGr) r r>rrrrr?r9rrrr!r!r"rs8      zAudioMixin.save_to_audio)rCrDrErFrrr#rrIrhrrrJrrr!r!r!r"rs rc@seZdZdZdgZdedefddZe dd e e e e e e e ffffd d Z ed e e e e e e e ffffd d Z d dddZdedede e efdee e e e e e e fffd df ddZdS) MarkdownMixinz6Mixin class for adding Markdown handling capabilities.Zmarkdown_textsrrcOs2t|i||_t|i||_|j|jdS)a&Initializes the Markdown writer and appends the save_to_markdown method to the save functions. Args: *args: Positional arguments to be passed to the MarkdownWriter constructor. **kwargs: Keyword arguments to be passed to the MarkdownWriter constructor. N)r_markdown_writerrrrrsave_to_markdownrr!r!r"r#szMarkdownMixin.__init__TFr%cCr&)a Convert the result to markdown format. Args: pretty (Optional[bool]): whether to pretty markdown by HTML, default by True. Returns: Dict[str, Union[str, Dict[str, Any]]]: A dictionary containing markdown text and image data. r()r prettyshow_formula_numberr!r!r" _to_markdowns zMarkdownMixin._to_markdowncCr-)zProperty to access the markdown data. Returns: Dict[str, Union[str, Dict[str, Any]]]: A dictionary containing markdown text and image data. )rr*r!r!r"markdownrOzMarkdownMixin.markdownNc Osdtfdd}||s-t|}||r|jnd}|j} t|} | | |}||_n||_|j|jj|j j|j|j ||dg|Ri|dS)aSave the markdown data to a file. Args: save_path (Union[str, Path]): The path where the markdown file will be saved. *args: Additional positional arguments for saving. **kwargs: Additional keyword arguments for saving. r%cSsFhd}tjt|\}}||vrdStt|\}}|dkS)a Check if a file is a markdown file based on its extension or MIME type. Args: file_path (Union[str, Path]): The path to the file. Returns: bool: True if the file is a markdown file, False otherwise. >z.mdown.mdz.mkdz .markdownTz text/markdown)ospathsplitextrIrr~r)rZmarkdown_extensionsrextrr!r!r"_is_markdown_file(s   z9MarkdownMixin.save_to_markdown.._is_markdown_filer)rrN) rrr8rr9r> _save_datarrTrr) r r>rrrrrr?rr9rr!r!r"rs&    zMarkdownMixin.save_to_markdown save_mkd_func save_img_funcr>datac Ost|}|dur dS|D]:\}}||jvr%|||g|Ri|t|trH|j} |D]\} } | rG|| | | g|Ri|q1qdS)aInternal method to save markdown and image data. Args: save_mkd_func (Callable): Function to save markdown text. save_img_func (Callable): Function to save image data. save_path (Union[str, Path]): The base path where the data will be saved. data (Optional[Dict[str, Union[str, Dict[str, Any]]]]): The markdown data to save. *args: Additional positional arguments for saving. **kwargs: Additional keyword arguments for saving. N)rrqMARKDOWN_SAVE_KEYSr=rgrHparent) r rrr>rrrrvaluerZimg_pathZimg_datar!r!r"rJs(   zMarkdownMixin._save_data)TFrB)rCrDrErFrrGrHr#rrrIr rrrJrrrrrrr!r!r!r"rs4  (  - r).rwrnr~rabcrpathlibrtypingrrrrrr numpyrhrrlZPILr utilsr Zutils.iorrrrrrrrrrrKrVr]rtrrrrrrrrr!r!r!r"s2     , /*!u=>@AE?B