a d!! @s4dZddlmZddlZddlZddlZddlZddlZddlm Z ddl Z ddl Z ddl m Z ddlmZddlmZe rddlmZeed d Zeed d Zed dZdZGddde jZddddddZddddddZd*dddddddddd d!Zd"dd"d"d#d$d%Zd"d&d'd(d)ZdS)+z^ Defines helper methods useful for setting up ports, launching servers, and creating tunnels. ) annotationsN) TYPE_CHECKING)ServerFailedToStartError)App)Tunnel)BlocksZGRADIO_SERVER_PORTZ7860ZGRADIO_NUM_PORTSZ100ZGRADIO_SERVER_NAMEz 127.0.0.1z(https://api.gradio.app/v2/tunnel-requestc@s$eZdZddZddZddZdS)ServercCsdS)Nselfr r Z/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/gradio/networking.pyinstall_signal_handlers!szServer.install_signal_handlerscCsRtj|jdd|_|jt}|jsNtdt|dkr$tdq$dS)NT)targetdaemongMbP?z@Server failed to start. Please check that the port is available.) threadingThreadrunthreadstarttimestartedsleepr)r rr r r run_in_thread$s  zServer.run_in_threadcCsd|_|jdS)NT)Z should_exitrjoinr r r r close/sz Server.closeN)__name__ __module__ __qualname__r rrr r r r r s rint)initialfinalreturnc Cs|t||D]R}z:t}|tjtjd|t|f||WStyZYq 0q td|d|dddS)a- Gets the first open port in a specified range of port numbers Parameters: initial: the initial value in the range of port numbers final: final (exclusive) value in the range of port numbers, should be greater than `initial` Returns: port: the first open port in the range zAll ports from z to z! are in use. Please close a port.N) rangesocket setsockopt SOL_SOCKET SO_REUSEADDRbindLOCALHOST_NAMErOSError)r r!portsr r r get_first_available_port4s   r.rr)appblocksr"cCsX|j}|dur0t|s(dd|D|_q6||_nd|_||_t|_|j|_i|_|S)NcSsi|]}|d|dqS)rr#r ).0accountr r r Oz!configure_app..)authcallabler0osgetcwdcwdZ favicon_pathtokens)r/r0r5r r r configure_appKs r;z str | Nonez int | Nonez dict | Nonez!tuple[str, int, str, App, Server])r0 server_name server_port ssl_keyfile ssl_certfilessl_keyfile_password app_kwargsr"c Csh|dur|durtd|pt}|dkr,dn|}|drR|drR|dd}n|}tj||d } |durr|gn tttt} | D]} zht } | t j t j d| t| f| tj| | |d |||d d } t| d }|Wq*WqttfyYq0qtdt| dt| d|durHd|d| d}nd|d| d}|| || |fS)arLaunches a local server running the provided Interface Parameters: blocks: The Blocks object to run on the server server_name: to make app accessible on local network, set this to "0.0.0.0". Can be set by environment variable GRADIO_SERVER_NAME. server_port: will start gradio app on this port (if available). Can be set by environment variable GRADIO_SERVER_PORT. auth: If provided, username and password (or list of username-password tuples) required to access the Blocks. Can also provide function that takes username and password and returns True if valid login. ssl_keyfile: If a path to a file is provided, will use this as the private key file to create a local server running on https. ssl_certfile: If a path to a file is provided, will use this as the signed certificate for https. Needs to be provided if ssl_keyfile is provided. ssl_keyfile_password: If a password is provided, will use this with the ssl certificate for https. app_kwargs: Additional keyword arguments to pass to the gradio.routes.App constructor. Returns: port: the port number the server is running on path_to_local_server: the complete address that the local server can be accessed at app: the FastAPI app object server: the server object that is a subclass of uvicorn.Server (used to close the server) Nz9ssl_certfile must be provided if ssl_keyfile is provided.z0.0.0.0 localhost[]r#)rAwarningi@)r/r,hostZ log_levelr>r?r@Z ws_max_size)configz!Cannot find empty port in range: -z. You can specify a different port by setting the GRADIO_SERVER_PORT environment variable or passing the `server_port` parameter to `launch()`.zhttps://:/zhttp://) ValueErrorr* startswithendswithrZ create_appr$INITIAL_PORT_VALUE TRY_NUM_PORTSr%r&r'r(r)ruvicornZConfigrrr+rminmax)r0r<r=r>r?r@rAZ url_host_namerGr/Z server_portsr,r-rHserverZpath_to_local_serverr r r start_server[sP    rUstr) local_host local_port share_tokenr"c Cstt}|r|jdkrz@|d}|dt|d}}t|||||}|}|WSty} zt t | | WYd} ~ n d} ~ 00t ddS)NrrGr,z0Could not get share link from Gradio API Server.) requestsgetGRADIO_API_SERVER status_codejsonrrZ start_tunnel Exception RuntimeErrorrV) rWrXrYresponsepayload remote_host remote_portZtunneladdresser r r setup_tunnels   $rhbool)urlr"c CszptdD]b}t*tdtj|ddd}Wdn1sF0Y|jdvrbWdStdq Wnt tj j fyYdS0dS) NrignoreF)timeoutverify)rZii.Tg?) r$warningscatch_warningsfilterwarningsr[headr^rrConnectionError exceptions)rj_rr r r url_oks   . rw)NNNNNN) __doc__ __future__rr7r%rrrotypingrr[rQZgradio.exceptionsrZ gradio.routesrZgradio.tunnelingrZ gradio.blocksrrgetenvrOrPr*r]rr.r;rUrhrwr r r r s<       Z