login

README.md

@@ -7,6 +7,16 @@ sdk: gradio
 app_file: app.py
 pinned: false
 license: apache-2.0
+
+hf_oauth: true
+# optional, default duration is 8 hours/480 minutes. Max duration is 30 days/43200 minutes.
+hf_oauth_expiration_minutes: 480
+# optional, see "Scopes" below. "openid profile" is always included.
+hf_oauth_scopes:
+ - read-repos
+ - write-repos
+ - manage-repos
+ - inference-api
 ---
 
 An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).

app.py

@@ -1,25 +1,41 @@
 import gradio as gr
-from transformers import HfApi
+from huggingface_hub import list_models
+
+
+def hello(profile: gr.OAuthProfile | None) -> str:
+    # ^ expect a gr.OAuthProfile object as input to get the user's profile
+    # if the user is not logged in, profile will be None
+    if profile is None:
+        return "I don't know you."
+    return f"Hello {profile.name}"
+
+
+def list_private_models(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
+    # ^ expect a gr.OAuthToken object as input to get the user's token
+    # if the user is not logged in, oauth_token will be None
+    if oauth_token is None:
+        return "Please log in to list private models."
+    models = [
+        f"{model.id} ({'private' if model.private else 'public'})"
+        for model in list_models(author=profile.username, token=oauth_token.token)
+    ]
+    return "Models:\n\n" + "\n - ".join(models) + "."
 
-# Hugging Faceにログインする関数
-def login_to_huggingface(token):
-    api = HfApi()
-    try:
-        # トークンを使用してログイン
-        api.set_access_token(token)
-        user = api.whoami()
-        return f"ログイン成功: {user['name']}"
-    except Exception as e:
-        return f"ログイン失敗: {str(e)}"
 
-# Gradioインターフェースの作成
 with gr.Blocks() as demo:
-    gr.Markdown("# Hugging Face ログイン")
-    token = gr.Textbox(label="Hugging Face トークン", placeholder="ここにトークンを入力")
-    output = gr.Textbox(label="ログイン状態")
-    submit_button = gr.Button("ログイン")
-    
-    submit_button.click(login_to_huggingface, inputs=token, outputs=output)
+    gr.Markdown(
+        "# Gradio OAuth Space"
+        "\n\nThis Space is a demo for the **Sign in with Hugging Face** feature. "
+        "Duplicate this Space to get started."
+        "\n\nFor more details, check out:"
+        "\n- https://www.gradio.app/guides/sharing-your-app#o-auth-login-via-hugging-face"
+        "\n- https://huggingface.co/docs/hub/spaces-oauth"
+    )
+    gr.LoginButton()
+    # ^ add a login button to the Space
+    m1 = gr.Markdown()
+    m2 = gr.Markdown()
+    demo.load(hello, inputs=None, outputs=m1)
+    demo.load(list_private_models, inputs=None, outputs=m2)
 
-# アプリを起動
-demo.launch()
+demo.launch()