o An0j5R@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_infoselfr`/home/wildlama/miniconda3/envs/lam_a2e/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)NZserialized_inputinfotyperr getr r typesrrrinput_api_info+zSerializable.input_api_infocCr)NZserialized_outputrrrrrrr!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'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 LzSimpleSerializable.api_inforcCs dddS)NrawZ serializedrr rrrrRz!SimpleSerializable.example_inputsNrr>r6r9r:r;__doc__r rrrrrr=I r=c@r<) StringSerializablez?Expects a string as input/output but performs no serialization.rr>cCr?)NrKFr@rAr rrrr \rCzStringSerializable.api_inforcC dddS)NHowdy!rDrr rrrrbrFz!StringSerializable.example_inputsNrGr6rHrrrrrKYrJrKc@r<) ListStringSerializablezHExpects a list of strings as input/output but performs no serialization.rr>cCr?)NrNFr@rAr rrrr lrCzListStringSerializable.api_inforcCsddgddgdS)NrMZMerhabarDrr rrrrrsz%ListStringSerializable.example_inputsNrGr6rHrrrrrNirJrNc@r<) BooleanSerializablez@Expects a boolean as input/output but performs no serialization.rr>cCr?)NrOFr@rAr rrrr |rCzBooleanSerializable.api_inforcCrL)NTrDrr rrrrrFz"BooleanSerializable.example_inputsNrGr6rHrrrrrOyrJrOc@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)Nstringz+filepath on your computer (or URL) of imager descriptionrr rrrrrFzImgSerializable.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__rrrfszFileSerializable.__init__cCr rb)_single_file_serialized_infor rrrrz FileSerializable.serialized_infocCr?)NZSingleFileSerializableTr@rAr rrr_single_file_api_inforCz&FileSerializable._single_file_api_infocCrS)NrT*filepath on your computer (or URL) of filerUrr rrrrirFz-FileSerializable._single_file_serialized_infocCsddddddS)Narrayz&List of filepath(s) or URL(s) to filesrTrlrU)rrVitemsrr rrr_multiple_file_serialized_infos z/FileSerializable._multiple_file_serialized_infocCr?)NZMultipleFileSerializableTr@rAr rrr_multiple_file_api_inforCz(FileSerializable._multiple_file_api_infodict[str, dict | bool]cCr rb)rkr rrrr rjzFileSerializable.api_inforcCr rb)_single_file_example_inputsr rrrrrjzFileSerializable.example_inputscCdtjdddSNFis_filedatazMhttps://github.com/gradio-app/gradio/raw/main/test/test_files/sample_file.pdfrDrZ BASE64_FILEr rrrrr z,FileSerializable._single_file_example_inputscCsdtjdgdgdSrtrxr 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)r^rw orig_namesize) isinstancedictrrZstrrstatst_sizeZencode_url_or_file_to_base64r^)r r'r(r*filenamer~rrr_serialize_singles z"FileSerializable._serialize_singlecCs t||Srb)rZdownload_byte_stream)r urlr3rrr _setup_stream&s zFileSerializable._setup_streamNr/r2r1r3c 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|)Nr[rvr^zThe 'name' field is missing in zfile=)r3r\ is_streamz2name and root_url and save_dir must all be presentzstream/)r3T)parentsexist_okr}outputrwzThe 'data' field is missing in zLA FileSerializable component can only deserialize a string or a dict, not a z: )rrrr]r^rr ValueErrorZdownload_tmp_copy_of_fileZcreate_tmp_copy_of_filercrdrnextrtempfile gettempdirsecretsZ token_hexmkdir write_bytesr) r r'r/r1r3 file_namefilepathchunkpathrwrrr_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-rrrr.\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)r/r1r3)rrr3r1r/r rrrs z0FileSerializable.deserialize..r)rrrrrr4rrrr5qs  zFileSerializable.deserialize)rrarrqr6r7)r'r{r(r)r*r+rr|r8) r'r{r/r2r1r2r3r2rr2)r'rr(r)r*r+rr) r'rr/rr1r2r3r2rr)r9r:r;rIrfrrkrirorpr rrrrzrrrr.r5 __classcell__rrrgrr`s6       6 r`csTeZdZddZdddZddd Z d d!fdd Z   d"d#fdd ZZS)$VideoSerializablecCrS)NrTz0filepath on your computer (or URL) of video filerUrr rrrrrFz!VideoSerializable.serialized_inforrqcCr?)Nr`Tr@rAr rrrr rWzVideoSerializable.api_inforcCrs)NFruzNhttps://github.com/gradio-app/gradio/raw/main/test/test_files/video_sample.mp4rD)rZ BASE64_VIDEOr rrrrryz VideoSerializable.example_inputsr&Fr'r2r(r)r*r+tuple[FileData | None, None]cst|||dfSrb)rer.r-rgrrr.szVideoSerializable.serializeN.tuple[FileData | None, FileData | None] | Noner/rr1r3*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)rtuplerlenrrer5)r r'r/r1r3Z x_as_listZdeserialized_filergrrr5s  zVideoSerializable.deserializerr6r7)r'r2r(r)r*r+rrr8) r'rr/rr1r2r3r2rr) r9r:r;rr rr.r5rrrrgrrs  rc@sJeZdZddZdddZddd Z dd ddZ   d!d"ddZdS)#JSONSerializablecCrS)NrTzfilepath to JSON filerUrr rrrrs z JSONSerializable.serialized_inforrqcCr?)NrTr@rAr rrrr rWzJSONSerializable.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_inforrqcCr?)NrTr@rAr rrrr rCzGallerySerializable.api_inforcCstjgddgddS)NrrXrDrYr 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*)ropenjsonloadrnr`r.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)rruuiduuid4rrrrr`r5rrdumposrabspath) 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 timeseriesZ fileexplorerstateZbuttonZ uploadbutton)Z colorpickerlabelZhighlightedtextrhtmlZgalleryZchatbotZmodel3dZplotZbarplotZlineplotZ scatterplotmarkdowncodeZannotatedimagezdict[str, type]COMPONENT_MAPPING))rI __future__rrrrrrpathlibrtypingrZ gradio_clientrrZgradio_client.data_classesrr__file__parentrrrBrr=rKrNrOrPrRr`rrrZSERIALIZER_MAPPING__subclasses__clsr9subclsr__annotations__rrrrs     5;?+3?