B eJd @s`ddlmZddlZddlmZmZyddlmZdZWne k rPdZYnXd d d Z dS))BytesION)Writer from_array)ImageTFpilpngc Cs|dks|dkrtdd}|jdkr,d}nH|jdkrJ|jddkrJd }n*|jdkrl|jdd krld }d }ntd |dkrtrdnd}|dkr|dkrtd|dkrD|j}|j}|dkr||d|d|df}t|d|d|dk||d}t||d} d} t,} || | j | t |  d} WdQRXnxtsRtdt|} |dksp|dkrzd} d}nd} d}t.} | j| ||d| t |  d} WdQRX| S)aConverts a numpy array of uint8 into a base64 png or jpg string. Parameters ---------- img: ndarray of uint8 array image backend: str 'auto', 'pil' or 'pypng'. If 'auto', Pillow is used if installed, otherwise pypng. compression: int, between 0 and 9 compression level to be passed to the backend ext: str, 'png' or 'jpg' compression format used to generate b64 string r z*compression level must be between 0 and 9.FLRGBrZRGBATzInvalid image shapeautorZpypngr z6jpg binary strings are only available with PIL backendr)Z greyscalealpha compression)modezdata:image/png;base64,zutf-8Nzapillow needs to be installed to use `backend='pil'. Pleaseinstall pillow or use `backend='pypng'.ZjpgZjpegzdata:image/jpeg;base64,)formatZcompress_level) ValueErrorndimshape pil_importedZreshaperrrwriterowsbase64 b64encodegetvaluedecode ImportErrorrZ fromarraysave)ZimgbackendrextrrrshwZimg_pngprefixstreamZ base64_stringZpil_imgr'J/var/www/html/venv/lib/python3.7/site-packages/_plotly_utils/data_utils.pyimage_array_to_data_uri sP    $ "r))rrr ) iorrr rrZPILrrrr)r'r'r'r(s