l0jxddlmZddlZddlZddlZddlZddlZddlmZddl m Z ej dZ ej dZej dZej d Zd%d Zd&dZd&dZGddejZdddddddddddddd d!d dd"d# d$ZdS)') annotationsNRedirectResponse) get_spaceOAUTH_CLIENT_IDOAUTH_CLIENT_SECRET OAUTH_SCOPESOPENID_PROVIDER_URLappfastapi.FastAPIc` ddlm}n"#t$r}td|d}~wwxYwtt |nt |||tjtpd  dddS)Nr)SessionMiddlewareCannot initialize OAuth to due a missing library. Please run `pip install gradio[oauth]` or add `gradio[oauth]` to your requirements.txt file in order to install the required dependencies.noneT) secret_key same_site https_only) starlette.middleware.sessionsr ImportErrorr_add_oauth_routes_add_mocked_oauth_routesadd_middlewarehashlibsha256r encode hexdigest)r res O/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/gradio/oauth.py attach_oauthr!sCCCCCCC  k   {{# %%% >#6#<""D"D"F"FGGQQSS s (#(returnNonec ddlm}n"#t$r}td|d}~wwxYwd}t"t |dt "t |dt"t |dt"t |d | d tt d titd z | ddfd }| ddfd }| ddd}dS)zIAdd OAuth routes to the FastAPI app (login, callback handler and logout).r)OAuthrNzOAuth is required but {} environment variable is not set. Make sure you've enabled OAuth in your Space by setting `hf_oauth: true` in the Space metadata.rr r r huggingfacescopez!/.well-known/openid-configuration)name client_id client_secret client_kwargsserver_metadata_url/login/huggingfacerequestfastapi.RequestcKt|d}d|vr|dd}j||d{VS)z)Endpoint that redirects to HF OAuth page.oauth_redirect_callbackz .hf.spacezhttp://zhttps://N)strurl_forreplacer&authorize_redirect)r. redirect_urioauths r oauth_loginz&_add_oauth_routes..oauth_loginRsn7??+DEEFF , & &'// :FFL&99'<PPPPPPPPP/login/callbackr"rcKj|d{V}|d|jd<||jd<tdS))Endpoint that handles the OAuth callback.Nuserinfo oauth_profile oauth_token/)r&authorize_access_tokensessionr)r.tokenr7s r r1z2_add_oauth_routes..oauth_redirect_callback[s\'>>wGGGGGGGG+0+<(). &$$$r9/logoutcK|jdd|jddtdSz=Endpoint that logs out the user (e.g. delete cookie session).r>Nr?r@rBpoprr.s r oauth_logoutz'_add_oauth_routes..oauth_logoutcF OT222M4000$$$r9r.r/r.r/r"r) %authlib.integrations.starlette_clientr%rr ValueErrorformatr r r registerget)r r%rmsgr8r1rJr7s @r rr/s???????  k    ;$566777"$9::;;;N33444"$9::;;; EGGE NN !) -/2UU  WW !""QQQQQ#"Q WW %%%%% % WWY%%%%%%s )$)ctjd|dd d}|ddd }|d dd }d S)aAdd fake oauth routes if Gradio is run locally and OAuth is enabled. Clicking on a gr.LoginButton will have the same behavior as in a Space (i.e. gets redirected in a new tab) but instead of authenticating with HF, a mocked user profile is added to the session. zGradio does not support OAuth features outside of a Space environment. To help you debug your app locally, the login and logout buttons are mocked with a fake user profile.r-r.r/c$KtdS)z.Fake endpoint that redirects to HF OAuth page.r:rrIs r r8z-_add_mocked_oauth_routes..oauth_loginws 1222r9r:r"rclKtd|jd<t|jd<tdS)r<r=r>r?r@)MOCKED_OAUTH_TOKENrBrrIs r r1z9_add_mocked_oauth_routes..oauth_redirect_callback|s3,>j+I(); &$$$r9rDcK|jdd|jddtdSrFrGrIs r rJz._add_mocked_oauth_routes..oauth_logoutrKr9NrLrM)warningswarnrR)r r8r1rJs r rrks  M p  WW !""333#"3 WW %%% %  WWY%%%%%%r9ceZdZdZdS) OAuthProfilea  A Gradio OAuthProfile object that can be used to inject the profile of a user in a function. If a function expects `OAuthProfile` or `Optional[OAuthProfile]` as input, the value will be injected from the FastAPI session if the user is logged in. If the user is not logged in and the function expects `OAuthProfile`, an error will be raised. Example: import gradio as gr from typing import Optional def hello(profile: Optional[gr.OAuthProfile]) -> str: if profile is None: return "I don't know you." return f"Hello {profile.name}" with gr.Blocks() as demo: gr.LoginButton() gr.LogoutButton() gr.Markdown().attach_load_event(hello, None) N)__name__ __module__ __qualname____doc__r9r r\r\sr9r\#hf_oauth_AAAAAAAAAAAAAAAAAAAAAAAAAAbeareriAAAAAAAAAAAAAAAAAAAAAAAAAAzopenid profileid11111111111111111111111zFake Gradio UserFakeGradioUserz%https://huggingface.co/FakeGradioUserzAhttps://huggingface.co/front/assets/huggingface_logo-noborder.svgrz$00000000-0000-0000-0000-000000000000i daaaaaaaaaaaaaaaaaaazhttps://huggingface.co) subr(preferred_usernameprofilepicturewebsiteaud auth_timenonceiatexpiss) access_token token_type expires_inid_tokenr' expires_atr=)r r )r r r"r#) __future__rrostypingrYfastapifastapi.responsesrutilsrenvironrRrr r r r!rrDictr\rWrar9r rsx"""""" ......*..!233jnn%:;;z~~n-- jnn%:;;89%9%9%9%x%%%%@6;6:, (".:V5&'  r9