o ٷi5R@s>UdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZee ejd ZeeZWdn1sPwYGd d d ZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!iZ"e#D]Z$e$e"e$j%<e$#D]Z&e&e"e&j%<qqee"d <ee"d<ee"d <id!ed"ed#ed$ed%ed&ed'ed(ed)ed*ed+ed,e d-e d.e d/ed0ed1eee e e ee!e ee e e e eee d2Z'd3e(d4<dS)5z:Included for backwards compatibility with 3.x spaces/apps.) annotationsN)Path)Any) media_datautils)FileDataz types.jsonc@sZeZdZddZd!ddZd"dd Zd#d d Zd#d dZd$d%ddZ   d&d'dd Z dS)( SerializablecC|Sz 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_infoselfrM/home/ubuntu/.local/lib/python3.10/site-packages/gradio_client/serializing.pyserialized_infoszSerializable.serialized_inforeturndict[str, list[str]]cCtr NotImplementedErrorr rrrr zSerializable.api_infodict[str, Any]cCr)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 rr rrrexample_inputs#rzSerializable.example_inputstuple[str, str]cC2|}|d|ddgd}|d|dfS)Nserialized_inputinfotyperr getr r typesrrrinput_api_info+zSerializable.input_api_infocCr)Nserialized_outputrrrrrr r"rrroutput_api_info1r%zSerializable.output_api_infoFxrload_dir str | Path allow_linksboolcC|S)z] Convert data from human-readable format to serialized format for a browser. rr r)r*r,rrr serialize6szSerializable.serializeNsave_dirstr | Path | Noneroot_url str | Nonehf_tokencCr.)z] Convert data from serialized format for a browser to human-readable format. rr r)r1r3r5rrr deserialize<s zSerializable.deserialize)rrrr)rrr(F)r)rr*r+r,r-NNN)r)rr1r2r3r4r5r4) __name__ __module__ __qualname__rr rr$r'r0r7rrrrrs      rc@$eZdZdZd ddZd ddZd S) SimpleSerializablezIGeneral class that does not perform any serialization or deserialization.rdict[str, bool | dict]cCtdddS)Nr?Frrserializer_typesr rrrr LzSimpleSerializable.api_inforcCs dddS)Nraw serializedrr rrrrRz!SimpleSerializable.example_inputsNrr@r8r;r<r=__doc__r rrrrrr?I r?c@r>) StringSerializablez?Expects a string as input/output but performs no serialization.rr@cCrA)NrNFrBrCr rrrr \rEzStringSerializable.api_inforcC dddS)NHowdy!rFrr rrrrbrIz!StringSerializable.example_inputsNrJr8rKrrrrrNYrMrNc@r>) ListStringSerializablezHExpects a list of strings as input/output but performs no serialization.rr@cCrA)NrQFrBrCr rrrr lrEzListStringSerializable.api_inforcCsddgddgdS)NrPMerhabarFrr rrrrrsz%ListStringSerializable.example_inputsNrJr8rKrrrrrQirMrQc@r>) BooleanSerializablez@Expects a boolean as input/output but performs no serialization.rr@cCrA)NrSFrBrCr rrrr |rEzBooleanSerializable.api_inforcCrO)NTrFrr rrrrrIz"BooleanSerializable.example_inputsNrJr8rKrrrrrSyrMrSc@r>) NumberSerializablezKExpects a number (int/float) as input/output but performs no serialization.rr@cCrA)NrTFrBrCr rrrr rEzNumberSerializable.api_inforcCrO)NrFrr rrrrrIz!NumberSerializable.example_inputsNrJr8rKrrrrrTrMrTc@sNeZdZdZddZdddZdd d Z ddddZ   d d!ddZdS)"ImgSerializablezJExpects a base64 string as input/output which is serialized to a filepath.cC dddS)Nstringz+filepath on your computer (or URL) of imager descriptionrr rrrrrIzImgSerializable.serialized_inforr@cCrA)NrVTrBrCr rrrr zImgSerializable.api_inforcCs tjddS)NPhttps://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.pngrFr BASE64_IMAGEr rrrrszImgSerializable.example_inputsr(Fr)r4r*r+r,r-cCs.|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_likeencode_url_to_base64encode_file_to_base64rr/rrrr0s    zImgSerializable.serializeNr1r2r3r5cCs(|dus|dkr dStj||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)r1r3r5filerrrr7szImgSerializable.deserializerJr8r9)r)r4r*r+r,r-rr4r:) r)r4r1r2r3r4r5r4rr4) r;r<r=rLrr rr0r7rrrrrVs    rVcseZdZdZd6fdd ZddZdd Zd d Zd d ZddZ d7ddZ d8ddZ d8ddZ d8ddZ  d9d:d#d$Zd%d&Z ' ' 'd;dd4d5ZZS)?FileSerializablezfExpects a dict with base64 representation of object as input/output which is serialized to a filepath.rNonecsd|_d|_tdSN)stream stream_namesuper__init__r  __class__rrrmszFileSerializable.__init__cCr ri)_single_file_serialized_infor rrrrz FileSerializable.serialized_infocCrA)NSingleFileSerializableTrBrCr rrr_single_file_api_inforEz&FileSerializable._single_file_api_infocCrW)NrX*filepath on your computer (or URL) of filerYrr rrrrprIz-FileSerializable._single_file_serialized_infocCsddddddS)Narrayz&List of filepath(s) or URL(s) to filesrXrtrY)rrZitemsrr rrr_multiple_file_serialized_infos z/FileSerializable._multiple_file_serialized_infocCrA)NMultipleFileSerializableTrBrCr rrr_multiple_file_api_inforEz(FileSerializable._multiple_file_api_infodict[str, dict | bool]cCr ri)rsr rrrr rqzFileSerializable.api_inforcCr ri)_single_file_example_inputsr rrrrrqzFileSerializable.example_inputscCdtjdddSNFis_filedatazMhttps://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdfrFr BASE64_FILEr rrrr{ z,FileSerializable._single_file_example_inputscCsdtjdgdgdSr}rr rrr_multiple_file_example_inputss z.FileSerializable._multiple_file_example_inputsr(Fr)str | FileData | Noner*r+r,r-FileData | NonecCsn|dus t|tr |St|r|}d}ntt||}t|j}|p'd|r+dnt|t|j |dS)N)rer orig_namesize) isinstancedictrr_strrstatst_sizeencode_url_or_file_to_base64re)r r)r*r,filenamerrrr_serialize_singles z"FileSerializable._serialize_singlecCs t||Sri)rdownload_byte_stream)r urlr5rrr _setup_stream&s zFileSerializable._setup_streamNr1r4r3r5c Cs|durdSt|trtj||dj}|St|tr|drJ|d}|dur/td||durAtj|d|||d}|Stj ||d}|S|dr|drW|rW|s[td|j re|j |dkrx|j |d |d|d |_ |d|_ t |j }t|pttd }|jd d d ||dd}||t|}|S|d} | durtd|tj| |dj}|Stdt|d|)NrbrrezThe 'name' field is missing in zfile=)r5rc is_streamz2name and root_url and save_dir must all be presentzstream/)r5T)parentsexist_okroutputrzThe 'data' field is missing in zLA FileSerializable component can only deserialize a string or a dict, not a z: )rrrrdrerr! ValueErrordownload_tmp_copy_of_filecreate_tmp_copy_of_filerjrkrnextrtempfile gettempdirsecrets token_hexmkdir write_bytesr) r r)r1r3r5 file_namefilepathchunkpathrrrr_deserialize_single)sX '         z$FileSerializable._deserialize_single3str | FileData | None | list[str | FileData | None]'FileData | None | list[FileData | None]csB|dus|dkr dSt|trfdd|DS|S)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)r.0fr,r*r rr msz.FileSerializable.serialize..)rlistrr/rrrr0\s  zFileSerializable.serializePath | str | Nonestr | None | list[str | None]csR|durdSttrtt|tr fdd|DSj|dS)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)r1r3r5)rrr5r3r1r rrrs z0FileSerializable.deserialize..r)rrrrrr6rrrr7qs  zFileSerializable.deserialize)rrhrrzr8r9)r)rr*r+r,r-rrr:) r)rr1r4r3r4r5r4rr4)r)rr*r+r,r-rr) r)rr1rr3r4r5r4rr)r;r<r=rLrmrrsrprwryr rr{rrrrr0r7 __classcell__rrrnrrgs6       6 rgcsTeZdZddZdddZddd Z d d!fdd Z   d"d#fdd ZZS)$VideoSerializablecCrW)NrXz0filepath on your computer (or URL) of video filerYrr rrrrrIz!VideoSerializable.serialized_inforrzcCrA)NrgTrBrCr rrrr r[zVideoSerializable.api_inforcCr|)NFr~zNhttps://github.com/gradio-app/gradio/raw/main/test/test_files/video_sample.mp4rF)r BASE64_VIDEOr rrrrrz VideoSerializable.example_inputsr(Fr)r4r*r+r,r-tuple[FileData | None, None]cst|||dfSri)rlr0r/rnrrr0szVideoSerializable.serializeN.tuple[FileData | None, FileData | None] | Noner1rr3r5*str | tuple[str | None, str | None] | Nonecspt|ttfrt|dkrtd||d|dg}ntd|t||||}t|tr6|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)rtuplerlenrrlr7)r r)r1r3r5 x_as_listdeserialized_filernrrr7s  zVideoSerializable.deserializerr8r9)r)r4r*r+r,r-rrr:) r)rr1rr3r4r5r4rr) r;r<r=rr rr0r7rrrrnrrs  rc@sJeZdZddZdddZddd Z dd ddZ   d!d"ddZdS)#JSONSerializablecCrW)NrXzfilepath to JSON filerYrr rrrrs z JSONSerializable.serialized_inforrzcCrA)NrTrBrCr rrrr r[zJSONSerializable.api_inforcCsdddddS)Nrr)abrFrr rrrrszJSONSerializable.example_inputsr(Fr)r4r*r+r,r-dict | list | NonecCs&|dus|dkr dStt||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()r file_to_jsonrr/rrrr0s zJSONSerializable.serializeNstr | dict | listr1r2r3r5cCs|durdStj||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 Nrb)rdict_or_str_to_json_filerer6rrrr7szJSONSerializable.deserializerr8r9)r)r4r*r+r,r-rrr:) r)rr1r2r3r4r5r4rr4r;r<r=rr rr0r7rrrrrs    rc@sHeZdZddZdddZddd Z dd ddZ  d!d"ddZdS)#GallerySerializablecCrW)NrXz^path to directory with images and a file associating images with captions called captions.jsonrYrr rrrrrIz#GallerySerializable.serialized_inforrzcCrA)NrTrBrCr rrrr rEzGallerySerializable.api_inforcCstjgddgddS)Nrr\rFr]r rrrrs z"GallerySerializable.example_inputsr(Fr)r4r*r+r,r-list[list[str | None]] | Nonec Cs|dus|dkr dSg}t|d}|d }t|}Wdn1s'wY|D]\}} tj||d} || | gq0|S)Nr( captions.jsonr)r,)ropenjsonloadrvrgr0append) r r)r*r,files captions_file captions_jsoncaptionsrcaptionimgrrrr0s   zGallerySerializable.serializeNr1rr3r5 None | strc Cs|durdSt|tt}|jdddi}|D]=}t|ttfr)|\}}nd}tj ||||d} ||| <|d} | d} t || Wdn1sSwYqt j|S)NT)rr)r3r5rw)rruuiduuid4rrrrrgr7rrdumposrabspath) r r)r1r3r5 gallery_pathrimg_datarrerrrrrr7s&   zGallerySerializable.deserializerr8r9)r)r4r*r+r,r-rr)r(NN) r)rr1rr3r4r5r4rrrrrrrrs   rFile UploadButtontextboxnumberslidercheckbox checkboxgroupradiodropdownimagevideoaudiorf dataframe timeseries fileexplorerstatebutton uploadbutton) colorpickerlabelhighlightedtextrhtmlgallerychatbotmodel3dplotbarplotlineplot scatterplotmarkdowncodeannotatedimagezdict[str, type]COMPONENT_MAPPING))rL __future__rrrrrrpathlibrtypingr gradio_clientrrgradio_client.data_classesrr__file__parentrrrDrr?rNrQrSrTrVrgrrrSERIALIZER_MAPPING__subclasses__clsr;subclsr__annotations__rrrrs     5;?+3?