Update app.py

app.py

@@ -2,19 +2,18 @@ import gradio as gr
 from typing import Optional
 from huggingface_hub import HfApi
 
-def grant_access_to_repo(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, repo_id, username):
+def grant_access_to_repo(token, repo_id, username):
     """
     Grant access to a user/organization for a specific repo
     """
     try:
-        gr.Info("SBM req: " + repo_id + "|" + username + "|" + oauth_token.token)
         api = HfApi()
-        api.grant_access(repo_id=repo_id, user=username, token=oauth_token.token)
+        api.grant_access(repo_id=repo_id, user=username, token=token)
         return f"✅ Successfully granted access to {username} for {repo_id}"
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-def grant_access_multiple_repos(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, repo_list, username):
+def grant_access_multiple_repos(token, repo_list, username):
     """
     Grant access to multiple repositories at once
     """
@@ -25,7 +24,7 @@ def grant_access_multiple_repos(profile: Optional[gr.OAuthProfile], oauth_token:
     
     for repo_id in repos:
         try:
-            api.grant_access(repo_id=repo_id, user=username, token=oauth_token.token)
+            api.grant_access(repo_id=repo_id, user=username, token=token)
             results.append(f"✅ {repo_id}: Success")
         except Exception as e:
             results.append(f"❌ {repo_id}: {str(e)}")
@@ -36,7 +35,13 @@ with gr.Blocks(title="HF Repo Access Manager") as demo:
     gr.Markdown("# 🔐 Hugging Face Repo Access Manager")
     gr.Markdown("Grant access to users/organizations for your repositories")
 
-    gr.LoginButton(elem_id="login")
+    hf_token = gr.Textbox(
+        label="Huggingface token",
+        placeholder="hf_xxxxxxxxxxxxxxxx",
+        type="password",
+        info="Your HF token with write permissions",
+        max_lines=1
+    )
     
     with gr.Tab("Single Repository"):
         with gr.Row():
@@ -77,13 +82,13 @@ with gr.Blocks(title="HF Repo Access Manager") as demo:
     # Connect the buttons
     single_btn.click(
         fn=grant_access_to_repo,
-        inputs=[repo_input, user_input],
+        inputs=[hf_token, repo_input, user_input],
         outputs=single_output
     )
     
     multi_btn.click(
         fn=grant_access_multiple_repos,
-        inputs=[multi_repo_input, multi_user_input],
+        inputs=[hf_token, multi_repo_input, multi_user_input],
         outputs=multi_output
     )