Update app.py

app.py

@@ -1,19 +1,81 @@
-import os
-from config import SheamiConfig
+import asyncio
+import json
+import gradio as gr
+from huggingface_hub import whoami
+
+from common import get_db
 from home import build_home_page
-from login import start_server
-from ui import build
-
-if __name__ == "__main__":
-    # build(
-    #     user_email="petersenclarence@example.org", patient_id="68a4265e04b98fb066f75f78"
-    # ).queue().launch(favicon_path=SheamiConfig.favicon_path)
-    print("✅ App built, launching on port", os.environ.get("PORT", 7860))
-    os.makedirs(SheamiConfig._output_dir, exist_ok=True)
-
-    # build_home_page().queue().launch(
-    #     # favicon_path=SheamiConfig.favicon_path,
-    #     server_name="0.0.0.0",
-    #     server_port=int(os.environ.get("PORT", 7860)),
-    # )
-    start_server()
+from modules.models import SheamiUser
+from ui import (
+    get_app_theme,
+    get_app_title,
+    get_css,
+    render_logo,
+)  # Optional: to show retrieving org info
+
+
+def get_logged_in_user_name(profile: gr.OAuthProfile | None) -> str:
+    if profile is None:
+        return None  
+    return f"{profile.name}"
+
+
+def get_sheami_user(profile: gr.OAuthProfile | None):
+    if profile is None:
+        return None
+    return SheamiUser(
+        email=f"{profile.username}",
+        name=f"{profile.name}",
+        picture_url=f"{profile.picture}",
+    )
+
+
+def list_organizations(oauth_token: gr.OAuthToken | None) -> 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)}."
+
+
+def get_user_from_request(profile: gr.OAuthProfile):
+    print("profile.username = ", profile.username)
+    return profile.username or None
+
+
+async def build_securely():
+    with gr.Blocks(
+        title=get_app_title(), theme=get_app_theme(), css=get_css(), fill_height=True
+    ) as demo:
+        # Top menu bar with logo and login/logout buttons
+        with gr.Row():
+            with gr.Column(scale=5):
+                render_logo()
+            with gr.Column(scale=1):
+                gr.LoginButton()
+                logged_in_user_name = gr.Markdown()
+                logged_in_user_id = gr.State()
+                m2 = gr.Markdown()
+
+        @gr.render(inputs=logged_in_user_id)
+        def render_home_page(user: SheamiUser | None):
+            if user:
+                asyncio.run(register_user(logged_in_user=user))
+                build_home_page(logged_in_user=user)
+            else:
+                pass
+
+        demo.load(get_logged_in_user_name, inputs=None, outputs=logged_in_user_name)
+        demo.load(get_sheami_user, inputs=None, outputs=logged_in_user_id)
+        # demo.load(list_organizations, inputs=None, outputs=m2)
+
+    return demo
+
+
+async def register_user(logged_in_user: SheamiUser):
+    user = await get_db().get_user_by_email(email=logged_in_user.email)
+    if not user:
+        # register the user
+        await get_db().add_user(email=logged_in_user.email, name=logged_in_user.name)
+
+demo  = asyncio.run(build_securely())
+demo.queue().launch()