index.ts

import { join } from "node:path";
import {
    AssemblyAIProvider,
    StarlingSTTProvider,
    WhisperSTTProvider,
    TranscriptionProvider,
    convertFile,
    getAudioFilePaths,
    saveFileToS3,
} from "./utils";
import { exists, mkdir } from "node:fs/promises";

async function main() {
    const language = "nl";
    const audioPaths = await getAudioFilePaths("audio-" + language);
    const providers: TranscriptionProvider[] = [
        new AssemblyAIProvider(),
        // new StarlingSTTProvider(),
        new WhisperSTTProvider(language),
    ];
    const results = await Promise.allSettled(
        audioPaths.map(async (path) => {
            try {
                const converted = await convertFile(path);
                const uploaded = await saveFileToS3(converted);

                const transcripts = await Promise.allSettled(
                    providers.map((el) => el.transcribe(uploaded)),
                );

                const transcriptMap: any = {};
                // add to the map both the name and the transcript
                [...providers.map((p) => p.name)].forEach(
                    (p, idx) => (transcriptMap[p] = transcripts[idx]),
                );

                return {
                    path,
                    uploaded,
                    transcriptions: transcriptMap,
                    status: "success",
                };
            } catch (error) {
                console.error(`Failed to process ${path}:`, error);
                return { path, error, status: "failed" };
            }
        }),
    );

    // Separate successful and failed files
    const successful = results.filter(
        (r) => r.status === "fulfilled" && r.value.status === "success",
    );
    const failed = results.filter(
        (r) =>
            r.status === "rejected" ||
            (r.status === "fulfilled" && r.value.status === "failed"),
    );

    if (failed.length > 0) {
        console.warn(`${failed.length} files failed to process`);
    }

    console.log("success", successful);
    console.log("failed", failed);

    const outputDir = join(import.meta.dir, "outputs");

    if (!(await exists(outputDir))) {
        await mkdir(outputDir);
    }

    const file = join(outputDir, language + Date.now().toString() + ".json");

    console.log();
    await Bun.write(
        file,
        JSON.stringify(
            {
                successful,
                failed,
            },
            null,
            4,
        ),
    );
}

main()
    .then(() => console.log("✅ Done"))
    .catch((e) => {
        console.error("❌ Something went wrong:", e);
        process.exit(1);
    });