Training Custom Voices

const fs = require('fs');
const apiKey = process.env.VOICEDUB_API_KEY;
const baseUrl = 'https://api.voicedub.ai';

async function trainCustomVoice(name, audioFiles, maxMinutes = 10) {
  // Step 1: Create the voice
  const createResponse = await fetch(`${baseUrl}/v1/me/voices`, {
    method: 'POST',
    headers: {
      'Authorization': `Api-Key ${apiKey}`,
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({
      name: name,
      maxMinutes: maxMinutes,
      separate: true,
      numberOfFiles: audioFiles.length
    })
  });
  
  const { voice, uploadUrls } = await createResponse.json();
  console.log('Voice created:', voice.id);
  console.log('Required credits:', voice.requiredCredits);
  
  // Step 2: Upload training files
  for (let i = 0; i < audioFiles.length; i++) {
    const fileData = fs.readFileSync(audioFiles[i]);
    
    await fetch(uploadUrls[i], {
      method: 'PUT',
      headers: { 'Content-Type': 'audio/mpeg' },
      body: fileData
    });
    
    console.log(`Uploaded file ${i + 1}/${audioFiles.length}`);
  }
  
  // Step 3: Start training
  await fetch(`${baseUrl}/v1/me/voices/${voice.id}/clone`, {
    method: 'POST',
    headers: { 'Authorization': `Api-Key ${apiKey}` }
  });
  
  console.log('Training started...');
  
  // Step 4: Wait for completion
  let status = 'queued';
  while (!['done', 'error'].includes(status)) {
    await new Promise(resolve => setTimeout(resolve, 30000)); // Wait 30 seconds
    
    const statusResponse = await fetch(`${baseUrl}/v1/me/voices/${voice.id}`, {
      headers: { 'Authorization': `Api-Key ${apiKey}` }
    });
    
    const voiceData = await statusResponse.json();
    status = voiceData.voice.status;
    
    console.log('Training status:', status);
    
    if (status === 'done') {
      console.log('Training completed! Voice ID:', voice.id);
      return voice.id;
    } else if (status === 'error') {
      throw new Error(`Training failed: ${voiceData.voice.errorMessage}`);
    }
  }
}

// Usage
const audioFiles = [
  './training-audio-1.mp3',
  './training-audio-2.mp3', 
  './training-audio-3.mp3'
];

trainCustomVoice('My Custom Voice', audioFiles, 15)
  .then(voiceId => console.log('Success! Voice ID:', voiceId))
  .catch(err => console.error('Error:', err));