gN=UddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZmZmZddlmZddlmZmZdd lmZdd lmZdd lmZe rdd l mZddlm Zdd lm Z m!Z!m"Z"ddl#m$Z%edZ&de'd<ee&dejPe)e e*fZ+de'd<edZ,de'd<edZ-de'd<ee-dfZ.de'd<ee+ee+fZ/de'd<dZ0de'd<GddeZ1d e1jd_3d!e1jh_3d"e1jj_3d0d#Z6d0d$Z7 d1d%Z8 d2 d3d&Z9d4d'Z:d5d6d(Z;d7d)Zd:d,Z? d;d-Z@dd/ZBy)?) annotationsN)Sequence)IntEnum)Path) TYPE_CHECKINGFinalLiteralUnioncast) TypeAlias)runtimeurl_util)StreamlitAPIException)caching) NumpyShape)Any)GifImagePluginImage ImageFile) ImageList)zImageFile.ImageFilez Image.ImagezGifImagePlugin.GifImageFiler PILImagenpt.NDArray[Any] AtomicImage)RGBBGRChannels)JPEGPNGGIF ImageFormatautoImageFormatOrAutoImageOrImageListiz Final[int]MAXIMUM_CONTENT_WIDTHc$eZdZdZdZdZdZdZdZy) WidthBehaviorz| Special values that are recognized by the frontend and allow us to change the behavior of the displayed image. N) __name__ __module__ __qualname____doc__ORIGINALCOLUMNAUTOMIN_IMAGE_OR_CONTAINERMAX_IMAGE_OR_CONTAINERW/var/www/openai/venv/lib/python3.12/site-packages/streamlit/elements/lib/image_utils.pyr&r&=s$ H F Dr6r&z'Display the image at its original widthz5Display the image at the width of the column it's in.zDisplay the image at its original width, unless it would exceed the width of its column in which case clamp it to its column widthc|jdvS)N)RGBALAP)modeimages r7_image_may_have_alpha_channelr?Ss ::, ,,r6c |jdk(S)Nr)formatr=s r7 _image_is_gifrBWs <<5  r6c|j}|dvrtt|S|dk(ryt|tr,ddlm}|jtj|}n|}t|ryt|ryy)apReturn 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. >rrJPGrrrrr) upperr r isinstancebytesPILropenioBytesIOrBr?) image_datarAr pil_images r7_validate_image_format_stringrO[sv\\^F  K((*e$EJJrzz*56  Y$Y/ r6ctj}|dk(rt|r|jd}|j ||||j S)zConvert a PIL image to bytes.rrrAquality)rKrLr?convertsavegetvalue)r>rArRtmps r7 _PIL_to_bytesrW}sM **,C9%@ e$ JJs67J3 <<>r6cD|jd|jSNr)seekrU)datas r7_BytesIO_to_bytesr\sIIaL ==?r6cddl}ddlm}|j|j |j }t ||}t||S)NrrE)numpyrIr fromarrayastypeuint8rOrW)array output_formatnprimgrAs r7_np_array_to_bytesrfs> %//%,,rxx0 1C *3 >F f %%r6c|j}t|dvr tdt|dk(r#|ddvrtd|dt|fzt|dk(r|ddk(r |dddddf}|S) N)z'Numpy shape has to be of length 2 or 3.rir')riz2Channel can only be 1, 3, or 4 got %d. Shape is %srjr)shapelenrstr)rbrls r7_verify_np_shaperos E 5z#$MNN 5zQ59I5# @Ry#e*% &   5zQ59>aAg Lr6c(d|jS)z2Get the mimetype string for the given ImageFormat.zimage/)lower) image_formats r7_get_image_format_mimetyperss L&&() **r6crddlm}|jtj|}|j \}}|dkr|t kDrt }|dkDrF||kDrAtd|z|z|z }|j||f|j}t||dS|j|k7rt||dS|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. rrE?)resampleZrQ) rIrrJrKrLsizer$intresizeBILINEARrWrA)rMwidthrrrrN actual_width actual_height new_heights r7_ensure_image_size_and_formatrs$%**RZZ %;>$R Y|RHH<'Y|RHH r6cddl}|}t|jj|jrP|r|j |dd}n3|j |dks|j|dkDr td|dz}|S|r|j |dd}|S|j |dks|j|dkDr td|S)Nrrugz0Data is outside [0.0, 1.0] and clamp is not set.z.Data is outside [0, 255] and clamp is not set.) r^ issubclassdtypetypefloatingclipaminamax RuntimeError)r>clamprdr[s r7 _clip_imagers D%++""BKK0 775!S)Dwwu~#rwwu~';"#UVVcz K 775!S)D Kwwu~!RWWU^c%9"#STT Kr6cddl}ddlm}m}t |t r t |}t |t rtjj|stj|dr|S|jdrCtjj|r$t|5} | j}dddtj d|rOd|vr|j#d d d }ddl} | j'|j)d j+d } d | S t|d5} | j} dddnt ||j|jfrt?||}tA||} nt |tBjDr tG|} nt ||jHrhtKtM||}|dk(r@tOtQtR|jTdk(r|ddddgdf}n tWdtY||} n|} t? |}t[| ||} t]|}t3j^rHt3j4j6j9| ||}t;j<| |||Sy#1swYxYw#1swYzxYw#t,$rgddl}|j1|\}}|d}t3j4j6j9|||}t;j<||||cYSwxYw)a/Return 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.) rN)rr)httphttpsr[)allowed_schemasz.svgz2(^\s?(<\?xml[\s\S]*\s)xmlnszr|rchannelsrcimage_idrdrrtextfilerimage_b64_encodedfrMrmimetype_urlrArrs r7 image_to_urlrs$ %E %ww~~e$ #<* L >>& !bggnnU&;e  99JE Re# F  & 0 0g1F G N Nw W /0A/BC C eT"aVVX #" EI//= >.umD"5&1 E2:: &&u-  E2:: &,U3U; u 4 EKK01Q6aIo.+4 (e=Q  1]KL.z5,OJ),7H~~""$3377 HhW Hh? ]*#"  #..u5KHa5&&(77;;E8XVC  # #E8X >J s7,J3+ K 7KK 3J=K K A-L=<L=c ttd|jdDcgc]}||ddddddfc}Scc}wrY)rangerl)rbis r7_4d_to_list_3drZs8',Q A'? @'?!E!Q1* '? @@ @s5c ddl}tt|j}t |t t tfr t |} nKt ||jr2ttt|jdk(r t|} n|g} t |t r|} nt |tr|g} nkt ||jr7ttt|jdk(r|j} n|dgt| z} n t|g} t | t sJdt| t| k(sJdt| t| fzt||_t#t%| | D]P\} \}}|j&j)} |t|| _d|| fz} t-|||||| | _Ry)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. rNrkrjz1If image is a list then caption should be as wellz'Cannot pair %d captions with %d images.z%s-%i)r^r rrFrGlistsettuplerrmrrlrrntolistryr| enumeratezipimgsrcaptionrr) coordinatesr>rr| proto_imgsrrrcrdimagescaptions coord_suffix proto_imgrs r7marshall_imagesr^s\Hhnn./H%$U+,e E2:: &3tJ /L+MQR+R&'4 )0 GS !9 GRZZ (Sj'--1P-QUV-V>># 6CK'L> h %; % x=CK ')R H  F V* ' 5zJ*3C4I*J& &ugOO'')   #G I k<88$ 5%=( +Kr6)r>rreturnbool)rMzbytes | PILImagerArnrr )rd)r>rrAr rRryrrH)r[z io.BytesIOrrH)r)rbrrcrnrrH)rbrrr)rrr rrn)rMrHr|ryrrr rrH)r>rrrrr)r>rr|ryrrrrrcr"rrnrrn)rbrrzlist[npt.NDArray[Any]])rr!)rrnr>r#rz)str | npt.NDArray[Any] | list[str] | Noner|zint | WidthBehaviorrImageListProtorrrrrcr"rNone)C __future__rrKrrcollections.abcrenumrpathlibrtypingrrr r r typing_extensionsr streamlitr rstreamlit.errorsrstreamlit.runtimerstreamlit.type_utilrr numpy.typingnptrIrrrstreamlit.proto.Image_pb2rrr__annotations__rLrnrHrrr r"r#r$r&r0r/r1r2r?rBrOrWr\rfrorsrrrrrr5r6r7rsu# $==''2%*44EG)  "**c4> Yl+)+ !56 Y6&{F':;9;#K+1F$FG)G %,z+  G  "O ?  -! *-H!   " &"+ !1< B(o o o o o % o  o odA'-[ [ [ 7[   [  [  [ [ %[  [ r6