Upload connect_huggingface.py with huggingface_hub

connect_huggingface.py

@@ -0,0 +1,88 @@
+from dotenv import load_dotenv
+import os
+from huggingface_hub import login, HfApi
+import sys
+
+def setup_huggingface():
+    """Configure l'authentification Hugging Face."""
+    try:
+        print("=== Début de la configuration Hugging Face ===")
+        print(f"Python version: {sys.version}")
+        print(f"Working directory: {os.getcwd()}")
+        
+        # Charger les variables d'environnement depuis .env
+        load_dotenv(verbose=True, override=True)
+        print("Variables d'environnement chargées depuis .env")
+        
+        # Récupérer le token depuis les variables d'environnement
+        print("\nRecherche du token...")
+        
+        # Liste des noms possibles pour le token
+        token_names = ["HF_TOKEN", "HUGGINGFACE_API_TOKEN", "HUGGINGFACE_TOKEN"]
+        
+        # Afficher toutes les variables d'environnement liées à Hugging Face
+        env_vars = {k: v if k != "HF_TOKEN" else f"{v[:4]}...{v[-4:]}" for k, v in os.environ.items() 
+                   if k.startswith("HF") or "HUGGING" in k}
+        print("\nVariables d'environnement disponibles :")
+        for k, v in env_vars.items():
+            print(f"- {k}: {v}")
+        
+        # Essayer de récupérer le token
+        hf_token = None
+        for name in token_names:
+            value = os.environ.get(name)
+            if value:
+                print(f"\nToken trouvé dans {name}")
+                if len(value.strip()) > 0:
+                    hf_token = value.strip()
+                    break
+                else:
+                    print(f"Token trouvé dans {name} mais vide")
+            else:
+                print(f"Token non trouvé dans {name}")
+        
+        if not hf_token:
+            print("\nErreur : Token Hugging Face non trouvé ou vide dans les variables d'environnement")
+            return False
+            
+        print(f"\nToken trouvé (longueur: {len(hf_token)})")
+        print(f"Début du token : {hf_token[:4]}...")
+        print(f"Fin du token : ...{hf_token[-4:]}")
+            
+        # Configurer le token comme HF_TOKEN
+        os.environ["HF_TOKEN"] = hf_token
+        print("Token configuré comme HF_TOKEN")
+        
+        # Se connecter à Hugging Face
+        print("\nTentative de connexion à Hugging Face...")
+        try:
+            login(token=hf_token, add_to_git_credential=True)
+            print("Login réussi")
+        except Exception as e:
+            print(f"Erreur lors du login : {str(e)}")
+            raise
+        
+        # Vérifier la connexion
+        print("\nVérification de la connexion...")
+        try:
+            api = HfApi()
+            user_info = api.whoami()
+            print(f"Connecté avec succès en tant que : {user_info['name']}")
+            print(f"Type de compte : {user_info.get('account_type', 'inconnu')}")
+            print(f"Organisations : {', '.join(user_info.get('orgs', []))}")
+        except Exception as e:
+            print(f"Erreur lors de la vérification de la connexion : {str(e)}")
+            raise
+        
+        print("\n=== Configuration Hugging Face terminée avec succès ===")
+        return True
+        
+    except Exception as e:
+        print(f"\nErreur lors de la configuration Hugging Face : {str(e)}")
+        import traceback
+        print("\nTrace complète :")
+        print(traceback.format_exc())
+        return False
+
+if __name__ == "__main__":
+    setup_huggingface()