o i+@sdZddlZddlZddlZddlmZddlmZddlZ ddl Z ddl m Z ddl mZddlmZd d lmZd d lmZd d lmZd d lmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&ddZ'd d!Z(d"e)d#e)fd$d%Z*d&edBd'e+d(e)d)e)d*e)d+e)d,e,e)e)ffd-d.Z-    /dEd0e.d1e j/dBd2edBd3edBd4e+d,ef d5d6Z0d7e)d,dfd8d9Z1d:d:d;d7e)dd?Z2d1e j/dBd,e3e)e)ffd@dAZ4dBe3e)e)fd,e,e)dBe3e)efffdCdDZ5dS)F N) timedelta)Any)version) assert_never)Config)utils) APIClient)GPUSize) AuthLevel) QuotaInfos)ScheduleResponse)info)error) get_event) supports_authz X-IP-Token<zGPU device not usedzNo GPU was availablezNo GPU was available after 60szDTry re-running outside of examples if it happened after clicking onezCreate a free accountzhttps://huggingface.co/joinzSubscribe to Proz4https://huggingface.co/settings/billing/subscriptioncCs(tjdusJtjtjddd}t|S)NrF)base_urltimeoutverify)rzero_device_api_urlhttpxClientr ) httpx_clientrF/home/ubuntu/.local/lib/python3.10/site-packages/spaces/zero/client.py api_client)srcCspd\}}t}|}tjjur,td|d}|kr"td|}tjjus|tjjur6tddS)N)rrr z*Error while initializing ZeroGPU: NotFoundz)Error while initializing ZeroGPU: Unknown) rstartup_reportrcodes NOT_FOUNDtimesleep RuntimeErrorOK)retries max_retriesclientstatusrrrr/s  r html_contents text_contentscsGfdddt}||S)NcseZdZfddZdS)zhtml_string..HTMLStringcsS)Nr)selfr+rr__str__<sz'html_string..HTMLString.__str__N)__name__ __module__ __qualname__r.rr-rr HTMLString;sr2)str)r*r+r2rr-r html_string:sr4auth supports_html pro_message unlogged_desc logged_descendingreturnc Cstr|dkr ||fSd}|durtnt}|durtnt}|dur#|n|} | d|d7} dgd} |rA|d| d|d 7}||7}|rK|d 7}|d| 7}d |d |d | } || fS)Npro .;)zwhite-space: nowrapztext-underline-offset: 2pxzcolor: var(--body-text-color)z z[z](z) )rSIGNUP_ON_HF_URLSUBSCRIBE_TO_PRO_URLSIGNUP_ON_HF_TXTSUBSCRIBE_TO_PRO_TXTjoin) r5r6r7r8r9r:htmllinktextdescstylemarkdownrrr _toast_actionAs rMTtask_idrequestdurationgpu_size_first_attemptc Csttj}jdkstd|tdk}|tdk}t|}t|\} } | dur?| d} r?d| d} t d| d d t j t || |rKd nd |durS|jnd|d \} }|j}t| trd| St| tr|durp|jnt}| jtdkrd|d} td| | durdt} td| | ddur| jdkrd} n>|dkrdn|dkrdnd}d|d|d| jd| j}|durd}n |dkrd }n|}| d!d"k}|r|nt||} td| d#d$t| tjs|jd%vrt d&d'n|jd(krt d&d)nt|jt}|dur|durt !d* zt"| }Wn%t#y.td+tj$yG|s=td,t |||d-d.YSw|j%d/kra|dur_|j&s_| 'td0q|j%d1kr| durxt(d2t} td| t)||d3d4d5d6d7\}}t(d2|}t(d8|}t||} td9| d#d$|j%d:kr|j*dusJ|dur|j&st+|j*j,td;t d&dt.| tjj/urtd>d?td@| dAtj0| d)BNz)ZeroGPU is only compatible with Gradio 4+z4.39z5.16.1rz!Falling back to IP-based quotas ()ZeroGPU client warningwarninglevelrr ) cgroup_pathrNtoken token_versionduration_secondsrQrzThe requested GPU duration (z%s) is larger than the maximum allowedzZeroGPU illegal durationz%Space app has reached its GPU limit. zZeroGPU quota exceededuserz"You have exceeded your runs limit.r<zPro GPUregularzfree GPUGPUzYou have exceeded your z quota (zs requested vs. zs left). Try again in zUnlogged user is runnning out of daily ZeroGPU quotas. Signup for free on https://huggingface.co/join or login on https://huggingface.co/login to get more ZeroGPU quota now.zwUser is runnning out of daily ZeroGPU quotas. Visit https://huggingface.co/subscribe/pro to get more ZeroGPU quota now.z x-gradio-usermcpT)rG)nodeNz ZeroGPU queuez%Waiting for a GPU to become available concurrencyz$Waiting for a GPU slot on this Spacez-ZeroGPU: Cannot get Gradio app Queue instancezUnexpected end of streamz,Error while re-trying after queue disconnectF)rRpingz+Connection closed by visitor while queueingfailedz. z Retry laterzto get a higherzto get the highestzpriority in ZeroGPU queues)r5r6r7r8r9r:r>zZeroGPU queue timeout succeededz-Connection closed by visitor on queue successzSuccessfully acquired a GPUsuccessZeroGPU client errorzExpired ZeroGPU proxy tokenzZeroGPU API /schedule error:  ()1rparsegr __version__majorr$Version _get_headers_get_token_and_payloadgetrrscheduler self_cgroup_device_pathsecondsr5 isinstancerrDEFAULT_SCHEDULE_DURATIONwaitrrEXAMPLES_RETRY_MESSAGEleftr4rr queuing_reasonrrwarningswarnnext StopIterationRemoteProtocolErroreventalivecloseNO_GPU_MESSAGE_INQUEUErMdatarelease allowTokenSERVICE_UNAVAILABLENO_GPU_MESSAGE_REGULAR UNAUTHORIZEDget_reason_phrase)rNrOrPrQrRgradio_versionGRADIO_HTML_TOASTSGRADIO_HANDSHAKEheadersrZpayload token_errormessageresmetar5 requestedgpu message_gui message_mcpmcp_userconnection_eventr details_htmldetails_markdown message_html message_textrrrrq_s                        &  rq allow_tokencCs8t}|dks Jdtj||dtjjusJdS)Nr z8Allowing PID 1 on ZeroGPU will end up killing your Spacerpid)osgetpidrallowrr r%rrrrrs rF)fail allow_404rrcCstj||d}|tjjur)z tdtddWn ty Ynwt tt dS|tjj ur8|s6t ddStj |r@dSt d|dtj|d) N)rrrUrVrWz+ZeroGPU API /release warning: 404 Not FoundzZeroGPU API /release error: rhrT)rrrr NO_CONTENTrUNUSED_MESSAGEAttributeErrorrzr{RuntimeWarningr! is_successr$r)rrrrrrrrs&      rcCsH|duriSt|dd}|dust|dstddt|ds"|j}|S)Nr__dict__rgzInternal Gradio errorrp)getattrhasattrrr)rOrrrrrn s   rnrcCsX|t}durdifSz t|}W||fSty+td|ifYSw)Nz#Error while decoding X-IP-Token JWT)rp TOKEN_HEADERlowerr jwt_payload Exceptionrzr{)rrZrrrrros   ro)NNNT)6__doc__rr"rzdatetimertypingrgradiorjr packagingrtyping_extensionsrconfigrr=r apir r r rrrrrrrrurrrrwrDrBrErCrrr3r4booltuplerMintRequestrqrrdictrnrorrrrs                      2