Spaces:
Running
Running
oauth in iframe
Browse files
app.py
CHANGED
@@ -36,7 +36,8 @@ def show_profile(request: gr.Request) -> str:
|
|
36 |
|
37 |
def js_open(url: str) -> str:
|
38 |
# Taken from https://cmgdo.com/external-link-in-gradio-button/
|
39 |
-
return f"function() {{window.
|
|
|
40 |
|
41 |
|
42 |
with gr.Blocks() as demo:
|
|
|
36 |
|
37 |
def js_open(url: str) -> str:
|
38 |
# Taken from https://cmgdo.com/external-link-in-gradio-button/
|
39 |
+
return f"function() {{window.location.assign('{url}');}}"
|
40 |
+
# return f"function() {{window.open('{url}', '_blank');}}"
|
41 |
|
42 |
|
43 |
with gr.Blocks() as demo:
|
auth.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
from authlib.integrations.starlette_client import OAuth
|
4 |
from fastapi import FastAPI
|
5 |
from fastapi.requests import Request
|
6 |
-
from fastapi.responses import
|
7 |
from starlette.middleware.sessions import SessionMiddleware
|
8 |
|
9 |
|
@@ -30,7 +30,7 @@ oauth.register(
|
|
30 |
|
31 |
# Hack to close the login/logout page once the user is logged in/out.
|
32 |
# TODO: can it be less hacky?
|
33 |
-
CLOSE_WINDOW_HTML = HTMLResponse("<script>window.close();</script>")
|
34 |
|
35 |
|
36 |
async def oauth_login(request: Request):
|
@@ -40,15 +40,15 @@ async def oauth_login(request: Request):
|
|
40 |
return await oauth.huggingface.authorize_redirect(request, redirect_uri)
|
41 |
|
42 |
|
43 |
-
async def oauth_logout(request: Request) ->
|
44 |
request.session.pop("user", None)
|
45 |
-
return
|
46 |
|
47 |
|
48 |
-
async def oauth_redirect_callback(request: Request) ->
|
49 |
token = await oauth.huggingface.authorize_access_token(request)
|
50 |
request.session["user"] = token["userinfo"] # TODO: we should store entire token
|
51 |
-
return
|
52 |
|
53 |
|
54 |
def attach_oauth(app: FastAPI) -> None:
|
|
|
3 |
from authlib.integrations.starlette_client import OAuth
|
4 |
from fastapi import FastAPI
|
5 |
from fastapi.requests import Request
|
6 |
+
from fastapi.responses import RedirectResponse
|
7 |
from starlette.middleware.sessions import SessionMiddleware
|
8 |
|
9 |
|
|
|
30 |
|
31 |
# Hack to close the login/logout page once the user is logged in/out.
|
32 |
# TODO: can it be less hacky?
|
33 |
+
# CLOSE_WINDOW_HTML = HTMLResponse("<script>window.close();</script>")
|
34 |
|
35 |
|
36 |
async def oauth_login(request: Request):
|
|
|
40 |
return await oauth.huggingface.authorize_redirect(request, redirect_uri)
|
41 |
|
42 |
|
43 |
+
async def oauth_logout(request: Request) -> RedirectResponse:
|
44 |
request.session.pop("user", None)
|
45 |
+
return RedirectResponse("/")
|
46 |
|
47 |
|
48 |
+
async def oauth_redirect_callback(request: Request) -> RedirectResponse:
|
49 |
token = await oauth.huggingface.authorize_access_token(request)
|
50 |
request.session["user"] = token["userinfo"] # TODO: we should store entire token
|
51 |
+
return RedirectResponse("/")
|
52 |
|
53 |
|
54 |
def attach_oauth(app: FastAPI) -> None:
|