a  dP @sUddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZeeejdZeeZWdn1s0YGdd d ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ iZ!e"D]*Z#e#e!e#j$<e#"D]Z%e%e!e%j$<qvq`ee!d <ee!d<ee!d<eeeeeeeeeeeeeeeeeeeeee eeeeeeeeed Z&d!e'd"<dS)#) annotationsN)Path)Any) media_datautils)FileDataz types.jsonc@sveZdZddZddddZdddd Zd dd d Zd dd dZdddddddZddddddddZ dS) SerializablecCs|S) The typing information for this component as a dictionary whose values are a list of 2 strings: [Python type, language-agnostic description]. Keys of the dictionary are: raw_input, raw_output, serialized_input, serialized_output )api_infoselfr b/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/gradio_client/serializing.pyserialized_infoszSerializable.serialized_infozdict[str, list[str]]returncCs tdS)r NNotImplementedErrorr r r rr szSerializable.api_infodict[str, Any]cCs tdS)z The example inputs for this component as a dictionary whose values are example inputs compatible with this component. Keys of the dictionary are: raw, serialized Nrr r r rexample_inputs!szSerializable.example_inputsztuple[str, str]cCs2|}|d|ddgd}|d|dfS)NZserialized_inputinfotyperr getr r typesr r rinput_api_info)szSerializable.input_api_infocCs2|}|d|ddgd}|d|dfS)NZserialized_outputrrrrrrrr r routput_api_info/szSerializable.output_api_infoFr str | Pathbool)xload_dir allow_linkscCs|S)z] Convert data from human-readable format to serialized format for a browser. r r r#r$r%r r r serialize4szSerializable.serializeNstr | Path | None str | None)r#save_dirroot_urlhf_tokencCs|S)z] Convert data from serialized format for a browser to human-readable format. r r r#r*r+r,r r r deserialize:s zSerializable.deserialize)r F)NNN) __name__ __module__ __qualname__rr rrrr'r.r r r rrs rc@s,eZdZdZddddZddddZd S) SimpleSerializablezIGeneral class that does not perform any serialization or deserialization.dict[str, bool | dict]rcCstdddS)Nr2Frrserializer_typesr r r rr JszSimpleSerializable.api_inforcCs dddS)NrawZ serializedr r r r rrPsz!SimpleSerializable.example_inputsNr/r0r1__doc__r rr r r rr2Gsr2c@s,eZdZdZddddZddddZd S) StringSerializablez?Expects a string as input/output but performs no serialization.r3rcCstdddS)Nr;Fr4r5r r r rr ZszStringSerializable.api_inforcCs dddS)NHowdy!r7r r r r rr`sz!StringSerializable.example_inputsNr9r r r rr;Wsr;c@s,eZdZdZddddZddddZd S) ListStringSerializablezHExpects a list of strings as input/output but performs no serialization.r3rcCstdddS)Nr=Fr4r5r r r rr jszListStringSerializable.api_inforcCsddgddgdS)Nr<ZMerhabar7r r r r rrpsz%ListStringSerializable.example_inputsNr9r r r rr=gsr=c@s,eZdZdZddddZddddZd S) BooleanSerializablez@Expects a boolean as input/output but performs no serialization.r3rcCstdddS)Nr>Fr4r5r r r rr zszBooleanSerializable.api_inforcCs dddS)NTr7r r r r rrsz"BooleanSerializable.example_inputsNr9r r r rr>wsr>c@s,eZdZdZddddZddddZd S) NumberSerializablezKExpects a number (int/float) as input/output but performs no serialization.r3rcCstdddS)Nr?Fr4r5r r r rr szNumberSerializable.api_inforcCs dddS)Nr7r r r r rrsz!NumberSerializable.example_inputsNr9r r r rr?sr?c@sbeZdZdZddZddddZddd d Zdd ddd dddZdd dd d d dddZdS)ImgSerializablezJExpects a base64 string as input/output which is serialized to a filepath.cCs dddS)Nstringzfilepath or URL to imager descriptionr r r r rrszImgSerializable.serialized_infor3rcCstdddS)NrATr4r5r r r rr szImgSerializable.api_inforcCs tjddS)NPhttps://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.pngr7rZ BASE64_IMAGEr r r rrszImgSerializable.example_inputsr Fr)r!r"r#r$r%rcCs.|sdSt|rt|Stt||S)z Convert from human-friendly version of a file (string filepath) to a serialized representation (base64). Parameters: x: String path to file to serialize load_dir: Path to directory containing x N)ris_http_url_likeZencode_url_to_base64Zencode_file_to_base64rr&r r rr's   zImgSerializable.serializeNr(r#r*r+r,rcCs(|dus|dkrdStj||d}|jS)a Convert from serialized representation of a file (base64) to a human-friendly version (string filepath). Optionally, save the file to the directory specified by save_dir Parameters: x: Base64 representation of image to deserialize into a string filepath save_dir: Path to directory to save the deserialized image to root_url: Ignored hf_token: Ignored Nr dir)rdecode_base64_to_filename)r r#r*r+r,filer r rr.szImgSerializable.deserialize)r F)NNN) r/r0r1r:rr rr'r.r r r rrAs rAcseZdZdZddfdd ZddZdd Zd d Zd d ZddZ ddddZ ddddZ ddddZ ddddZ d2ddddd d!d"Zd#d$Zd3dd&d&d&d&d'd(d)Zd4d*ddd+d d,d-Zd5d*d.d&d&d/d'd0d1ZZS)6FileSerializablezfExpects a dict with base64 representation of object as input/output which is serialized to a filepath.Nonercsd|_d|_tdSN)stream stream_namesuper__init__r  __class__r rrUszFileSerializable.__init__cCs|SrQ)_single_file_serialized_infor r r rrsz FileSerializable.serialized_infocCstdddS)NZSingleFileSerializableTr4r5r r r r_single_file_api_infosz&FileSerializable._single_file_api_infocCs dddS)NrBfilepath or URL to filerCr r r r rrXsz-FileSerializable._single_file_serialized_infocCsddddddS)Narrayz&List of filepath(s) or URL(s) to filesrBrZrC)rrDitemsr r r r r_multiple_file_serialized_infosz/FileSerializable._multiple_file_serialized_infocCstdddS)NZMultipleFileSerializableTr4r5r r r r_multiple_file_api_infosz(FileSerializable._multiple_file_api_infodict[str, dict | bool]cCs|SrQ)rYr r r rr szFileSerializable.api_inforcCs|SrQ)_single_file_example_inputsr r r rrszFileSerializable.example_inputscCsdtjdddSNFis_filedatazMhttps://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdfr7rZ BASE64_FILEr r r rr`s z,FileSerializable._single_file_example_inputscCsdtjdgdgdSrarer r r r_multiple_file_example_inputss z.FileSerializable._multiple_file_example_inputsr Fzstr | FileData | Noner!r"zFileData | NonerGcCsl|dust|tr|St|r*|}d}ntt||}t|j}||rRdnt|t|j ||dS)N)rMrd orig_namercsize) isinstancedictrrHstrrstatst_sizeZencode_url_or_file_to_base64rM)r r#r$r%filenamerhr r r_serialize_singles z"FileSerializable._serialize_singlecCs t||SrQ)rZdownload_byte_stream)r urlr,r r r _setup_streamszFileSerializable._setup_streamNr)rIc Cs|dur dSt|tr*tj||dj}nft|trx|dr|d}|dus`Jd||durtj|d|||d}ntj||d}n|drD|dr|r|sJ|j r|j |dkr|j |d|d|d |_ |d|_ t |j }t |pttd }|jd d d ||d d}||t|}n2|d} | dusfJd|tj| |dj}ntdt|d||S)NrJrcrMzThe 'name' field is missing in zfile=)r,rK is_streamzstream/)r,T)parentsexist_okrgoutputrdzThe 'data' field is missing in zLA FileSerializable component can only deserialize a string or a dict, not a z: )rirkrrLrMrjrZdownload_tmp_copy_of_fileZcreate_tmp_copy_of_filerRrSrqnextrtempfile gettempdirsecretsZ token_hexmkdir write_bytes ValueErrorr) r r#r*r+r, file_namefilepathchunkpathrdr r r_deserialize_singlesF           z$FileSerializable._deserialize_singlez3str | FileData | None | list[str | FileData | None]z'FileData | None | list[FileData | None]csF|dus|dkrdSt|tr4fdd|DS|SdS)aG Convert from human-friendly version of a file (string filepath) to a serialized representation (base64) Parameters: x: String path to file to serialize load_dir: Path to directory containing x allow_links: Will allow path returns instead of raw file content Nr csg|]}|qSr )ro.0fr%r$r r r ^z.FileSerializable.serialize..)rilistror&r rrr'Ms  zFileSerializable.serializePath | str | Nonezstr | None | list[str | None]csV|dur dSttrtt|tr@fdd|DSj|dSdS)aV Convert from serialized representation of a file (base64) to a human-friendly version (string filepath). Optionally, save the file to the directory specified by `save_dir` Parameters: x: Base64 representation of file to deserialize into a string filepath save_dir: Path to directory to save the deserialized file to root_url: If this component is loaded from an external Space, this is the URL of the Space. hf_token: If this component is loaded from an external private Space, this is the access token for the Space Ncsg|]}j|dqS)r*r+r,)rrr,r+r*r r rrwsz0FileSerializable.deserialize..r)rirrkrrr-r rrr.bs  zFileSerializable.deserialize)r F)NNN)r F)NNN)r/r0r1r:rUrrYrXr]r^r rr`rfrorqrr'r. __classcell__r r rVrrOs4 1rOcsjeZdZddZddddZdddd Zdd d dddfdd Zdddd d ddfdd ZZS)VideoSerializablecCs dddS)NrBzfilepath or URL to video filerCr r r r rrsz!VideoSerializable.serialized_infor_rcCstdddS)NrOTr4r5r r r rr szVideoSerializable.api_inforcCsdtjdddS)NFrbzNhttps://github.com/gradio-app/gradio/raw/main/test/test_files/video_sample.mp4r7)rZ BASE64_VIDEOr r r rrs z VideoSerializable.example_inputsr Fr)r!r"ztuple[FileData | None, None]rGcst|||dfSrQ)rTr'r&rVr rr'szVideoSerializable.serializeNz.tuple[FileData | None, FileData | None] | Nonerz*str | tuple[str | None, str | None] | NonerIcspt|ttfr:t|dks(Jd||d|dg}ntd|t||||}t|trl|dSdS)z Convert from serialized representation of a file (base64) to a human-friendly version (string filepath). Optionally, save the file to the directory specified by `save_dir` rz&Expected tuple of length 2. Received: rrN)rituplerlenr}rTr.)r r#r*r+r,Z x_as_listZdeserialized_filerVr rr.s  zVideoSerializable.deserialize)r F)NNN) r/r0r1rr rr'r.rr r rVrrsrc@s^eZdZddZddddZdddd Zdd d dddddZdddd d d dddZdS)JSONSerializablecCs dddS)NrBzfilepath to JSON filerCr r r r rrsz JSONSerializable.serialized_infor_rcCstdddS)NrTr4r5r r r rr szJSONSerializable.api_inforcCsdddddS)Nrr)abr7r r r r rrszJSONSerializable.example_inputsr Fr)r!r"zdict | list | NonerGcCs&|dus|dkrdStt||S)a Convert from a a human-friendly version (string path to json file) to a serialized representation (json string) Parameters: x: String path to json file to read to get json string load_dir: Path to directory containing x Nr )rZ file_to_jsonrr&r r rr's zJSONSerializable.serializeNzstr | dict | listr(rIcCs|dur dStj||djS)av Convert from serialized representation (json string) to a human-friendly version (string path to json file). Optionally, save the file to the directory specified by `save_dir` Parameters: x: Json string save_dir: Path to save the deserialized json file to root_url: Ignored hf_token: Ignored NrJ)rZdict_or_str_to_json_filerMr-r r rr.szJSONSerializable.deserialize)r F)NNNr/r0r1rr rr'r.r r r rrs rc@s^eZdZddZddddZdddd Zdd d dddddZdddd d ddddZdS)GallerySerializablecCs dddS)NrBz^path to directory with images and a file associating images with captions called captions.jsonrCr r r r rrsz#GallerySerializable.serialized_infor_rcCstdddS)NrTr4r5r r r rr szGallerySerializable.api_inforcCstjgddgddS)NrrEr7rFr r r rrs  z"GallerySerializable.example_inputsr Fr)r!r"zlist[list[str | None]] | NonerGc Cs|dus|dkrdSg}t|d}|d}t|}Wdn1sN0Y|D]&\}} tj||d} || | gq`|S)Nr captions.jsonr)r%)ropenjsonloadr\rOr'append) r r#r$r%files captions_file captions_jsoncaptionsr~captionimgr r rr's  (zGallerySerializable.serializeNrkz None | strrIc Cs|dur dSt|tt}|jdddi}|D]z}t|ttfrR|\}}nd}tj ||||d} ||| <|d} | d} t || Wdq61s0Yq6t j|S)NT)rurt)r+r,rw)rrkuuiduuid4r{rirrrOr.rrdumposrabspath) r r#r*r+r,Z gallery_pathrZimg_datarrMrrr r rr.s"  ,zGallerySerializable.deserialize)r F)r NNrr r r rrs rFileZ UploadButton)ZtextboxnumberZsliderZcheckboxZ checkboxgroupZradioZdropdownimagevideoaudiorNZ dataframeZ timeseriesstatebuttonZ uploadbuttonZ colorpickerlabelZhighlightedtextrhtmlZgalleryZchatbotZmodel3dZplotZbarplotZlineplotZ scatterplotZmarkdowncodeZannotatedimagezdict[str, type]COMPONENT_MAPPING)( __future__rrrrzrxrpathlibrtypingrZ gradio_clientrrZgradio_client.data_classesrr__file__parentrrr6rr2r;r=r>r?rArOrrrZSERIALIZER_MAPPING__subclasses__clsr/subclsr__annotations__r r r rs~   (585'3?