;3hlKddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZm Z dd l!m"Z"m#Z#mZddl$m%Z%ddl&m'Z'ddl(m)Z)e rddl$m*Z*m+Z+ddl&m,Z,ejZj]dxs e/e ej`dz Z1GddZ2y)N)Sequence)BytesIO)Path) TYPE_CHECKINGAny)types)Server)SseServerTransport)Image) Starlette) JSONResponseResponse)MountRoute)processing_utils route_utilsutils) BlockFunction)State)FileData) BlockContextBlocks) ComponentGRADIO_TEMP_DIRgradioc eZdZdZd!dZd"dZdefdZdede d e ddfd Z d e dd fd Z e de ddedefdZe de ddedefdZ d#d e dee e fdzdeee efeee ezffdZdefdZdee efdeee efeee ezffdZdedeee ezdefdZe de dej4dzfdZe dej4de de fdZdedeej<ej>zfd Z y)$GradioMCPServerz A class for creating an MCP server around a Gradio app. Args: blocks: The Blocks app to create the MCP server for. cR||_|jj|_|j|_d|_d|_tj}|r1|jdd}tjdd||_ nd|_ |jy)N/z [^a-zA-Z0-9]_)blocks get_api_infoapi_infocreate_mcp_server mcp_serverrequestroot_urlr get_spacesplitresub tool_prefixwarn_about_state_inputs)selfr#r.s J/var/www/html/audio-gradio/venv/lib/python3.12/site-packages/gradio/mcp.py__init__zGradioMCPServer.__init__+s  002 002  oo' %++C04K!vvosKHD !D  $$&returnNc<|jr|jdjD]o\}}|j|jdz}|ds*|j |}|s>t d|j Ds[tjdqyy)z= Warn about tools that have gr.State inputs. named_endpointsrshow_apic3<K|]}t|tywN) isinstancer).0inputs r1 z:GradioMCPServer.warn_about_state_inputs..Ds(5: 5%0(szThis MCP server includes a tool that has a gr.State input, which will not be updated between tool calls. The original, default value of the State will be used each time.N) r%itemsr.lstripget_block_fn_from_tool_nameanyinputswarningswarn)r0 endpoint_name endpoint_info tool_nameblock_fns r1r/z'GradioMCPServer.warn_about_state_inputs9s ==04 !1eg , }!,,}/C/CC/HH  ,#?? JHC(>Foo(%! .  r3c tttjjxsd}|j dtdt tt fdttjtjzffd }|jdttjffd }|S)z Create an MCP server for the given Gradio Blocks app. Parameters: blocks: The Blocks app to create the MCP server for. Returns: The MCP server. z Gradio Appname argumentsr4cjK j|\}} j||} j|}d|j j} j r>| j dvr- j d|d}t j|d|}ng}|td| j|j|} jj|| jd{} j|j|} j| dS74w) z Call a tool on the Gradio app. Args: name: The name of the tool to call. arguments: The arguments to pass to the tool. rr6 parametersN"Unknown tool for this Gradio app: )rHrBr(data)get_input_schemaconvert_strings_to_filedatar@ removeprefixr.r% client_utilsconstruct_args ValueErrorinsert_empty_staterBr# process_apir(pop_returned_statepostprocess_output_data) rJrKr!filedata_positionsprocessed_kwargsrHrEparameters_infoprocessed_argsoutputr0s r1 call_toolz4GradioMCPServer.create_mcp_server..call_toolYsH%)$9$9$$? !A!#??-  77=H 1 1$2B2B CDEM}}$--@Q2R!R"&--0A"B="Q #".!r.r?r@fnrget_function_descriptionrQappendrTool) toolsrErFrGrHrbrMschemar!r0s r1 list_toolsz5GradioMCPServer.create_mcp_server..list_toolss == E04 !1eg , }!,,}/C/CC/HH  ,#?? JH'8;;+> .3.L.L /+K!% 5 5i LIFALL !*(3(. &Ls ACBC) r strr#titler`dictrlistr TextContent ImageContentrjrg)r0serverr`rjs` r1r&z!GradioMCPServer.create_mcp_serverMsDKK--=>?    $ @$ @"&sCx.$ @ %##e&8&88 9$ @ $ @L     $uzz"2   : r3appsubpath root_pathc d}t|fd}|j|ttdjtd|t dj gy) z Launch the MCP server on the SSE transport. Parameters: app: The Gradio app to mount the MCP server on. subpath: The subpath to mount the MCP server on. E.g. "/gradio_api/mcp" z /messages/cK|_tj|d_ j |j |j |j4d{}jj|d|djjd{dddd{tS7a7 7#1d{7swYtSxYw#t$r}tdt|d}~wwxYww)Nz/gradio_api/mcp/sse)r( route_pathrtrzMCP SSE connection error: )r(r get_root_urlr) connect_ssescopereceive_sendr'runcreate_initialization_optionsr Exceptionprintrk)r(streamsertr0sses r1 handle_ssez5GradioMCPServer.launch_mcp_on_sse..handle_sses"DL'440#DM ??MM7??GMM//--  EEG z! z! 23q6(;< s%D 5C$CC$"?C!C"C& C$1C2 C$?D C$CC$C! C C! C$ D !C$$ D -DD  D z/schema)endpointz/sse)rr)routesN)r mountr rget_complete_schemarhandle_post_message)r0rrrsrt messages_pathrrs` ` @r1launch_mcp_on_ssez!GradioMCPServer.launch_mcp_on_ssesg%  / *  !!%!9!9&:6,C,C,CD   r3rGzBlockFunction | NonecztfdjjjDd}|S)a Get the BlockFunction for a given tool name. Parameters: tool_name: The name of the tool to get the BlockFunction for. Returns: The BlockFunction for the given tool name, or None if it is not found. c3rK|].}|jjjk(r|0ywr9)api_namerSr.)r;rdr0rGs r1r=z>GradioMCPServer.get_block_fn_from_tool_name..s6 ;;)"8"89I9I"JJ s47N)nextr#fnsvalues)r0rGrHs`` r1r@z+GradioMCPServer.get_block_fn_from_tool_names9 ++//002    r3rBzComponent | BlockContextrPcrt|D](\}}t|ts|j|d*|Sr9) enumerater:rinsertrBrPiinput_component_types r1rWz"GradioMCPServer.insert_empty_states>(1'8 % #A#.6 At$ % r3cpt|D]'\}}t|ts|j|)|Sr9)rr:rpoprs r1rYz"GradioMCPServer.pop_returned_states;(1'8  #A#.6   r3rMc 2d|j|j}|jd}|j|}|t d|d|dDcic]"}|di|d|r|d|vr d||dini$c}d }|j |Scc}w) a Get the input schema of the Gradio app API, appropriately formatted for MCP. Parameters: tool_name: The name of the tool to get the schema for, e.g. "predict" parameters: The description and parameters of the tool to get the schema for. Returns: - The input schema of the Gradio app API. - A list of positions of FileData objects in the input schema. rr6rOobjectrMparameter_nametyperb)r properties)rSr.r%getrVsimplify_filedata_schema)r0rGrMrEr6rFpris r1rQz GradioMCPServer.get_input_schemasI2243C3CDEF --(9:'++M:  A)MN N'|4 "#&i&&!,<*=*K' 15E3F(GH &   ,,V44 s'BcK|js tiSi}|jdjD]\}}|j|j dz}|ds*|j |}| |j Jtj|j \}}|j||\} } | ||<|||d<t|Sw)a Get the complete schema of the Gradio app API. (For debugging purposes) Parameters: request: The Starlette request object. Returns: A JSONResponse containing a dictionary mapping tool names to their input schemas. r6rr7rb) r%r r>r.r?r@rdrrerQ) r0r(schemasrErFrGrHrbrMrir!s r1rz#GradioMCPServer.get_complete_schemas}}# #,0MM:K,L,R,R,T @ (M=((=+?+?+DDIZ(;;IF#x{{':*/*H*H*U' Z 11)ZH %+ "4? "=1 @G$$s A C#A1Cric dtdtttfdtfd d dtdttt zdzdtttfdzdtffd g|}|fS) a Parses a schema of a Gradio app API to identify positions of FileData objects. Replaces them with base64 strings while keeping track of their positions so that they can be converted back to FileData objects later. Parameters: schema: The original schema of the Gradio app API. Returns: A tuple containing the simplified schema and the positions of the FileData objects. objdefsr4c0t|tsyd|vr>|d}|jdr'|jdd}|j |i}ny|j di}|j di}d|vr>|d}|jdr'|jdd}|j |i}ny|j dij di}|j d ij d}|j d d k(xs|d k(S) NFz$refz#/$defs/rr rmeta_typedefaultconstzgradio.FileData)r:rm startswithr+r)rrrefkeypropsr type_field default_types r1is_gradio_filedatazDGradioMCPServer.simplify_filedata_schema..is_gradio_filedataDsc4(}&k>>*-))C.,C((3+C GGL"-E99VR(D~6l>>*-))C.,C88C,D ,377DJ88Ir266w?Lw'+<<5#44 r3Nnodepathc|g}|i}t|trd|vr|j|d ||rB j|j dD]}|j |dd|d<d|d<i}d|vxrd|v}|j D]H\}}|r|dk(r |||||<|j| |||||<|j J|St|trQg}t|D]?\}}|j||j ||||j A|S|S)N$defs)radditional_descriptionrstringrza http or https url to a fileformatr) r:rmupdaterfcopyrr>rnr) rrrrresultis_schema_rootvalueritemr[rtraverses r1rz:GradioMCPServer.simplify_filedata_schema..traversebs] ||$%d?KKW .%dD1&--diik:P,d+,#+DL%DDN!'4!HLD4H"&**,#JC%#*=&.udD&As  C(&.udD&As  # D$'(GAtKKNMM(4t"<=HHJ Kr3)NN)rrmrkboolrnint)r0risimplified_schemar[rrs @@@r1rz(GradioMCPServer.simplify_filedata_schema5s C tCH~ $ @,0*.( ( sSy/D(( sCx.4'(  ( T57$V, "444r3rr[clddtdtttzdzdtffd |S)a' Convert specific string values back to FileData objects based on their positions. This is used to convert string values (as base64 encoded strings) to FileData dictionaries so that they can be passed into .preprocess() logic of a Gradio app. Parameters: value: The input data to process, which can be an arbitrary nested data structure that may or may not contain strings that should be converted to FileData objects. filedata_positions: List of paths to positions in the input data that should be converted to FileData objects. Returns: The processed data with strings converted to FileData objects where appropriate. Base64 encoded strings are first saved to a temporary file and then converted to a FileData object. Example: >>> convert_strings_to_filedata( {"image": "data:image/jpeg;base64,..."}, [["image"]] ) >>> {'image': FileData(path='')}, Nrrr4c |g}t|tr/|jDcic]\}}||||gzc}}St|tr)t |Dcgc]\}}|||gzc}}St|t rd|vr`|j dr$ttj|tS|j dr t|Std||Scc}}wcc}}w)Nzdata:)r)zhttp://zhttps://zSInvalid file data format, provide a url ('http://...' or 'https://...'). Received: ) r:rmr>rnrrkrrrsave_base64_to_cacheDEFAULT_TEMP_DIRrV)rrrrrrr[rs r1rz=GradioMCPServer.convert_strings_to_filedata..traverses|$%IM;E3C%66D$'BKD/Rwq$tqcz2RRD#&43E+E??7+$-BB "2 __%<=#..$mnrmstK)Ss C4#C:r9)rrnrkr)r0rr[rs `@r1rRz+GradioMCPServer.convert_strings_to_filedatas62 3 d39o&<  6r3 file_pathctjj|sytjj|j d}|t j vry t j|S#t$rYywxYw)ze If a filepath is a valid image, returns a PIL Image object. Otherwise returns None. Nrx) osrexistssplitextlowerr registered_extensionsopenr)rexts r1 get_imagezGradioMCPServer.get_imagesq ww~~i(ggy01!4 e113 3 ::i( (  s)A>> B  B imagerct}|j||tj|j j dS)z? Returns a base64 encoded string of the image. )rzutf-8)rsavebase64 b64encodegetvaluedecode)rrbuffers r1get_base64_datazGradioMCPServer.get_base64_datas>  6& ) 1299'BBr3chg}|jr!tj||jd}|D]}tj|r|j |dx}rr|j xsd}|j||}d|j}tjd||tjdd|d xs|d g}nMtjdt|d xs|d g}n!tjdt| g}|j||S) z Postprocess the output data from the Gradio app to convert FileData objects back to base64 encoded strings. Parameters: data: The output data to postprocess. Nrpngzimage/r)rrPmimeTypetextz Image URL: url)rr)r)r add_root_urlrTis_file_obj_with_metarrrrrrprorkextend) r0rP return_valuesr_r image_format base64_datamimetype return_values r1rZz'GradioMCPServer.postprocess_output_datas9 ==#00t}}dKD /F11&9 NN6&>::5:#(<<#85L"&"6"6ul"KK!' (:(:(<'=>H**!({X))!'#.ve}/Nv.O!P $L))!'c&-2Q6&>.R$L !& 1 1vCK PQ   ./ /0r3)r#r)r4Nr9)!__name__ __module__ __qualname____doc__r2r/r r&r rkrr@ staticmethodrrnrWrYrmtuplerrrQr rrrRr rrrrorprZrNr3r1rr#s  '(Q6Qf, Y, , , QU, \S=S(34<@ 34<@ -1$5$5cNT)$5 tCH~tDsO44 5 $5L%L%6Y538nY5 tCH~tDsO44 5Y5v44.24c ?.C4 4l S U[[4%7  Cu{{CCCCCC$$ e%"4"44 5$r3r)3rrr,tempfilerCcollections.abcriorpathlibrtypingrrgradio_client.utilsrrTmcpr mcp.serverr mcp.server.sser PILr starlette.applicationsr starlette.responsesr rstarlette.routingrrrrr gradio.blocksrgradio.componentsrgradio.data_classesrrrrenvironrrk gettempdirrrrNr3r1rs $%*-,6*77'#(2+::>>"34    (*9 __r3