o #1 i @sddlZddlZddlmZddlZddlmZddlmZm Z m Z ddl m Z ddl mZdedejfd d Zd eejeefdejfd d Zdejde defddZe ejeee edfZdS)N)Union)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)r r responseimger`/home/app/PaddleOCR-VL-test/.venv_paddleocr/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. contextZPNG)formatrmax_image_b64_lenNz...Zadd_format_prefixz data:image/z;base64,) hasattrr*rrr+saver# b64encodegetvaluedecoderlower)r(r)r*streamZ im_formatZim_b64r,rrrserialize_image_to_byte_strEs  r4zCA normal PIL image that supports serialization to b64 bytes string.)r#rtypingrr ZPILrZpydanticrrrZtyping_extensionsrZmistral_commonrstrrr"r'r4ZSerializableImagerrrrs$