kh`ddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZmZddlmZddlmZddlZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0mZddl1m2Z2ddl3m4Z4ddl5m6Z6erddl1m7Z7m8Z8ddl3m9Z9ejtjwdxs e<e ejzdz Z>GddZ?y)N) AsyncIteratorSequence)BytesIO)Path) TYPE_CHECKINGAny)unquote)types)Server)SseServerTransport)StreamableHTTPSessionManager)Image) Starlette)Request) JSONResponseResponse)MountRoute)ReceiveScopeSend)processing_utils route_utilsutils) BlockFunction)State)FileData) BlockContextBlocks) ComponentGRADIO_TEMP_DIRgradioc PeZdZdZd(dZdedefdZdeeeffdZ d)dZ de fd Z d e d ed eddfd ZdeddfdZededdedefdZededdedefdZ d*dedeeefdzdeeeefeeeezffdZdefdZdeeefdeeeefeeeezffdZdedeeeezdefdZededej:dzfd Zed!ededzfd"Z ed#ej:d$edefd%Z!ded&edee"jFe"jHzfd'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. cT||_|jj|_|j|_d|_t j}|r4|jdddz}tjdd||_ nd|_ |j|_ |jt|jddd t d t"d t$d dffd }t&j(dt*d t,dffd }||_|_||_y)N/_z [^a-zA-Z0-9]FT)app json_response statelessscopereceivesendreturncHKj|||d{y7wN)handle_request)r-r.r/managers F/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/gradio/mcp.pyhandle_streamable_httpz8GradioMCPServer.__init__..handle_streamable_httpCs!((> > >s " "r*cKj4d{ d dddd{y7#wxYw7 #1d{7swYyxYww)z7Context manager for managing session manager lifecycle.N)run)r*r4s r5lifespanz*GradioMCPServer.__init__..lifespanHsR{{}           sLA5A>7> A<A9>AAA A A)blocks get_api_infoapi_infocreate_mcp_server mcp_server root_pathr get_spacesplitresub tool_prefixget_tool_to_endpointtool_to_endpointwarn_about_state_inputsr rrr contextlibasynccontextmanagerrrr9r4r6)selfr:rDr6r9r4s @r5__init__zGradioMCPServer.__init__1s  002 002oo' %++C04s:K!vvosKHD !D  $ 9 9 ; $$&.u  ? ?#* ?26 ?  ?  ' '   mD.A  ( !  &<#requestr0ctjt|j}|j d}t|j d}|j dryy)z Gets the route path of the MCP server based on the incoming request. Can be different depending on whether the request is coming from the MCP SSE transport or the HTTP transport. N)queryr'z/gradio_api/mcp/messagesz/gradio_api/mcp/http)httpxURLstrurl copy_withrstripendswith)rJrMrSs r5get_route_pathzGradioMCPServer.get_route_pathUsP iiGKK()mm$m'#hooc" <<2 3-)rLci}|jdjD]\}}|ds |j|}| |j,t |jddxsLt |jdxr!t |jj ddxs|jd}|j|z}||vr |dz}||vr |||<|S)z Gets all of the tools that are exposed by the Gradio app and also creates a mapping from the tool names to the endpoint names in the API docs. named_endpointsshow_apiN__name__ __class__r'r)) r<itemsget_block_fn_from_endpoint_namefngetattrhasattrr\lstriprD)rJrF endpoint_name endpoint_infoblock_fnfn_name tool_names r5rEz$GradioMCPServer.get_tool_to_endpointbs ,0MM:K,L,R,R,T < (M=Z(?? N#x{{':HKKT:1 [9M#HKK$9$9:tL1 %++C0 !,,w6 #33 )CI #33.; +! <" rLNc|jjD]K\}}|j|}|std|jDs7t j dMy)z= Warn about tools that have gr.State inputs. c3<K|]}t|tywr2) isinstancer).0inputs r5 z:GradioMCPServer.warn_about_state_inputs..sVU 5% 8VszThis 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)rFr]r^anyinputswarningswarn)rJr)rcres r5rGz'GradioMCPServer.warn_about_state_inputs{s\!% 5 5 ; ; =  A};;MJHCVhooVV & rLc 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 argumentsr0cK jjj}| td j |}t j || j} j|\}} j||} jj|}|td| j|} | J| jdvr- jd|d} tj| d|} ng} j!| j"| } j$j'| | |d{} j)| j"| } j+| d |S75w) 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. NzCould not find the request object in the MCP server context. This is not expected to happen. Please raise an issue: https://github.com/gradio-app/gradio.)rM route_pathr?"Unknown tool for this Gradio app: rY parameters)rerorMdata)r>request_contextrM ValueErrorrWr get_root_urlr?get_input_schemaconvert_strings_to_filedatarFgetr^r< client_utilsconstruct_argsinsert_empty_stateror: process_apipop_returned_statepostprocess_output_data)rsrtcontext_requestrvroot_urlr)filedata_positionsprocessed_kwargsrcreparameters_infoprocessed_argsoutputrJs r5 call_toolz4GradioMCPServer.create_mcp_server..call_tools#oo==EEO& p,,_=J"//'%..H %)$9$9$$? !A!#??-  !1155d;M$ #EdV!LMM;;MJH' '' .? @@"&--0A"B="Q #".!.list_toolss E,0,A,A,G,G,I ( =?? N+ 0GGG383Q3QKK40 Z*33C8c&'))G,-K !11)ZH  JJ&$/$* (LsCC) r rRr:titlerdictrlistr TextContent ImageContentrr)rJserverrrs` r5r=z!GradioMCPServer.create_mcp_serversDKK--=>?    2 J2 J"&sCx.2 J %##e&8&88 92 J 2 Jh     $uzz"2   6 rLr*subpathr?c d}t||_fd}|j|tt dj t d|t djt djgy ) a! 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" root_path: The root path of the Gradio Blocks app. z /messages/cK j|j|j|j4d{}jj |d|djj d{dddd{tS7a7 7#1d{7swYtSxYw#t$r}tdt|d}~wwxYww)NrzMCP SSE connection error: ) connect_sser-r._sendr>r8create_initialization_optionsr ExceptionprintrR)rMstreamserJsses r5 handle_ssez5GradioMCPServer.launch_mcp_on_sse..handle_sses ??MM7??GMM//--  EEG z! z! 23q6(;< sC)5CBC?B"=B>B" C B  CC)CB" C"B=(B+ )B=0 C<C)=C C& C!!C&&C)z/schema)endpointz/sse)r*z/http/)routesN) r r?mountrrget_complete_schemarhandle_post_messager6)rJr*rr? messages_pathrrs` @r5launch_mcp_on_ssez!GradioMCPServer.launch_mcp_on_sses~%  /"   !!%!9!9&:6,C,C,CD((C(CD  rLrczBlockFunction | Nonecvtfd|jjjDd}|S)a$ Get the BlockFunction for a given endpoint name (e.g. "/predict"). Parameters: endpoint_name: The name of the endpoint to get the BlockFunction for. Returns: The BlockFunction for the given endpoint name, or None if it is not found. c3^K|]$}|jjdk(r|&yw)r'N)api_namerb)rkr_rcs r5rmzBGradioMCPServer.get_block_fn_from_endpoint_name..s0 ;;-"6"6s";; s*-N)nextr:fnsvalues)rJrcres ` r5r^z/GradioMCPServer.get_block_fn_from_endpoint_names9 ++//002    rLrozComponent | BlockContextrzcrt|D](\}}t|ts|j|d*|S)z Insert None placeholder values for any State input components, as State inputs are not included in the endpoint schema. N) enumeraterjrinsertrorziinput_component_types r5rz"GradioMCPServer.insert_empty_state(s>(1'8 % #A#.6 At$ % rLcpt|D]'\}}t|ts|j|)|S)z Remove any values corresponding to State output components from the data as State outputs are not included in the endpoint schema. )rrjrpoprs r5rz"GradioMCPServer.pop_returned_state5s;(1'8  #A#.6   rLrgrxc X|jj|}|td||jd}|j|}|Jd|dDcic]4}|di|d|r|d|vr d||dinid|vr |drd |dini6c}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. rwrYobjectrxparameter_nametyperparameter_defaultdefault)r properties)rFrr|r<simplify_filedata_schema)rJrgrxrcrYrdprs r5r~z GradioMCPServer.get_input_schemaBs--11)<  A)MN N--(9:'++M: ((('|4"# &i &&!,<*=*K' 15E3F(GH &/!3:M8N#A&9$:; &  &,,V44#s9B'cK|js tiSg}|jjD]\}}|j |}| |j Jt j|j \}}}|r-||jdrdnddzdj|zz }|j||\} } ||| d} |j| t|Sw)a Get the complete schema of the Gradio app API. For debugging purposes, also used by the Hugging Face MCP server to get the schema for MCP Spaces without needing to establish an SSE connection. Parameters: request: The Starlette request object. Returns: A JSONResponse containing a dictionary mapping tool names to their input schemas. rr&rrr) r<rrFr]r^r_rrrVrr~r) rJrMschemasrgrcrerrxrrr)infos r5rz#GradioMCPServer.get_complete_schemams}}# #(,(=(=(C(C(E ! $I};;MJH'HKK,C CC/4/M/M 0 ,KW&//4R#"#ii() --iDIFA!*%D NN4 % !(G$$sC&C(rc 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. objdefsr0c0t|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/r'r(rmeta_typerconstzgradio.FileData)rjr startswithrAr)rrrefkeypropsr type_field default_types r5is_gradio_filedatazDGradioMCPServer.simplify_filedata_schema..is_gradio_filedatasc4(}&k>>*-))C.,C((3+C GGL"-E99VR(D~6l>>*-))C.,C88C,D ,377DJ88Ir266w?Lw'+<<5#44 rLNnodepathc|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) rjrupdatercopyrr]rr) rrrrresultis_schema_rootvalueritemrrtraverses r5rz:GradioMCPServer.simplify_filedata_schema..traverses] ||$%d?KKW .%dD1&--diik:P,d+,#+DL%DDN!'4!HLD4H"&**,#JC%#*=&.udD&As  C(&.udD&As  # D$'(GAtKKNMM(4t"<=HHJ KrL)NN)rrrRboolrint)rJrsimplified_schemarrrs @@@r5rz(GradioMCPServer.simplify_filedata_schemas C tCH~ $ @,0*.( ( sSy/D(( sCx.4'(  ( T57$V, "444rLrrclddtdtttzdzdtffd |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='')}, Nrrr0c |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: ) rjrr]rrrRrrrsave_base64_to_cacheDEFAULT_TEMP_DIRr|)rrrrrrrrs r5rz=GradioMCPServer.convert_strings_to_filedata..traverses|$%IM;E3C%66D$'BKD/Rwq$tqcz2RRD#&43E+E??7+$-BB "2 __%<=#..$mnrmstK)Ss C4#C:r2)rrrRr)rJrrrs `@r5rz+GradioMCPServer.convert_strings_to_filedatas62 3 d39o&<  6rL 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. Nr) osrexistssplitextlowerrregistered_extensionsopenr)rexts r5 get_imagezGradioMCPServer.get_image$sq ww~~i(ggy01!4 e113 3 ::i( (  s)A>> B  B  file_datact|tra|jdx}rNt|tr=|j dr,t |j dddjSyy)z If a file_data is a valid FileDataDict with a url that is a data:image/svg+xml, returns bytes of the svg. Otherwise returns None. rSzdata:image/svg,rN)rjrrrRrr rAencode)rrSs r5get_svgzGradioMCPServer.get_svg3sa i &9==3G,GC,G#s#7G(Hsyya034;;==rLimagerct}|j||tj|j j dS)z? Returns a base64 encoded string of the image. )rutf-8)rsavebase64 b64encodegetvaluedecode)rrbuffers r5get_base64_datazGradioMCPServer.get_base64_data@s>  6& ) 1299'BBrLrcdg}tj||d}|D]}|j|x}rtj|j d}d}tj ||dt}|d|} tjd||tjdd |  g} ntj|r|j|d x} rr| jxsd } |j| | }d | j!}tjd||tjdd|dxs|d  g} nMtjdt#|dxs|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. Nrz image/svg+xml orig_namez/gradio_api/file=r)rrzmimeTypetextzSVG Image URL: )rrrpngzimage/z Image URL: rS)r add_root_urlrrr r save_bytes_to_cacherr rrris_file_obj_with_metarrr rrRextend) rJrzr return_valuesr svg_bytes base64_datamimetypesvg_pathsvg_url return_valuer image_formats r5rz'GradioMCPServer.postprocess_output_dataIs ,,T8TB' /F LL00y0$..y9@@I *+??&"5!69I&J&7zB&&$;%%#.wi8   33F; NN6&>::5:#(<<#85L"&"6"6ul"KK!' (:(:(<'=>H**!({X))!'#.ve}/Nv.O!P $L))!'c&-2Q6&>.R$L !& 1 1vCK PQ   .O' /PrL)r:r)r0Nr2)&r[ __module__ __qualname____doc__rKrrRrWrrErGr r=rrr^ staticmethodrrrrtuplerrr~rrrrrrbytesrr r rrrryrLr5r$r$)se"=H *g *# * d38n 2 ]6]~) Y) ) ) QU) V  , 34 <@     34 <@    -1)5)5cNT))5 tCH~tDsO44 5 )5V$%L$%LY538nY5 tCH~tDsO44 5Y5v44.24c ?.C4 4l S U[[4%7   3 54<  Cu{{CCCCCC33#&3 e%"4"44 53rLr$)@rrHrrBtempfilerpcollections.abcrriorpathlibrtypingrr urllib.parser gradio_client.utilsrrrPmcpr mcp.serverr mcp.server.sser "mcp.server.streamable_http_managerr PILrstarlette.applicationsrstarlette.requestsrstarlette.responsesrrstarlette.routingrrstarlette.typesrrrr"rr gradio.blocksrgradio.componentsrgradio.data_classesrrrr environrrR gettempdirrr$ryrLr5r;s  3% * -K,&6*0077'#(2+::>>"34    (*9 S S rL