a dx@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z ej dZ ej dZej dZej d Zd d d d Zd ddddZd ddddZGdddejZddddddddddd d!d"d#d$d#dd%d& d'ZdS)() annotationsNRedirectResponse) get_spaceOAUTH_CLIENT_IDOAUTH_CLIENT_SECRET OAUTH_SCOPESOPENID_PROVIDER_URLzfastapi.FastAPI)appc Cszddlm}Wn.ty>}ztd|WYd}~n d}~00tdurTt|nt||j|tt pld  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 erU/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/gradio/oauth.py attach_oauths"  rNone)r returnc szddlm}Wn.ty>}ztd|WYd}~n d}~00d}tdurZt|dtdurpt|dtdurt|dtdurt|d |j d ttd titd d | dddfdd }| ddddfdd }| dddddd}dS)zIAdd OAuth routes to the FastAPI app (login, callback handler and logout).r)OAuthr NzOAuth 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.rrr r huggingfacescopez!/.well-known/openid-configuration)nameZ client_idZ client_secretZ client_kwargsZserver_metadata_url/login/huggingfacefastapi.Requestrequestcs6t|d}d|vr"|dd}j||IdHS)z)Endpoint that redirects to HF OAuth page.oauth_redirect_callbackz .hf.spacezhttp://zhttps://N)strZurl_forreplacerZauthorize_redirect)r$Z redirect_uriZoauthrr oauth_loginRs z&_add_oauth_routes..oauth_login/login/callbackrr$rcs2j|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/logoutcs$|jdd|jddtdSz=Endpoint that logs out the user (e.g. delete cookie session).r.Nr/r0r1poprr#rrr oauth_logoutcsz'_add_oauth_routes..oauth_logout) Z%authlib.integrations.starlette_clientrrr ValueErrorformatrr r registerget)r rrmsgr)r%r7rr(rr/s@ rcCsZtd|ddddd}|dddd d d }|d ddd d 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#cstdS)z.Fake endpoint that redirects to HF OAuth page.r*rr#rrrr)wsz-_add_mocked_oauth_routes..oauth_loginr*rr+cs td|jd<t|jd<tdS)r,r-r.r/r0)MOCKED_OAUTH_TOKENr1rr#rrrr%|s z9_add_mocked_oauth_routes..oauth_redirect_callbackr3cs$|jdd|jddtdSr4r5r#rrrr7sz._add_mocked_oauth_routes..oauth_logoutN)warningswarnr;)r r)r%r7rrrrksrc@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__rrrrr@sr@Z#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_tokenrZ expires_atr-) __future__rrostypingr>ZfastapiZfastapi.responsesrutilsrenvironr;rrr r rrrDictr@r=rrrrsF       <