#!/usr/bin/env python3
"""
Test simplifié de connexion HuggingFace sans numpy
"""

import os

def test_hf_connection():
    print("🧪 Test de connexion simple au dataset HuggingFace")
    
    dataset_id = "HackathonCRA/2024"
    hf_token = os.environ.get("HF_TOKEN")
    
    print(f"📦 Dataset: {dataset_id}")
    print(f"🔑 Token configuré: {'✅ Oui' if hf_token else '❌ Non'}")
    
    try:
        # Import ici pour éviter les conflits
        from huggingface_hub import HfApi
        
        api = HfApi(token=hf_token)
        
        # Test d'accès au dataset
        print("🔄 Test d'accès au dataset...")
        dataset_info = api.dataset_info(dataset_id)
        
        print(f"✅ Dataset accessible !")
        print(f"📝 Description: {dataset_info.description or 'Pas de description'}")
        print(f"🏷️ Tags: {dataset_info.tags}")
        print(f"📊 Dernière modification: {dataset_info.last_modified}")
        
        # Lister les fichiers
        files = api.list_repo_files(dataset_id, repo_type="dataset")
        csv_files = [f for f in files if f.endswith('.csv')]
        
        print(f"📄 Fichiers CSV trouvés ({len(csv_files)}):")
        for f in csv_files[:5]:  # Afficher les 5 premiers
            print(f"  - {f}")
        if len(csv_files) > 5:
            print(f"  ... et {len(csv_files) - 5} autres")
        
        return True
        
    except Exception as e:
        print(f"❌ Erreur: {e}")
        print()
        print("💡 Solutions possibles:")
        print("1. Vérifiez que le dataset existe: https://huggingface.co/datasets/HackathonCRA/2024")
        print("2. Configurez votre token: export HF_TOKEN='votre_token'")
        print("3. Vérifiez vos permissions d'accès")
        return False

if __name__ == "__main__":
    test_hf_connection()