from huggingface_hub import HfApi
import csv
import os
from pathlib import Path

# 🔐 Configuración segura del token
HF_TOKEN = os.getenv("HF_TOKEN")
if not HF_TOKEN:
    raise ValueError("⚠️ Por favor, configura la variable de entorno HF_TOKEN")

# 🗂️ Configurar repositorio
api = HfApi()
repo_id = "Segizu/facial-recognition"

try:
    # ✅ Listar todos los archivos del dataset
    files = api.list_repo_files(repo_id=repo_id, repo_type="dataset", token=HF_TOKEN)
    
    # Filtrar imágenes .jpg
    image_files = [f for f in files if f.lower().endswith(".jpg")]
    
    # Guardar metadata.csv
    metadata_path = Path("metadata.csv")
    with open(metadata_path, "w", newline="") as f:
        writer = csv.writer(f)
        writer.writerow(["image"])
        for img in image_files:
            writer.writerow([img])
    
    print(f"✅ metadata.csv generado con {len(image_files)} imágenes.")

except Exception as e:
    print(f"❌ Error: {str(e)}")
    if "401" in str(e):
        print("⚠️ Error de autenticación. Verifica que tu token de Hugging Face sea válido.")
    elif "404" in str(e):
        print("⚠️ No se encontró el repositorio. Verifica que el nombre del repositorio sea correcto.")
    else:
        print("⚠️ Ocurrió un error inesperado.")