o i @sddlZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z de dejfdd Zd eje BeBdejfd d Zd ejdede fddZe ejeeeedfZdS)N)Image)BeforeValidatorPlainSerializerSerializationInfo) Annotated) __version__urlreturnc Csddti}ztj||d}|tt|j}|WStj j y7}z t d|d|dd}~wt yI}zt d|d}~ww) 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)rr responseimgerH/home/ubuntu/.local/lib/python3.10/site-packages/mistral_common/image.pydownload_image s rxc Cst|tjr|St|tr"z tt|WSty!tdwzttt | d}|WStyL}z td|ddd|d}~ww)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)rimagerrrr"maybe_load_image_from_str_or_bytes%s*    r&iminfocCst|dr |jp i}ni}t}|jpd}|j||dt| d}|r8| d}r8|d|dS|rI| drId | d |}|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)formatrmax_image_b64_lenNz...add_format_prefixz data:image/z;base64,) hasattrr)rrr+saver" b64encodegetvaluedecoder lower)r'r(r)stream im_formatim_b64r,rrrserialize_image_to_byte_strDs  r7zCA normal PIL image that supports serialization to b64 bytes string.)r"rr PILrpydanticrrrtyping_extensionsrmistral_commonrstrrr!r&r7SerializableImagerrrrs"