U ARdM @sUdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlZddlmZmZmZddlmZmZmZddlmZdd lmZdd lmZdd l m!Z"dd l#m$Z$dd l%m&Z&erddlm'Z'ddl(mZ)ddl*m+Z+ee,Z-ee.d<dZ/ee0e.d<e ejejej1fZ2ee.d<e e2dej3e4fZ5ee.d<e e5e e5fZ6ee.d<e e ede7fZ8ee.d<edZ9ee.d<edZ:ee.d<ee:dfZ;ee.d<Gdd d eZ_d#ed?ZJeCe0e:eCd@dAdBZKde7ddCdDdEZLe5e0e7e9e;e4e4dFdGdHZMdOe4e6e e e4de e4fe e0e` OR a list of one of the above, to display multiple images. caption : str or list of str Image caption. If displaying multiple images, caption should be a list of captions (one for each image). width : int or None Image width. None means use the image width, but do not exceed the width of the column. Should be set for SVG images, as they have no default image width. use_column_width : 'auto' or 'always' or 'never' or bool If 'auto', set the image's width to its natural size, but do not exceed the width of the column. If 'always' or True, set the image's width to the column width. If 'never' or False, set the image's width to its natural size. Note: if set, `use_column_width` takes precedence over the `width` parameter. clamp : bool Clamp image pixel values to a valid range ([0-255] per channel). This is only meaningful for byte array images; the parameter is ignored for image URLs. If this is not set, and an image has an out-of-range value, an error will be thrown. channels : 'RGB' or 'BGR' If image is an nd.array, this parameter denotes the format used to represent color information. Defaults to 'RGB', meaning `image[:, :, 0]` is the red channel, `image[:, :, 1]` is green, and `image[:, :, 2]` is blue. For images coming from libraries like OpenCV you should set this to 'BGR', instead. output_format : 'JPEG', 'PNG', or 'auto' This parameter specifies the format to use when transferring the image data. Photos should use the JPEG format for lossy compression while diagrams should use the PNG format for lossless compression. Defaults to 'auto' which identifies the compression type based on the type and format of the image argument. Example ------- >>> import streamlit as st >>> from PIL import Image >>> >>> image = Image.open('sunrise.jpg') >>> >>> st.image(image, caption='Sunrise by the mountains') .. output:: https://doc-image.streamlitapp.com/ height: 710px rNrTrzImage width must be positive.imgs) r*r4r3r2rImageListProtomarshall_imagesdgZ_get_delta_path_strZ_enqueue) selfr8r9r:r;r<r=r>Zimage_list_protor5r5r6r8Xs(G zImageMixin.image)r?cCs td|S)zGet our DeltaGenerator.r)r)rDr5r5r6rCsz ImageMixin.dg)NNNFr"r)r.r/r0rrrrstrrintr!boolr$r)r8propertyrCr5r5r5r6r7Ws&\r7)r8r?cCs|jdkrdSdSdS)N)ZRGBAZLAPTF)moder8r5r5r6_image_may_have_alpha_channels rLcCst|jdkS)Nr')rGformatrKr5r5r6 _image_is_gifsrN) image_datarMr?cCsj|}|dks|dkr"tt|S|dkr.dSt|trJtt|}n|}t |rZdSt |rfdSdS)aqReturn either "JPEG", "PNG", or "GIF", based on the input `format` string. - If `format` is "JPEG" or "JPG" (or any capitalization thereof), return "JPEG" - If `format` is "PNG" (or any capitalization thereof), return "PNG" - For all other strings, return "PNG" if the image has an alpha channel, "GIF" if the image is a GIF, and "JPEG" otherwise. r%r&ZJPGr') upperrr( isinstancebytesr openioBytesIOrNrL)rOrMZ pil_imager5r5r6_validate_image_format_strings   rVr%d)r8rMqualityr?cCs:t}|dkr"t|r"|d}|j|||d|S)zConvert a PIL image to bytes.r%r"rMrX)rTrUrLconvertsavegetvalue)r8rMrXtmpr5r5r6 _PIL_to_bytess  r^)datar?cCs|d|S)Nr)seekr\)r_r5r5r6_BytesIO_to_bytess ra)arrayr?cCs&t|tj}t||}t||SN)r Z fromarrayZastypenpZuint8rVr^)rbr>imgrMr5r5r6_np_array_to_bytess rfcsfddtdjdDS)Ncs(g|] }|ddddddfqSrcr5).0irbr5r6 sz"_4d_to_list_3d..r)rangeshaperir5rir6_4d_to_list_3dsrmcCst|jdkrtdt|jdkrN|jddkrNtd|jdt|jft|jdkr|jddkr|dddddf}|S) N)z'Numpy shape has to be of length 2 or 3.ror+)roz2Channel can only be 1, 3, or 4 got %d. Shape is %srpr)lenrlrrErir5r5r6_verify_np_shapesrs) image_formatr?cCsd|S)z2Get the mimetype string for the given ImageFormat.zimage/)lower)rtr5r5r6_get_image_format_mimetypesrv)rOr:rtr?cCstt|}|j\}}|dkr.|tkr.t}|dkrt||krttd|||}|j||ftjd}t ||ddSt d|}|| krt ||ddS|S)zResize an image if it exceeds the given width, or if exceeds MAXIMUM_CONTENT_WIDTH. Ensure the image's format corresponds to the given ImageFormat. Return the (possibly resized and reformatted) image bytes. r?)ZresampleZrYN) r rSrTrUsizerrFresizeZBILINEARr^imghdrwhatru)rOr:rtr8Z actual_widthZ actual_heightZ new_heightextr5r5r6_ensure_image_size_and_formats   r~)r8r<r?cCs|}t|jjtjrV|r(t|dd}n$t|dksDt|dkrLtd|d}n8|rjt|dd}n$t|dkst|dkrtd|S)Nrrwgz0Data is outside [0.0, 1.0] and clamp is not set.z.Data is outside [0, 255] and clamp is not set.) issubclassZdtypetyperdZfloatingZclipZaminZamax RuntimeError)r8r<r_r5r5r6 _clip_image8s r)r8r:r<r=r>image_idr?c Cst|trzt|}|jr |WSWntk r6YnXz"t|d}|}W5QRXWnTtk rt |\} } | dkrd} t j || |} t|| || YSXnt|tjtjfrt||} t|| }nt|tjrt|}ntt|tjr`tt||}|dkrLt|jdkrD|dddddddgf}ntd ttd ||d }n|}t||} t ||| }t!| } t "rt j || |} t|| || Sd SdS) a0Return a URL that an image can be served from. If `image` is already a URL, return it unmodified. Otherwise, add the image to the MediaFileManager and return the URL. (When running in "raw" mode, we won't actually load data into the MediaFileManager, and we'll return an empty URL.) rbNzapplication/octet-streamr#rornrprzQWhen using `channels="BGR"`, the input image should have exactly 3 color channelsr)rbr>)#rQrEr schemeUnicodeDecodeErrorrSread Exception mimetypes guess_typerZ get_instanceZmedia_file_mgraddrZsave_media_datar r rVr^rTrUrardndarrayrrsrrrlrrfrr~rvexists)r8r:r<r=r>rpfrOmimetype_urlrMrtr5r5r6 image_to_urlJs\             rr") coordinatesr8r9r: proto_imgsr<r=r>r?c Cstt|}t|tr|}n*t|tjrBt|jdkrBt |}n|g}t |tkrZ|} nXt|t rl|g} nFt|tjrt|jdkr| } n"|dkrdgt|} n t |g} t | tkst dt| t|kst dt| t|ft||_tt|| D]\} \}}|j} |dk r0t || _d|| f} d} t|t r|dr|d st|}|}W5QRXtd |rd |ksd |krd || _n d || _d} | st|||||| | _qdS)aFill an ImageListProto with a list of images and their captions. The images will be resized and reformatted as necessary. Parameters ---------- coordinates A string indentifying the images' location in the frontend. image The image or images to include in the ImageListProto. caption Image caption. If displaying multiple images, caption should be a list of captions (one for each image). width The desired width of the image or images. This parameter will be passed to the frontend. Positive values set the image width explicitly. Negative values has some special. For details, see: `WidthBehaviour` proto_imgs The ImageListProto to fill in. clamp Clamp image pixel values to a valid range ([0-255] per channel). This is only meaningful for byte array images; the parameter is ignored for image URLs. If this is not set, and an image has an out-of-range value, an error will be thrown. channels If image is an nd.array, this parameter denotes the format used to represent color information. Defaults to 'RGB', meaning `image[:, :, 0]` is the red channel, `image[:, :, 1]` is green, and `image[:, :, 2]` is blue. For images coming from libraries like OpenCV you should set this to 'BGR', instead. output_format This parameter specifies the format to use when transferring the image data. Photos should use the JPEG format for lossy compression while diagrams should use the PNG format for lossless compression. Defaults to 'auto' which identifies the compression type based on the type and format of the image argument. rqrpNz1If image is a list then caption should be as wellz'Cannot pair %d captions with %d images.z%s-%iFz.svg)zhttp://zhttps://z2(^\s?(<\?xml[\s\S]*\s)xlinkxmlnszdata:image/svg+xml,T)rr$rPrQlistrdrrrrlrmrrEtolistAssertionErrorrFr: enumeratezipr@rr9endswith startswithrSrresearchZmarkuprr)rr8r9r:rr<r=r>ZimagesZcaptionsZ coord_suffixZ proto_imgrZis_svgZtextfiler5r5r6rBsZ0              rB)r%rW)r%)r"r)Or1r{rTrrenumrtypingrrrrrr urllib.parser ZnumpyrdZPILr r r Ztyping_extensionsr rrZ streamlitrZstreamlit.errorsrZstreamlit.loggerrZstreamlit.proto.Image_pb2rrAZstreamlit.runtimerZstreamlit.runtime.metrics_utilrrZ numpy.typingZnptZstreamlit.delta_generatorrr.r__annotations__rrFZ GifImageFilerrUrErrrGr!r$r(r)r*r2r3r4r7rLrNrRrVr^rarfrmrsrvr~rrrBr5r5r5r6s              d  "   j