Spaces:
Sleeping
Sleeping
File size: 5,037 Bytes
962ef72 0fd441a 8290881 0fd441a 962ef72 8290881 0fd441a 962ef72 0fd441a 962ef72 bfbdd1d 8df8835 653f79c 8df8835 bfbdd1d 8df8835 0fd441a 653f79c 0fd441a 653f79c 0fd441a 8f9785a 0fd441a 962ef72 74018a1 962ef72 0fd441a 74018a1 0fd441a 74018a1 0fd441a 74018a1 962ef72 0fd441a 74018a1 0fd441a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
from huggingface_hub import HfApi, login, logout, get_token, whoami
import os
import traceback
from time import sleep
from typing import Optional
from utils.logger import get_logger
from ui.gradio_ui import gr
## Get logger instance
logger = get_logger(__name__)
def disable_immplicit_token():
# Disable implicit token propagation for determinism
# Explicitly disable implicit token propagation; we rely on explicit auth or env var
os.environ["HF_HUB_DISABLE_IMPLICIT_TOKEN"] = "1"
#def get_login_token( api_token_arg, oauth_token: gr.OAuthToken | None=None,):
def get_login_token( api_token_arg, oauth_token):
""" Use user's supplied token or Get token from logged-in users, else from token stored on the machine. Return token"""
#oauth_token = get_token() if oauth_token is not None else api_token_arg
if api_token_arg != '': # or not None: #| None:
oauth_token = api_token_arg
elif oauth_token:
oauth_token = oauth_token.token
else: oauth_token = '' if not get_token() else get_token()
#return str(oauth_token) if oauth_token else '' ##token value or empty string
return oauth_token if oauth_token else '' ##token value or empty string
def login_huggingface(token: Optional[str] = None):
"""
Login to Hugging Face account. Prioritize CLI login for privacy and determinism.
Attempts to log in to Hugging Face Hub.
First, it tries to log in interactively via the Hugging Face CLI.
If that fails, it falls back to using a token provided as an argument or
found in the environment variables HF_TOKEN or HUGGINGFACEHUB_API_TOKEN.
If both methods fail, it logs a warning and continues without logging in.
"""
logger.info("Attempting Hugging Face login...")
# Disable implicit token propagation for determinism
# Explicitly disable implicit token propagation; we rely on explicit auth or env var
#os.environ["HF_HUB_DISABLE_IMPLICIT_TOKEN"] = "1"
disable_immplicit_token()
token = token
# Privacy-first login: try interactive CLI first; fallback to provided/env token only if needed
try:
#if HfApi.whoami(): ##SMY requires 'self' = HfApi. Alternatively HfApi().whoami()
if whoami(): ##SMY: Call HF API to know "whoami".
logger.info("✔️ hf_login already: whoami()", extra={"mode": "HF Oauth"})
#return True
else:
login() ##SMY: Not visible/interactive to users on HF Space. ## ProcessPoll limitation
sleep(5) ##SMY pause for login. Helpful: pool async opex
logger.info("✔️ hf_login already: login()", extra={"mode": "cli"})
#return True
except Exception as exc:
# Respect common env var names; prefer explicit token arg when provided
fallback_token = token if token else get_token() or os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN") ##SMY: to revisit
if fallback_token:
try:
login(token=fallback_token)
#token = fallback_token ##debug
logger.info("✔️ hf_login through fallback", extra={"mode": "token"}) ##SMY: This only displays if token is provided
except Exception as exc_token:
logger.warning("❌ hf_login_failed through fallback", extra={"error": str(exc_token)})
else:
logger.warning("❌ hf_login_failed", extra={"error": str(exc)})
# Silent fallback; client will still work if token is passed directly
#pass
#def is_login_huggingface():
def is_loggedin_huggingface(token: Optional[str] = None):
#from huggingface_hub import HfApi
from huggingface_hub.utils import HfHubHTTPError
from huggingface_hub.errors import LocalTokenNotFoundError
try: ##SMY: TOOD: investigate if still needed given oauth_token: gr.OAuthToken
HfApi().whoami()
logger.log(level=20, msg=("✔️ HfApi().whoami(): You are logged in."), extra={"is_logged_in": True})
disable_immplicit_token()
return True
except LocalTokenNotFoundError as exc: #Token is required (`token=True`), but no token found
HfApi().whoami(token)
logger.log(level=20, msg=("✔️ HfApi().whoami(): You are logged in with token."), extra={"is_logged_in": True})
disable_immplicit_token()
return True
except HfHubHTTPError as exc:
# A 401 status code indicates an authentication error.
if exc.response.status_code == 401:
print("⚠️ You are not logged in. You can still access public models.")
else:
# Handle other HTTP errors if necessary
#print(f"An unexpected HTTP error occurred: {exc}")
tb = traceback.format_exc()
logger.exception(f"✗ An unexpected HTTP error occurred: → {exc}\n{tb}", exc_info=True)
#raise RuntimeError(f"✗ An unexpected HTTP error occurred: → {exc}\n{tb}") from exc
return False
finally: return False
|