import { HfApi, HfFolder } from "@huggingface/hub";

async function uploadToHuggingFace() {
    try {
        console.log("Iniciando processo de upload para Hugging Face...");
        
        // Inicializar o cliente API
        const api = new HfApi();
        
        // Configurar suas credenciais
        // Você precisa ter um token de acesso do Hugging Face
        // Obtenha em: https://huggingface.co/settings/tokens
        const token = "seu_token_aqui"; // Substitua pelo seu token
        
        // Nome do seu repositório
        const repoId = "amos-fernadnes/meu-modelo";
        
        // Criar novo repositório
        console.log("Criando novo repositório...");
        await api.createRepo({
            token,
            name: "meu-modelo",
            organization: "amos-fernadnes", // seu username
            private: false,
        });
        
        // Upload dos arquivos do modelo
        console.log("Fazendo upload dos arquivos...");
        
        // Lista de arquivos para upload
        const files = [
            "config.json",
            "pytorch_model.bin",
            "special_tokens_map.json",
            "tokenizer.json",
            "tokenizer_config.json",
            "vocab.txt"
        ];
        
        for (const file of files) {
            console.log(`Uploading ${file}...`);
            await api.uploadFile({
                token,
                repo: repoId,
                path: file,
                file: `./modelo/${file}` // Caminho local dos seus arquivos
            });
        }
        
        console.log("Upload concluído com sucesso!");
        console.log(`Seu modelo está disponível em: https://huggingface.co/${repoId}`);
        
    } catch (error) {
        console.error("Erro durante o upload:", error);
    }
}

// Executar o upload
uploadToHuggingFace();