require('dotenv').config();

const apiKey = process.env.DJELIA_API_KEY;

// Alternatively: const apiKey = "your_api_key_here" (but shh, that's not safe!)

// Specify your audio file for transcription tests
const audioFilePath = process.env.TEST_AUDIO_FILE || "audio.wav";

const { Djelia } = require('djelia');

const djeliaClient = new Djelia(apiKey);

// if DJELIA_API_KEY is already set you can just do : (yes I know I'm making your life easy 😂)
const djeliaClient = new Djelia();

const languages = await djeliaClient.translation.getSupportedLanguages();
languages.forEach(lang => {
    console.log(`${lang.name}: ${lang.code}`);
});

const { TranslationRequest, Language, Versions } = require('djelia');

const request = new TranslationRequest(
    "Hello, how are you today?",
    Language.ENGLISH,
    Language.BAMBARA
);

const { TranslationResponse } = require('djelia');

try {
    const response = await djeliaClient.translation.translate(request, Versions.v1);
    console.log(`Original: ${request.text}`);
    console.log(`Translation: ${response.text}`);
} catch (error) {
    console.log(`Translation error: ${error.message}`);
}

const { Versions } = require('djelia');

try {
    const transcription = await djeliaClient.transcription.transcribe(
        audioFilePath,
        false,    // translateToFrench
        false,    // stream
        Versions.v2
    );
    console.log(`Transcribed ${transcription.length} segments:`);
    transcription.forEach(segment => {
        console.log(`[${segment.start.toFixed(2)}s - ${segment.end.toFixed(2)}s]: ${segment.text}`);
    });
} catch (error) {
    console.log(`Transcription error: ${error.message}`);
}

console.log("Streaming transcription (showing first 3 segments)...");
let segmentCount = 0;

try {
    const stream = await djeliaClient.transcription.transcribe(
        audioFilePath,
        false,    // translateToFrench
        true,     // stream
        Versions.v2
    );
    
    for await (const segment of stream) {
        segmentCount++;
        console.log(`Segment ${segmentCount}: [${segment.start.toFixed(2)}s]: ${segment.text}`);
        
        if (segmentCount >= 3) {  // Just showing first 3 for demo
            console.log("...and more segments!");
            break;
        }
    }
} catch (error) {
    console.log(`Streaming transcription error: ${error.message}`);
}

try {
    const frenchTranscription = await djeliaClient.transcription.transcribe(
        audioFilePath,
        true,     // translateToFrench
        false,    // stream
        Versions.v2
    );
    console.log(`French translation: ${frenchTranscription.text}`);
} catch (error) {
    console.log(`French transcription error: ${error.message}`);
}

const { TTSRequest } = require('djelia');

const ttsRequestV1 = new TTSRequest(
    "Aw ni ce, i ka kɛnɛ wa?",  // "Hello, how are you?" in Bambara
    1  // Choose from 0, 1, 2, 3, or 4
);

try {
    const audioFileV1 = await djeliaClient.tts.textToSpeech(
        ttsRequestV1,
        "hello_v1.wav",
        false,      // stream
        Versions.v1
    );
    console.log(`Audio saved to: ${audioFileV1}`);
} catch (error) {
    console.log(`TTS v1 error: ${error.message}`);
}

const { TTSRequestV2 } = require('djelia');

const ttsRequestV2 = new TTSRequestV2(
    "Aw ni ce, i ka kɛnɛ wa?",
    "Seydou speaks with a warm, welcoming tone",  // Must include Moussa, Sekou, or Seydou
    1.0  // Control speech pacing (0.1 - 2.0)
);

try {
    const audioFileV2 = await djeliaClient.tts.textToSpeech(
        ttsRequestV2,
        "hello_v2.wav",
        false,      // stream
        Versions.v2
    );
    console.log(`Natural audio saved to: ${audioFileV2}`);
} catch (error) {
    console.log(`TTS v2 error: ${error.message}`);
}

const streamingTtsRequest = new TTSRequestV2(
    "An filɛ ni ye yɔrɔ minna ni an ye an sigi ka a layɛ yala an bɛ ka baara min kɛ ɛsike a kɛlen don ka Ɲɛ wa, ...............", // a very long text 
    "Seydou speaks clearly and naturally",
    1.0
);

console.log("Streaming TTS generation...");
let chunkCount = 0;
let totalBytes = 0;
const maxChunks = 5;

try {
    const stream = await djeliaClient.tts.textToSpeech(
        streamingTtsRequest,
        "streamed_audio.wav",
        true,       // stream
        Versions.v2
    );
    
    for await (const chunk of stream) {
        chunkCount++;
        totalBytes += chunk.length;
        console.log(`Chunk ${chunkCount}: ${chunk.length} bytes`);
        
        if (chunkCount >= maxChunks) {
            console.log(`...and more chunks! (Total so far: ${totalBytes} bytes)`);
            break;
        }
    }
} catch (error) {
    console.log(`Streaming TTS error: ${error.message}`);
}

const { Versions } = require('djelia');

console.log(`Latest version: ${Versions.latest()}`);
console.log(`Available versions: ${Versions.allVersions().map(v => `v${v}`)}`);

// Use specific version
try {
    const transcription = await djeliaClient.transcription.transcribe(
        audioFilePath,
        false,    // translateToFrench
        false,    // stream
        Versions.v2
    );
    console.log(`Transcribed ${transcription.length} segments`);
} catch (error) {
    console.log(`Transcription error: ${error.message}`);
}

const { TranslationRequest, Language, TTSRequestV2, Versions } = require('djelia');

async function parallelOperations() {
    try {
        const translationRequest = new TranslationRequest(
            "Hello", 
            Language.ENGLISH, 
            Language.BAMBARA
        );
        const ttsRequest = new TTSRequestV2(
            "Aw ni ce, i ka kɛnɛ wa?",
            "Moussa speaks with a clear tone",
            1.0
        );
        
        const results = await Promise.allSettled([
            djeliaClient.translation.translate(translationRequest, Versions.v1),
            djeliaClient.transcription.transcribe(audioFilePath, false, false, Versions.v2),
            djeliaClient.tts.textToSpeech(ttsRequest, "parallel_tts.wav", false, Versions.v2)
        ]);
        
        results.forEach((result, i) => {
            if (result.status === 'fulfilled') {
                console.log(`Operation ${i+1} succeeded: ${result.value.constructor.name}`);
            } else {
                console.log(`Operation ${i+1} failed: ${result.reason.message}`);
            }
        });
    } catch (error) {
        console.log(`Parallel operations error: ${error.message}`);
    }
}

// Run it
parallelOperations();

const { 
    AuthenticationError, 
    APIError, 
    ValidationError, 
    LanguageError, 
    SpeakerError 
} = require('djelia');

try {
    const response = await djeliaClient.translation.translate(request, Versions.v1);
    console.log(`Translation: ${response.text}`);
} catch (error) {
    if (error instanceof AuthenticationError) {
        console.error('Authentication error (check API key):', error.message);
    } else if (error instanceof LanguageError) {
        console.error('Invalid or unsupported language:', error.message);
    } else if (error instanceof ValidationError) {
        console.error('Validation error (e.g., invalid input):', error.message);
    } else if (error instanceof APIError) {
        console.error(`API error (status ${error.statusCode}): ${error.message}`);
    } else {
        console.error('Unexpected error:', error.message);
    }
}