@remotion/install-whisper-cpp 4.0.141 → 4.0.143

package/dist/download-whisper-model.js

@@ -80,25 +80,23 @@ const downloadWhisperModel = async ({ model, folder, printOutput = true, onProgr
             // eslint-disable-next-line no-constant-condition
             while (true) {
                 const { done, value } = await reader.read();
-                if (!value) {
-                    throw new Error('Failed to read from stream');
-                }
-                downloaded += value.length;
-                if (printOutput) {
-                    if (downloaded - lastPrinted > 1024 * 1024 * 10 ||
-                        downloaded === totalFileSize) {
+                if (value) {
+                    downloaded += value.length;
+                    if (printOutput &&
+                        (downloaded - lastPrinted > 1024 * 1024 * 10 ||
+                            downloaded === totalFileSize)) {
                         console.log(`Downloaded ${downloaded} of ${contentLength} bytes (${((downloaded / Number(contentLength)) *
                             100).toFixed(2)}%)`);
                         lastPrinted = downloaded;
                     }
+                    fileStream.write(value, () => {
+                        onProgress === null || onProgress === void 0 ? void 0 : onProgress(downloaded, totalFileSize);
+                        if (downloaded === totalFileSize) {
+                            fileStream.end();
+                            resolve();
+                        }
+                    });
                 }
-                fileStream.write(value, () => {
-                    onProgress === null || onProgress === void 0 ? void 0 : onProgress(downloaded, totalFileSize);
-                    if (downloaded === totalFileSize) {
-                        fileStream.end();
-                        resolve();
-                    }
-                });
                 if (done) {
                     break;
                 }

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 export { Caption, convertToCaptions } from './convert-to-captions';
 export { OnProgress, WhisperModel, downloadWhisperModel, } from './download-whisper-model';
 export { installWhisperCpp } from './install-whisper-cpp';
+export type { Language } from './languages';
 export { TranscriptionJson, transcribe } from './transcribe';

package/dist/languages.d.ts

@@ -0,0 +1,5 @@
+/**
+ * List of supported languages of Whispher.
+ * https://github.com/ggerganov/whisper.cpp/blob/8f253ef3af1c62c04316ba4afa7145fc4d701a8c/whisper.cpp#L424
+ */
+export type Language = 'Afrikaans' | 'Albanian' | 'Amharic' | 'Arabic' | 'Armenian' | 'Assamese' | 'Azerbaijani' | 'Bashkir' | 'Basque' | 'Belarusian' | 'Bengali' | 'Bosnian' | 'Breton' | 'Bulgarian' | 'Burmese' | 'Castilian' | 'Catalan' | 'Chinese' | 'Croatian' | 'Czech' | 'Danish' | 'Dutch' | 'English' | 'Estonian' | 'Faroese' | 'Finnish' | 'Flemish' | 'French' | 'Galician' | 'Georgian' | 'German' | 'Greek' | 'Gujarati' | 'Haitian' | 'Haitian Creole' | 'Hausa' | 'Hawaiian' | 'Hebrew' | 'Hindi' | 'Hungarian' | 'Icelandic' | 'Indonesian' | 'Italian' | 'Japanese' | 'Javanese' | 'Kannada' | 'Kazakh' | 'Khmer' | 'Korean' | 'Lao' | 'Latin' | 'Latvian' | 'Letzeburgesch' | 'Lingala' | 'Lithuanian' | 'Luxembourgish' | 'Macedonian' | 'Malagasy' | 'Malay' | 'Malayalam' | 'Maltese' | 'Maori' | 'Marathi' | 'Moldavian' | 'Moldovan' | 'Mongolian' | 'Myanmar' | 'Nepali' | 'Norwegian' | 'Nynorsk' | 'Occitan' | 'Panjabi' | 'Pashto' | 'Persian' | 'Polish' | 'Portuguese' | 'Punjabi' | 'Pushto' | 'Romanian' | 'Russian' | 'Sanskrit' | 'Serbian' | 'Shona' | 'Sindhi' | 'Sinhala' | 'Sinhalese' | 'Slovak' | 'Slovenian' | 'Somali' | 'Spanish' | 'Sundanese' | 'Swahili' | 'Swedish' | 'Tagalog' | 'Tajik' | 'Tamil' | 'Tatar' | 'Telugu' | 'Thai' | 'Tibetan' | 'Turkish' | 'Turkmen' | 'Ukrainian' | 'Urdu' | 'Uzbek' | 'Valencian' | 'Vietnamese' | 'Welsh' | 'Yiddish' | 'Yoruba' | 'Zulu' | 'auto' | 'af' | 'am' | 'ar' | 'as' | 'az' | 'ba' | 'be' | 'bg' | 'bn' | 'bo' | 'br' | 'bs' | 'ca' | 'cs' | 'cy' | 'da' | 'de' | 'el' | 'en' | 'es' | 'et' | 'eu' | 'fa' | 'fi' | 'fo' | 'fr' | 'gl' | 'gu' | 'ha' | 'haw' | 'he' | 'hi' | 'hr' | 'ht' | 'hu' | 'hy' | 'id' | 'is' | 'it' | 'ja' | 'jw' | 'ka' | 'kk' | 'km' | 'kn' | 'ko' | 'la' | 'lb' | 'ln' | 'lo' | 'lt' | 'lv' | 'mg' | 'mi' | 'mk' | 'ml' | 'mn' | 'mr' | 'ms' | 'mt' | 'my' | 'ne' | 'nl' | 'nn' | 'no' | 'oc' | 'pa' | 'pl' | 'ps' | 'pt' | 'ro' | 'ru' | 'sa' | 'sd' | 'si' | 'sk' | 'sl' | 'sn' | 'so' | 'sq' | 'sr' | 'su' | 'sv' | 'sw' | 'ta' | 'te' | 'tg' | 'th' | 'tk' | 'tl' | 'tr' | 'tt' | 'uk' | 'ur' | 'uz' | 'vi' | 'yi' | 'yo' | 'zh';

package/dist/languages.js

@@ -0,0 +1,6 @@
+"use strict";
+/**
+ * List of supported languages of Whispher.
+ * https://github.com/ggerganov/whisper.cpp/blob/8f253ef3af1c62c04316ba4afa7145fc4d701a8c/whisper.cpp#L424
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/test/convert-to-captions.test.js

@@ -1,14 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const vitest_1 = require("vitest");
+const bun_test_1 = require("bun:test");
 const convert_to_captions_1 = require("../convert-to-captions");
 const example_payload_1 = require("./example-payload");
-(0, vitest_1.test)('Convert to captions - 200ms together', () => {
+(0, bun_test_1.test)('Convert to captions - 200ms together', () => {
     const { captions: transcript } = (0, convert_to_captions_1.convertToCaptions)({
         transcription: example_payload_1.examplePayload.transcription,
         combineTokensWithinMilliseconds: 200,
     });
-    (0, vitest_1.expect)(transcript).toEqual([
+    (0, bun_test_1.expect)(transcript).toEqual([
         { text: 'William', startInSeconds: 0.24 },
         { text: 'just', startInSeconds: 0.48 },
         { text: 'hit 100,000', startInSeconds: 0.7 },
@@ -120,12 +120,12 @@ const example_payload_1 = require("./example-payload");
         { text: 'beeping)', startInSeconds: 63.02 },
     ]);
 });
-(0, vitest_1.test)('Convert to captions - 0ms together', () => {
+(0, bun_test_1.test)('Convert to captions - 0ms together', () => {
     const { captions: transcript } = (0, convert_to_captions_1.convertToCaptions)({
         transcription: example_payload_1.examplePayload.transcription,
         combineTokensWithinMilliseconds: 0,
     });
-    (0, vitest_1.expect)(transcript).toEqual([
+    (0, bun_test_1.expect)(transcript).toEqual([
         { text: 'William', startInSeconds: 0.24 },
         { text: 'just', startInSeconds: 0.48 },
         { text: 'hit', startInSeconds: 0.7 },

package/dist/transcribe.d.ts

@@ -1,4 +1,5 @@
 import type { WhisperModel } from './download-whisper-model';
+import type { Language } from './languages';
 type Timestamps = {
     from: string;
     to: string;
@@ -57,7 +58,7 @@ export type TranscriptionJson<WithTokenLevelTimestamp extends boolean> = {
     result: Result;
     transcription: true extends WithTokenLevelTimestamp ? TranscriptionItemWithTimestamp[] : TranscriptionItem[];
 };
-export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inputPath, whisperPath, model, modelFolder, translateToEnglish, tokenLevelTimestamps, printOutput, tokensPerItem, }: {
+export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inputPath, whisperPath, model, modelFolder, translateToEnglish, tokenLevelTimestamps, printOutput, tokensPerItem, language, }: {
     inputPath: string;
     whisperPath: string;
     model: WhisperModel;
@@ -66,5 +67,6 @@ export declare const transcribe: <HasTokenLevelTimestamps extends boolean>({ inp
     translateToEnglish?: boolean | undefined;
     printOutput?: boolean | undefined;
     tokensPerItem?: (true extends HasTokenLevelTimestamps ? never : number | null) | undefined;
+    language?: Language | null | undefined;
 }) => Promise<TranscriptionJson<HasTokenLevelTimestamps>>;
 export {};

package/dist/transcribe.js

@@ -43,7 +43,7 @@ const readJson = async (jsonPath) => {
     const data = await node_fs_1.default.promises.readFile(jsonPath, 'utf8');
     return JSON.parse(data);
 };
-const transcribeToTempJSON = async ({ fileToTranscribe, whisperPath, model, tmpJSONPath, modelFolder, translate, tokenLevelTimestamps, printOutput, tokensPerItem, }) => {
+const transcribeToTempJSON = async ({ fileToTranscribe, whisperPath, model, tmpJSONPath, modelFolder, translate, tokenLevelTimestamps, printOutput, tokensPerItem, language, }) => {
     const modelPath = (0, download_whisper_model_1.getModelPath)(modelFolder !== null && modelFolder !== void 0 ? modelFolder : whisperPath, model);
     if (!node_fs_1.default.existsSync(modelPath)) {
         throw new Error(`Error: Model ${model} does not exist at ${modelFolder ? modelFolder : modelPath}. Check out the downloadWhisperModel() API at https://www.remotion.dev/docs/install-whisper-cpp/download-whisper-model to see how to install whisper models`);
@@ -60,6 +60,7 @@ const transcribeToTempJSON = async ({ fileToTranscribe, whisperPath, model, tmpJ
         tokenLevelTimestamps ? ['--dtw', model] : null,
         model ? [`-m`, `${modelPath}`] : null,
         translate ? '-tr' : null,
+        language ? ['-l', language.toLowerCase()] : null,
     ]
         .flat(1)
         .filter(Boolean);
@@ -107,7 +108,7 @@ const transcribeToTempJSON = async ({ fileToTranscribe, whisperPath, model, tmpJ
     });
     return { outputPath };
 };
-const transcribe = async ({ inputPath, whisperPath, model, modelFolder, translateToEnglish = false, tokenLevelTimestamps, printOutput = true, tokensPerItem, }) => {
+const transcribe = async ({ inputPath, whisperPath, model, modelFolder, translateToEnglish = false, tokenLevelTimestamps, printOutput = true, tokensPerItem, language, }) => {
     if (!(0, node_fs_1.existsSync)(whisperPath)) {
         throw new Error(`Whisper does not exist at ${whisperPath}. Double-check the passed whisperPath. If you havent installed whisper, check out the installWhisperCpp() API at https://www.remotion.dev/docs/install-whisper-cpp/install-whisper-cpp to see how to install whisper programatically.`);
     }
@@ -128,6 +129,7 @@ const transcribe = async ({ inputPath, whisperPath, model, modelFolder, translat
         tokenLevelTimestamps,
         printOutput,
         tokensPerItem: tokenLevelTimestamps ? 1 : tokensPerItem !== null && tokensPerItem !== void 0 ? tokensPerItem : 1,
+        language,
     });
     const json = (await readJson(tmpJSONPath));
     node_fs_1.default.unlinkSync(tmpJSONPath);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@remotion/install-whisper-cpp",
-  "version": "4.0.141",
+  "version": "4.0.143",
   "description": "Install helper for Whisper.cpp",
   "main": "dist/index.js",
   "sideEffects": false,
@@ -28,8 +28,7 @@
     "eslint-plugin-react": "7.32.2",
     "eslint-plugin-react-hooks": "4.4.0",
     "prettier": "3.2.5",
-    "prettier-plugin-organize-imports": "3.2.4",
-    "vitest": "0.31.1"
+    "prettier-plugin-organize-imports": "3.2.4"
   },
   "keywords": [
     "remotion",
@@ -44,6 +43,6 @@
     "lint": "eslint src --ext ts,tsx",
     "build": "tsc -d",
     "watch": "tsc -w",
-    "test": "vitest --run"
+    "test": "bun test src"
   }
 }