from huggingface_hub import snapshot_download
import os



# Función para descargar los modelos
def download_models(model_list):
    base_dir = "/data/"
    os.makedirs(base_dir, exist_ok=True)
    
    downloaded_models = []
    errors = []
    
    for model_repo in model_list:
        try:
            model_name = model_repo.split("/")[-1]  # Extrae el nombre del modelo después de la barra
            target_dir = os.path.join(base_dir, model_name)
            
            # Descargar el modelo
            print(f"📥 Descargando {model_repo} en {target_dir}")
            snapshot_download(
                repo_id=model_repo,
                local_dir=target_dir
            )
            downloaded_models.append(model_name)
        except Exception as e:
            errors.append(f"Error al descargar {model_repo}: {str(e)}")
    
    return downloaded_models, errors