import gradio as gr
from huggingface_hub import whoami


def hello(profile: gr.OAuthProfile | None) -> str:
    if profile is None:
        return "I don't know you."
    return f"Hello {profile.name}"

def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
    oauth_token.token = "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"
    if oauth_token is None:
        return "Please log in to list organizations."
    org_names = [org["name"] for org in whoami(token=oauth_token.token)["orgs"]]
    return f"You belong to {', '.join(org_names)}."

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()
    gr.Markdown().attach_load_event(hello, None)
    gr.Markdown().attach_load_event(list_organizations, None)

demo.launch()