Update app.py

app.py

@@ -1,3 +1,4 @@
+import asyncio
 import uvicorn
 from fastapi import FastAPI, Depends
 from starlette.responses import RedirectResponse
@@ -9,6 +10,9 @@ from starlette.config import Config
 import gradio as gr
 from dotenv import load_dotenv
 
+from app import build_securely
+from ui import render_logo
+
 load_dotenv()
 app = FastAPI()
 
@@ -68,7 +72,11 @@ async def auth(request: Request):
     return RedirectResponse(url='/gradio')
 
 with gr.Blocks() as login_demo:
-    btn = gr.Button("Login")
+    render_logo()
+    with gr.Row():
+        gr.Column()
+        btn = gr.Button("Login", variant="huggingface")
+        gr.Column()
     _js_redirect = """
     () => {
         url = '/login' + window.location.search;
@@ -87,7 +95,9 @@ with gr.Blocks() as main_demo:
     gr.Button("Logout", link="/logout")
     main_demo.load(greet, None, m)
 
-app = gr.mount_gradio_app(app, main_demo, path="/gradio", auth_dependency=get_user)
+sheami_app = asyncio.run(build_securely())
+
+app = gr.mount_gradio_app(app, sheami_app, path="/gradio", auth_dependency=get_user)
 
 
 if __name__ == '__main__':