o ZhR@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl Z ddlZddlmZGdddeZGd d d Z ddddZdS)) annotationsN)Enum)AnyDict)urlparse)Partc@seZdZdZdZdZdZdS)RoutezImage Loading RouteN)__name__ __module__ __qualname____doc__BASE64 LOCAL_FILEURLrrZ/var/www/html/lang_env/lib/python3.10/site-packages/langchain_google_genai/_image_utils.pyrs rc@sLeZdZdZdddZdd d Zdd d ZdddZdddZdddZ dS)ImageBytesLoaderzLoads image bytes from multiple sources given a string. Currently supported: - B64 Encoded image string - URL image_stringstrreturnbytescCsT||}|tjkr||S|tjkr||S|tjkr"tdtd|d)zRoutes to the correct loader based on the image_string. Args: image_string: Can be either: - B64 Encoded image string - URL Returns: Image bytes. zLoading from local files is no longer supported for security reasons. Please pass in images as Google Cloud Storage URI, b64 encoded image string (data:image/...), or valid image url.zImage string must be one of: Google Cloud Storage URI, b64 encoded image string (data:image/...), or valid image url.Instead got ''.) _routerr_bytes_from_b64r_bytes_from_urlr ValueErrorZ_bytes_from_file)selfrrouterrr load_bytes s     zImageBytesLoader.load_bytesrc Cs||}|tjkr||}|tjkr||}|tjkr$d}t|d|i}t |\}}|s;t |}|r;|j }|rA||d<t |dS)zGets Part for loading from Gemini. Args: image_string: Can be either: - B64 Encoded image string - URL zLoading from local files is no longer supported for security reasons. Please specify images as Google Cloud Storage URI, b64 encoded image string (data:image/...), or valid image url.data mime_type) inline_data)rrrrrrrr mimetypes guess_typefiletypeguessmimer) rrr bytes_msgr$r#_kindrrr load_partBs$        zImageBytesLoader.load_partrcCsB|drtjS||rtjStj|rtjSt d|d)N data:image/zhImage string must be one of: b64 encoded image string (data:image/...) or valid image url. Instead got 'r) startswithrr_is_urlrospathexistsrr)rrrrrrgs   zImageBytesLoader._route base64_imagecCs:d}t||}|dur|d}t|Std|)zGets image bytes from a base64 encoded string. Args: base64_image: Encoded image in b64 format. Returns: Image bytes zdata:image/\w{2,4};base64,(.*)Nr z1Error in b64 encoded image. Must follow pattern: )researchgroupbase64 b64decoder)rr5patternmatchencoded_stringrrrrws    z ImageBytesLoader._bytes_from_b64urlcCst|}|js ||jS)zGets image bytes from a public url. Args: url: Valid url. Raises: HTTP Error if there is one. Returns: Image bytes )requestsgetokraise_for_statuscontent)rr>responserrrrs z ImageBytesLoader._bytes_from_url url_stringboolcCs0z t|}t|j|jgWStyYdSw)zChecks if a url is valid. Args: url_string: Url to check. Returns: Whether the url is valid. F)rallschemenetloc Exception)rrEresultrrrr1s   zImageBytesLoader._is_urlN)rrrr)rrrr)rrrr)r5rrr)r>rrr)rErrrF) r r rrr!r.rrrr1rrrrrs  " %  rasciipng image_bytesrencodingr image_formatrcCs t||}d|d|S)aEncodes image bytes into a b64 encoded string. Args: image_bytes: Bytes of the image. encoding: Type of encoding in the string. 'ascii' by default. image_format: Format of the image. 'png' by default. Returns: B64 image encoded string. r/z;base64,)r9 b64encodedecode)rNrOrPZ encoded_bytesrrrimage_bytes_to_b64_strings rS)rLrM)rNrrOrrPrrr) __future__rr9r%r2r6enumrtypingrr urllib.parserr'r?Z)google.ai.generativelanguage_v1beta.typesrrrrSrrrrs