o ex@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z ej dZ ej dZej dZej d Zd(d d Zd)ddZd)ddZGdddejZddddddddddd d!d"d#d$d#dd%d& d'ZdS)*) annotationsNRedirectResponse) get_spaceOAUTH_CLIENT_IDOAUTH_CLIENT_SECRET OAUTH_SCOPESOPENID_PROVIDER_URLappfastapi.FastAPIc Csxzddlm}Wnty}ztd|d}~wwtdur$t|nt||j|tt p0d  ddddS)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)Z secret_keyZ same_siteZ https_only) Zstarlette.middleware.sessionsr ImportErrorr_add_oauth_routes_add_mocked_oauth_routesZadd_middlewarehashlibsha256rencode hexdigest)r r er.oauth_login/login/callbackrrcs4j|IdH}|d|jd<||jd<tdS))Endpoint that handles the OAuth callback.Nuserinfo oauth_profile oauth_token/)rZauthorize_access_tokensessionr)r#tokenr(rrr%[s  z2_add_oauth_routes..oauth_redirect_callback/logoutc&|jdd|jddtdSz=Endpoint that logs out the user (e.g. delete cookie session).r-Nr.r/r0poprr#rrr oauth_logoutcz'_add_oauth_routes..oauth_logoutr#r$r#r$rr) Z%authlib.integrations.starlette_clientrrr ValueErrorformatrr r registerget)r rrmsgr)r%r8rr(rr/sD rcCsJtd|dddd}|ddd d }|d dd d}dS)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$cs tdS)z.Fake endpoint that redirects to HF OAuth page.r*rr7rrrr)wsz-_add_mocked_oauth_routes..oauth_loginr*rrcs"td|jd<t|jd<tdS)r+r,r-r.r/)MOCKED_OAUTH_TOKENr0rr7rrrr%|s z9_add_mocked_oauth_routes..oauth_redirect_callbackr2cr3r4r5r7rrrr8r9z._add_mocked_oauth_routes..oauth_logoutNr:r;)warningswarnr?)r r)r%r8rrrrks  rc@seZdZdZdS) 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__rrrrrDsrDZ#hf_oauth_AAAAAAAAAAAAAAAAAAAAAAAAAAZbeareriZAAAAAAAAAAAAAAAAAAAAAAAAAAzopenid profileidZ11111111111111111111111zFake Gradio UserZFakeGradioUserz%https://huggingface.co/FakeGradioUserzAhttps://huggingface.co/front/assets/huggingface_logo-noborder.svgrz$00000000-0000-0000-0000-000000000000i dZaaaaaaaaaaaaaaaaaaazhttps://huggingface.co) subr!Zpreferred_usernameZprofileZpictureZwebsiteZaudZ auth_timenonceZiatexpZiss)Z access_token token_typeZ expires_inZid_tokenr Z expires_atr,)r r )r r rr) __future__rrostypingrBZfastapiZfastapi.responsesrutilsrenvironr?rrr r rrrDictrDrArrrrsH          <