update app

app.py

@@ -5,6 +5,15 @@ def hello(profile: gr.OAuthProfile | None) -> str:
         return "I don't know you."
     return f"Hello {profile.name}"
 
+def is_user_in_organization(oauth_token: gr.OAuthToken | None, required_org: str) -> bool:
+    if oauth_token is None:
+        return False
+
+    user_info = whoami(token=oauth_token.token)
+    user_orgs = [org["name"] for org in user_info["orgs"]]
+
+    return required_org in user_orgs
+
 def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
         return "Please log in to list organizations."
@@ -15,11 +24,24 @@ with gr.Blocks() as demo:
     gr.Markdown(
         "# Gradio OAuth Space\n\nThis Space is a demo for the new **Sign in with Hugging Face** feature."
     )
-    gr.LoginButton()
-    gr.LogoutButton()
-    m1 = gr.Markdown()
-    m2 = gr.Markdown()
-    demo.load(hello, inputs=None, outputs=m1)
-    demo.load(list_organizations, inputs=None, outputs=m2)
+    with gr.Row():
+        gr.LoginButton()
+        gr.LogoutButton()
+
+    with gr.Row():
+        m1 = gr.Markdown()
+        m2 = gr.Markdown()
+
+    required_org = "YourOrganizationName"
+
+    def update_interface(profile: gr.OAuthProfile | None):
+        if profile and is_user_in_organization(profile.token, required_org):
+            m1.update(f"Hello {profile.name}, welcome to the app.")
+            # Load other components or functions relevant to authenticated users
+        else:
+            m1.update("Access denied. You do not belong to the required organization.")
+
+
+    demo.load(update_interface, inputs=None, outputs=None)
 
 demo.launch()