Fix: allow fine grained write-access tokens (#68)

* Fix: allow fine grained write-access tokens

Refactor the logic in deploy_as_space to attempt reading info or
creation of a space, and handling the exception to present a login
screen. This way the server can handle the authenticity of the token and
we don't need logic to validate it.

* Check both 401 and 403 in both places

* simplify

* retry after acquiring token

---------

Co-authored-by: Abubakar Abid <abubakar@huggingface.co>

Author: znation

trackio/deploy.py

@@ -9,6 +9,7 @@
 from gradio_client import Client, handle_file
 from httpx import ReadTimeout
 from huggingface_hub.errors import RepositoryNotFoundError
+from requests import HTTPError
 
 from trackio.sqlite_storage import SQLiteStorage
 
@@ -27,25 +28,26 @@ def deploy_as_space(
     trackio_path = files("trackio")
 
     hf_api = huggingface_hub.HfApi()
-    whoami = None
-    login = False
+
     try:
-        whoami = hf_api.whoami()
-        if whoami["auth"]["accessToken"]["role"] != "write":
-            login = True
-    except OSError:
-        login = True
-    if login:
-        print("Need 'write' access token to create a Spaces repo.")
-        huggingface_hub.login(add_to_git_credential=False)
-        whoami = hf_api.whoami()
-
-    huggingface_hub.create_repo(
-        space_id,
-        space_sdk="gradio",
-        repo_type="space",
-        exist_ok=True,
-    )
+        huggingface_hub.create_repo(
+            space_id,
+            space_sdk="gradio",
+            repo_type="space",
+            exist_ok=True,
+        )
+    except HTTPError as e:
+        if e.response.status_code in [401, 403]:  # unauthorized or forbidden
+            print("Need 'write' access token to create a Spaces repo.")
+            huggingface_hub.login(add_to_git_credential=False)
+            huggingface_hub.create_repo(
+                space_id,
+                space_sdk="gradio",
+                repo_type="space",
+                exist_ok=True,
+            )
+        else:
+            raise ValueError(f"Failed to create Space: {e}")
 
     with open(Path(trackio_path, "README.md"), "r") as f:
         readme_content = f.read()
@@ -102,6 +104,15 @@ def create_space_if_not_exists(
         return
     except RepositoryNotFoundError:
         pass
+    except HTTPError as e:
+        if e.response.status_code in [401, 403]:  # unauthorized or forbidden
+            print("Need 'write' access token to create a Spaces repo.")
+            huggingface_hub.login(add_to_git_credential=False)
+            huggingface_hub.add_space_variable(
+                space_id, "TRACKIO_DATASET_ID", dataset_id
+            )
+        else:
+            raise ValueError(f"Failed to create Space: {e}")
 
     print(f"* Creating new space: {SPACE_URL.format(space_id=space_id)}")
     deploy_as_space(space_id, dataset_id)