import os
import base64
from huggingface_hub import HfApi, hf_hub_download

# Obtener el token de la variable de entorno y descodificarlo
token_base64 = os.getenv("TOKEN")
token = base64.b64decode(token_base64).decode("utf-8")

# Obtener el repo_id de la variable de entorno
repo_id = os.getenv("REPO_ID")

# Crear una instancia de HfApi para listar los archivos del repositorio
api = HfApi()
archivos = api.list_repo_files(repo_id=repo_id, token=token)

# Crear la carpeta "model" si no existe
destino = os.getenv("MODEL_HF_DIR")
os.makedirs(destino, exist_ok=True)

# Descargar todos los archivos del repositorio a la carpeta de destino
for archivo in archivos:
    ruta_archivo = hf_hub_download(
        repo_id=repo_id,
        filename=archivo,
        token=token,
        local_dir=destino,
    )
    print(f"Archivo {archivo} descargado a la ruta {ruta_archivo}.")

print(f"Todo el contenido del repositorio {repo_id} ha sido descargado en la carpeta '{destino}'.")