;3h5ddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl Z ddl mZddlmZmZddlmZej(j+dZej(j+dZej(j+d Zej(j+d Zd Zdd Zdd ZddZddZ d ddZe Gddej@Z!e GddZ"ddZ#y)) annotationsN) dataclassfield)RedirectResponse) get_tokenwhoami) get_spaceOAUTH_CLIENT_IDOAUTH_CLIENT_SECRET OAUTH_SCOPESOPENID_PROVIDER_URLc0 ddlm}t t |n t |t xsddz}|j|tj|jjddy#t$r}td|d}~wwxYw) 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.z-v4noneT) secret_key same_site https_only) starlette.middleware.sessionsr ImportErrorr _add_oauth_routes_add_mocked_oauth_routesr add_middlewarehashlibsha256encode hexdigest)appresession_secrets L/var/www/html/audio-gradio/venv/lib/python3.12/site-packages/gradio/oauth.py attach_oauthr$sC{# % */R58N>>."7"7"9:DDF +  k  sA;; B BBc^ ddlmddlm}d}t t |jdtt |jdtt |jd tt |jd |jd t td titd z|jddfd }|jddfd }|jddd}y#t$r}t d|d}~wwxYw)zIAdd OAuth routes to the FastAPI app (login, callback handler and logout).r)MismatchingStateError)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.r r r r huggingfacescopez!/.well-known/openid-configuration)name client_id client_secret client_kwargsserver_metadata_url/login/huggingfacecnKt|}jj||d{S7w)z)Endpoint that redirects to HF OAuth page.N)_generate_redirect_urir(authorize_redirect)request redirect_urioauths r# oauth_loginz&_add_oauth_routes..oauth_login^s2.g6 &&99'<PPPPs +535/login/callbackcK jj|d{}||jd <t'|S7# $r-t|jj D]/}|j ds|jj |1t|jjdd}|jjd}d|dzi}|r||d<dtjj|}|tkDrQtjjd}| t!d dd |j#d z}t%||zcYSt%|cYSwxYww) )Endpoint that handles the OAuth callback.N_state_huggingface _nb_redirectsr _target_urlz/login/huggingface? SPACE_HOSTzrGradio is not running in a Space (SPACE_HOST environment variable is not set). Cannot redirect to non-iframe view.https:/// oauth_info)r(authorize_access_tokenlistsessionkeys startswithpopint query_paramsgeturllibparse urlencode MAX_REDIRECTSosenviron RuntimeErrorrstripr_redirect_to_target) r3rAkey nb_redirects target_urlrI login_urihosthost_urlr&r5s r#oauth_redirect_callbackz2_add_oauth_routes..oauth_redirect_callbackest' /$00GGPPJR)3 %"7++UQ$% /GOO0023 ->>"67OO'', - w3377KLL --11-@J3B.oauth_logout' L$/"7++(*r3fastapi.Requestr3rdreturnr) 'authlib.integrations.base_client.errorsr&%authlib.integrations.starlette_clientr'rr ValueErrorformatr r r registerrJ) r r'r!msgr6rZr`r&r5s @@r#rr9s@Q? ; $5677"$9:;;N344"$9:;; GE NN !) -/2UU  WW !"Q#Q  WW -, -,^ WWY,,q  k  s D D, D''D,ctjdt|jdd d}|jdd fd }|jdd d}y) 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 your profile. To make it work, your machine must be logged in to Huggingface.r/czKt|}tdtjj d|izSw)z.Fake endpoint that redirects to HF OAuth page.z/login/callback?r<)r1rrKrLrM)r3r4s r#r6z-_add_mocked_oauth_routes..oauth_logins;.g6  !7!7 8U!V V  s9;r7c@K|jd<t|Sw)r9rA)rDrS)r3mocked_oauth_infos r#rZz9_add_mocked_oauth_routes..oauth_redirect_callbacks"): %"7++sr[cXK|jjddt|Swr]r^r_s r#r`z._add_mocked_oauth_routes..oauth_logoutrarbNrcre)warningswarn_get_mocked_oauth_inforJ)r r6rZr`rps @r#rrs{  MM T /0 WW !" #  WW , ,  WWY,,cLd|jvr|jd}n,dtjj|jz}|j dj |}t |}|jjdr|jdd}|S)Nr<z/?rZ)r<z .hf.spacezhttp://r?) rIrKrLrMurl_forinclude_query_paramsstrnetlocendswithreplace)r3targetr4redirect_uri_as_strs r#r1r1s,,,%%m4 ..w/C/CDD??#<=RRSLl+##K0199)ZP rucP|jjd|}t|S)Nr<)rIrJr)r3default_targetr}s r#rSrSs' ! ! % %m^ DF F ##ruceZdZUdZedZded<edZded<edZded<edZ ded<d d Z y ) 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. Attributes: name (str): The name of the user (e.g. 'Abubakar Abid'). username (str): The username of the user (e.g. 'abidlabs') profile (str): The profile URL of the user (e.g. 'https://huggingface.co/abidlabs'). picture (str): The profile picture URL of the user. 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.Markdown().attach_load_event(hello, None) F)initryr*usernameprofilepicturecv|j||d|_|d|_|d|_|d|_y)Nr*preferred_usernamerr)updater*rrr)selfdatas r#__init__zOAuthProfile.__init__s< DL 12 I I ruN)rdict) __name__ __module__ __qualname____doc__rr*__annotations__rrrrrur#rrsJ:5!D#!u%Hc%e$GS$e$GS$'rurc0eZdZUdZded<ded<ded<y) OAuthTokenak A Gradio OAuthToken object that can be used to inject the access token of a user in a function. If a function expects `OAuthToken` or `Optional[OAuthToken]` 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 `OAuthToken`, an error will be raised. Attributes: token (str): The access token of the user. scope (str): The scope of the access token. expires_at (int): The expiration timestamp of the access token. Example: import gradio as gr from typing import Optional from huggingface_hub import whoami def list_organizations(oauth_token: Optional[gr.OAuthToken]) -> str: if oauth_token is None: return "Please log in to list organizations." org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]] return f"You belong to {', '.join(org_names)}." with gr.Blocks() as demo: gr.LoginButton() gr.Markdown().attach_load_event(list_organizations, None) rytokenr)rH expires_atN)rrrrrrrur#rrs< J JOrurct}| tdt}|ddk7r td|ddddtt jd zd |d |d d |d |ddddddddd dS)Na Your machine must be logged in to HF to debug a Gradio app locally. Please run `huggingface-cli login` or set `HF_TOKEN` as environment variable with one of your access token. You can generate a new token in your settings page (https://huggingface.co/settings/tokens).typeuserzYour machine is not logged in with a personal account. Please use a personal access token. You can generate a new token in your settings page (https://huggingface.co/settings/tokens).beareriAAAAAAAAAAAAAAAAAAAAAAAAAAzopenid profileip11111111111111111111111fullnamer*zhttps://huggingface.co/ avatarUrlrz$00000000-0000-0000-0000-000000000000i daaaaaaaaaaaaaaaaaaaidzhttps://huggingface.co) subr*rrrwebsiteaud auth_timenonceiatexpiss) access_token token_type expires_inid_tokenr)ruserinfo)rrirrHtime)rrs r#rtrt*s KE } F   8D F|v 9  0!$))+&4,$"&v,0f?K(9#*+   ru)r fastapi.FastAPI)r rrfNone)r3rdrfry)r@)r3rdrryrfr)rfz typing.Dict)$ __future__rrrOrtyping urllib.parserKrr dataclassesrrfastapifastapi.responsesrhuggingface_hubrr gradio.utilsr rPrJr r r r rNr$rrr1rSDictrrrtrrur#rs" (.-"**..!23jjnn%:;zz~~n- jjnn%:; @`,F!,H&58$ $.1$$ ('6;;(' ('V !! !H'ru