o eP@s8UddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZeeejd ZeeZWdn1sNwYGdd d ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ 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+ed,ed-ed.ed/ed0eeeeee eeeeeeeeed1Z&d2e'd3<dS)4) 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_infoselfrID:\Projects\ConvertPro\env\Lib\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)NZserialized_inputinfotyperr getr r typesrrrinput_api_info)zSerializable.input_api_infocCr)NZserialized_outputrrrrrrr!rrroutput_api_info/r$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 serialize4szSerializable.serializeNsave_dirstr | Path | Noneroot_url str | Nonehf_tokencCr,)z] Convert data from serialized format for a browser to human-readable format. rr r'r/r1r3rrr deserialize:s zSerializable.deserialize)rrrr)rrr&F)r'rr(r)r*r+NNN)r'rr/r0r1r2r3r2) __name__ __module__ __qualname__rr rr#r%r.r5rrrrrs      rc@$eZdZdZd ddZd ddZd S) SimpleSerializablezIGeneral class that does not perform any serialization or deserialization.rdict[str, bool | dict]cCtdddS)Nr=Frrserializer_typesr rrrr JzSimpleSerializable.api_inforcCs dddS)NrawZ serializedrr rrrrPz!SimpleSerializable.example_inputsNrr>r6r9r:r;__doc__r rrrrrr=G r=c@r<) StringSerializablez?Expects a string as input/output but performs no serialization.rr>cCr?)NrKFr@rAr rrrr ZrCzStringSerializable.api_inforcC dddS)NHowdy!rDrr rrrr`rFz!StringSerializable.example_inputsNrGr6rHrrrrrKWrJrKc@r<) ListStringSerializablezHExpects a list of strings as input/output but performs no serialization.rr>cCr?)NrNFr@rAr rrrr jrCzListStringSerializable.api_inforcCsddgddgdS)NrMZMerhabarDrr rrrrpsz%ListStringSerializable.example_inputsNrGr6rHrrrrrNgrJrNc@r<) BooleanSerializablez@Expects a boolean as input/output but performs no serialization.rr>cCr?)NrOFr@rAr rrrr zrCzBooleanSerializable.api_inforcCrL)NTrDrr rrrrrFz"BooleanSerializable.example_inputsNrGr6rHrrrrrOwrJrOc@r<) NumberSerializablezKExpects a number (int/float) as input/output but performs no serialization.rr>cCr?)NrPFr@rAr rrrr rCzNumberSerializable.api_inforcCrL)NrDrr rrrrrFz!NumberSerializable.example_inputsNrGr6rHrrrrrPrJrPc@sNeZdZdZddZdddZdd d Z ddddZ   d d!ddZdS)"ImgSerializablezJExpects a base64 string as input/output which is serialized to a filepath.cC dddS)Nstringzfilepath or URL to imager descriptionrr rrrr zImgSerializable.serialized_inforr>cCr?)NrRTr@rAr rrrr zImgSerializable.api_inforcCs tjddS)NPhttps://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.pngrDrZ BASE64_IMAGEr rrrrszImgSerializable.example_inputsr&Fr'r2r(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_likeZencode_url_to_base64Zencode_file_to_base64rr-rrrr.s    zImgSerializable.serializeNr/r0r1r3cCs(|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'r/r1r3filerrrr5szImgSerializable.deserializerGr6r7)r'r2r(r)r*r+rr2r8) r'r2r/r0r1r2r3r2rr2) r9r:r;rIrr rr.r5rrrrrRs    rRcseZdZdZd6fdd 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__rrrgszFileSerializable.__init__cCr rc)_single_file_serialized_infor rrrrz FileSerializable.serialized_infocCr?)NZSingleFileSerializableTr@rAr rrr_single_file_api_inforCz&FileSerializable._single_file_api_infocCrS)NrTfilepath or URL to filerUrr rrrrjrWz-FileSerializable._single_file_serialized_infocCsddddddS)Narrayz&List of filepath(s) or URL(s) to filesrTrmrU)rrVitemsrr rrr_multiple_file_serialized_infosz/FileSerializable._multiple_file_serialized_infocCr?)NZMultipleFileSerializableTr@rAr rrr_multiple_file_api_inforCz(FileSerializable._multiple_file_api_infodict[str, dict | bool]cCr rc)rlr rrrr rkzFileSerializable.api_inforcCr rc)_single_file_example_inputsr rrrrrkzFileSerializable.example_inputscCdtjdddSNFis_filedatazMhttps://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdfrDrZ BASE64_FILEr rrrrs z,FileSerializable._single_file_example_inputscCsdtjdgdgdSruryr rrr_multiple_file_example_inputss z.FileSerializable._multiple_file_example_inputsr&Fr'str | FileData | Noner(r)r*r+FileData | NonecCsl|dus t|tr |St|r|}d}ntt||}t|j}||r)dnt|t|j ||dS)N)r_rx orig_namerwsize) isinstancedictrr[strrstatst_sizeZencode_url_or_file_to_base64r_)r r'r(r*filenamerrrr_serialize_singles z"FileSerializable._serialize_singlecCs t||Src)rZdownload_byte_stream)r urlr3rrr _setup_streams zFileSerializable._setup_streamNr/r2r1r3c Cs|durdSt|trtj||dj}|St|tr|drJ|d}|dus/Jd||durAtj|d|||d}|Stj||d}|S|dr|drW|rW|sYJ|j rc|j |dkrv|j |d|d|d |_ |d|_ t |j }t |pttd }|jd d d ||d d}||t|}|S|d} | dusJd|tj| |dj}|Stdt|d|)Nr\rwr_zThe 'name' field is missing in zfile=)r3r] is_streamzstream/)r3T)parentsexist_okr~outputrxzThe 'data' field is missing in zLA FileSerializable component can only deserialize a string or a dict, not a z: )rrrr^r_rr Zdownload_tmp_copy_of_fileZcreate_tmp_copy_of_filerdrernextrtempfile gettempdirsecretsZ token_hexmkdir write_bytes ValueErrorr) r r'r/r1r3 file_namefilepathchunkpathrxrrr_deserialize_singlesN "        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 ^sz.FileSerializable.serialize..)rlistrr-rrrr.Ms  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)r/r1r3)rrr3r1r/r rrrws z0FileSerializable.deserialize..r)rrrrrr4rrrr5bs  zFileSerializable.deserialize)rrbrrrr6r7)r'r|r(r)r*r+rr}r8) r'r|r/r2r1r2r3r2rr2)r'rr(r)r*r+rr) r'rr/rr1r2r3r2rr)r9r:r;rIrgrrlrjrprqr rrsr{rrrr.r5 __classcell__rrrhrras6       1 racsTeZdZddZdddZddd Z d d!fdd Z   d"d#fdd ZZS)$VideoSerializablecCrS)NrTzfilepath or URL to video filerUrr rrrrrWz!VideoSerializable.serialized_inforrrcCr?)NraTr@rAr rrrr rXzVideoSerializable.api_inforcCrt)NFrvzNhttps://github.com/gradio-app/gradio/raw/main/test/test_files/video_sample.mp4rD)rZ BASE64_VIDEOr rrrrrzz VideoSerializable.example_inputsr&Fr'r2r(r)r*r+tuple[FileData | None, None]cst|||dfSrc)rfr.r-rhrrr.szVideoSerializable.serializeN.tuple[FileData | None, FileData | None] | Noner/rr1r3*str | tuple[str | None, str | None] | Nonecspt|ttfrt|dksJd||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)rtuplerlenrrfr5)r r'r/r1r3Z x_as_listZdeserialized_filerhrrr5s  zVideoSerializable.deserializerr6r7)r'r2r(r)r*r+rrr8) r'rr/rr1r2r3r2rr) r9r:r;rr rr.r5rrrrhrrs  rc@sJeZdZddZdddZddd Z dd ddZ   d!d"ddZdS)#JSONSerializablecCrS)NrTzfilepath to JSON filerUrr rrrrrWz JSONSerializable.serialized_inforrrcCr?)NrTr@rAr rrrr rXzJSONSerializable.api_inforcCsdddddS)Nrr)abrDrr rrrrszJSONSerializable.example_inputsr&Fr'r2r(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&)rZ file_to_jsonrr-rrrr.s zJSONSerializable.serializeNstr | dict | listr/r0r1r3cCs|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 Nr\)rZdict_or_str_to_json_filer_r4rrrr5szJSONSerializable.deserializerr6r7)r'r2r(r)r*r+rrr8) r'rr/r0r1r2r3r2rr2r9r:r;rr rr.r5rrrrrs    rc@sHeZdZddZdddZddd Z dd ddZ  d!d"ddZdS)#GallerySerializablecCrS)NrTz^path to directory with images and a file associating images with captions called captions.jsonrUrr rrrrrFz#GallerySerializable.serialized_inforrrcCr?)NrTr@rAr rrrr rCzGallerySerializable.api_inforcCstjgddgddS)NrrYrDrZr rrrrs z"GallerySerializable.example_inputsr&Fr'r2r(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*)ropenjsonloadrorar.append) r r'r(r*files captions_file captions_jsoncaptionsrcaptionZimgrrrr.s   zGallerySerializable.serializeNr/rr1r3 None | strc Cs|durdSt|tt}|jdddi}|D]=}t|ttfr)|\}}nd}tj ||||d} ||| <|d} | d} t || Wdn1sSwYqt j|S)NT)rr)r1r3rw)rruuiduuid4rrrrrar5rrdumposrabspath) r r'r/r1r3Z gallery_pathrZimg_datarr_rrrrrr5s&   zGallerySerializable.deserializerr6r7)r'r2r(r)r*r+rr)r&NN) r'rr/rr1r2r3r2rrrrrrrrs   rFileZ UploadButtonZtextboxnumberZsliderZcheckboxZ checkboxgroupradioZdropdownimageZvideoZaudior`Z dataframeZ timeseriesstateZbuttonZ uploadbuttonZ colorpicker)labelZhighlightedtextrhtmlZgalleryZchatbotZmodel3dZplotZbarplotZlineplotZ scatterplotmarkdowncodeZannotatedimagezdict[str, type]COMPONENT_MAPPING)( __future__rrrrrrpathlibrtypingrZ gradio_clientrrZgradio_client.data_classesrr__file__parentrrrBrr=rKrNrOrPrRrarrrZSERIALIZER_MAPPING__subclasses__clsr9subclsr__annotations__rrrrs    585'3?