Kaggle support

+ Colab warnings

manage branches.py

@@ -61,17 +61,22 @@ if cord == 'create':
 clear_screen()
 
 #get token
+if os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None: #check if user in kaggle
+    from kaggle_secrets import UserSecretsClient
+    from kaggle_web_client import BackendError
+    try:
+        login(UserSecretsClient().get_secret("HF_TOKEN")) #login if token secret found
+    except BackendError: 
+        print('''
+            When using Kaggle, make sure to use the secret key HF_TOKEN with a 'WRITE' token.
+                   This will prevent the need to login every time you run the script.
+                   Set your secrets with the secrets add-on on the top of the screen.
+             ''')
 if get_token() is not None:
     #if the token is found then log in:
     login(get_token())
     tfound = "Where are my doritos?"
 else:
-    if os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None: #check if user in kaggle
-        print('''
-              When using Kaggle, make sure to use the secret key HF_TOKEN with a 'WRITE' token.
-              This will prevent the need to login every time you run the script.
-              Set your secrets with the secrets add-on on the top of the screen.
-              ''')
     #if the token is not found then prompt user to provide it:
     login(input("API token not detected. Enter your HuggingFace (WRITE) token: "))
     tfound = "false"
@@ -80,21 +85,26 @@ else:
 while True:
     if whoami().get('auth', {}).get('accessToken', {}).get('role', None) != 'write':
         clear_screen()
-        if os.environ.get('HF_TOKEN', None) is not None: #if environ finds HF_TOKEN as write then display following text and exit:
-            print(f'''
+        if os.environ.get('HF_TOKEN', None) is not None: #if environ finds HF_TOKEN as read-only then display following text and exit:
+            print('''
           You have the environment variable HF_TOKEN set.
           You cannot log in.
-          Either set the environment variable to a (WRITE) token or remove it.
-          ''')
+          Either set the environment variable to a 'WRITE' token or remove it.
+                  ''')
             input("Press enter to continue.")
             exit()
-            
-        if os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None: #check if user in kaggle
+        if os.environ.get('COLAB_BACKEND_VERSION', None) is not None:
             print('''
-                  When using Kaggle, make sure to use the secret key HF_TOKEN with a 'WRITE' token.
-                  This will prevent the need to login every time you run the script.
-                  Set your secrets with the secrets add-on on the top of the screen.
+                              Your Colab secret key is read-only
+                Please switch your key to 'write' or disable notebook access on the left.
+                               For now, you are stuck in a loop
                   ''')
+        elif os.environ.get('KAGGLE_KERNEL_RUN_TYPE', None) is not None:
+            print('''
+                                      Your Kaggle secret key is read-only
+                Please switch your key to 'write' or unattach from notebook in add-ons at the top.
+                          Having a read-only key attched will require login every time.
+                ''')
         print("You do not have write access to this repository. Please use a valid token with (WRITE) access.")
         login(input("Enter your HuggingFace (WRITE) token: "))
         continue
@@ -164,10 +174,10 @@ else:
 #if token wasn't found from line 60 then display following text:
 if tfound == 'false':
     print(f'''
-          You are now logged in as {whoami().get('fullname', None)}.
+              You are now logged in as {whoami().get('fullname', None)}.
           
           To logout, use the hf command line interface 'huggingface-cli logout'
-          To view your active account, use 'huggingface-cli whoami'
+               To view your active account, use 'huggingface-cli whoami'
           ''')
 
 input("Press enter to continue.")