*`i ddlZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z de dejfdZd eje zezdejfd Zd ejd ede fd Ze ejeeeedfZdS)N)Image)BeforeValidatorPlainSerializerSerializationInfo) Annotated) __version__urlreturnclddti} tj||}|t jt j|j}|S#tj j $r}td|d|dd}~wt$r}td|d}~wwxYw) zDownload an image from a URL and return it as a PIL Image. Args: url: The URL of the image to download. Returns: The downloaded image as a PIL Image object. z User-Agentzmistral-common/)headersz!Error downloading the image from z: .NzError converting to PIL image: ) rrequestsgetraise_for_statusropenioBytesIOcontent exceptionsRequestException RuntimeError Exception)r r responseimges h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/image.pydownload_imager s<{<<=G B<W555!!###jH$45566   /LLLJsJJaJJJKKK BBB@Q@@AAABs$AA%%B39B B3B..B3xc t|tjr|St|trD tjt j|S#t $rtdwxYw tjt jtj | d}|S#t $r!}td|ddd|d}~wwxYw)aULoad an image from a string or bytes. If the input is already a PIL Image, return it as is. Args: x: The input to load the image from. Can be a PIL Image, a string, or bytes. If it's a string, it's assumed to be a base64 encoded string of bytes. Returns: The loaded image as a PIL Image object. z3Encountered an error when loading image from bytes.asciizBEncountered an error when loading image from bytes starting with 'NzI'. Expected either a PIL.Image.Image or a base64 encoded string of bytes.) isinstancerbytesrrrrrbase64 b64decodeencode)rimagers r"maybe_load_image_from_str_or_bytesr(%s !U[!!!UV V:bjmm,, , V V VTUU U V 2:f&6qxx7H7H&I&IJJKK  (ssV ( ( (    s$%AA37A C C/C**C/iminfoct|dr |jpi}ni}tj}|jpd}|||t j| d}|r$| dx}r |d|dzS|r0| drd | d |z}|S) zSerialize an image to a base64 encoded string of bytes. Args: im: The image to serialize. info: The serialization info. Returns: The serialized image as a base64 encoded string of bytes. contextPNG)formatr max_image_b64_lenNz...add_format_prefixz data:image/z;base64,) hasattrr,rrr.saver$ b64encodegetvaluedecoderlower)r)r*r,stream im_formatim_b64r/s rserialize_image_to_byte_strr:DstY,$" Z\\F "UIGGF9G%%%  foo// 0 0 7 7 @ @F25H)I)II%2((()E11D7;;233D:y00:::VC MzCA normal PIL image that supports serialization to b64 bytes string.)r$rrPILrpydanticrrrtyping_extensionsrmistral_commonrstrrr#r(r:SerializableImager;rrCs7 HHHHHHHHHH''''''&&&&&&BB BBBB2%+*;e*C >EK7HS4 KO677O/00IKr;