pyannote-cpp-node 0.3.0 → 0.5.0

package/README.md

@@ -7,7 +7,7 @@ Node.js native bindings for integrated Whisper transcription + speaker diarizati
 
 ## Overview
 
-`pyannote-cpp-node` exposes the integrated C++ pipeline that combines streaming diarization and Whisper transcription into a single API.
+`pyannote-cpp-node` exposes the integrated C++ pipeline that combines Whisper transcription and speaker diarization into a single API.
 
 Given 16 kHz mono PCM audio (`Float32Array`), it produces cumulative and final transcript segments shaped as:
 
@@ -15,16 +15,21 @@ Given 16 kHz mono PCM audio (`Float32Array`), it produces cumulative and final t
 - segment start/duration in seconds
 - segment text
 
-The API supports both one-shot processing (`transcribe`) and incremental streaming (`createSession` + `push`/`finalize`). All heavy operations are asynchronous and run on libuv worker threads.
+The API supports three modes: **offline** batch processing (`transcribeOffline`), **one-shot** streaming (`transcribe`), and **incremental** streaming (`createSession` + `push`/`finalize`). All heavy operations are asynchronous and run on libuv worker threads.
 
 ## Features
 
 - Integrated transcription + diarization in one pipeline
 - Speaker-labeled transcript segments with sentence-level text
-- One-shot and streaming APIs with the same output schema
-- Incremental `segments` events plus separate real-time `audio` chunk streaming
+- **Offline mode**: runs Whisper on the full audio at once + offline diarization (fastest for batch)
+- **One-shot mode**: streaming pipeline with automatic chunking
+- **Streaming mode**: incremental push/finalize with real-time `segments` events and `audio` chunk streaming
 - Deterministic output for the same audio/models/config
 - CoreML-accelerated inference on macOS
+- **Shared model cache**: all models loaded once during `Pipeline.load()`, reused across offline/streaming/session modes
+- **Runtime backend switching**: switch Whisper between GPU-only and CoreML-accelerated without reloading the pipeline
+- **Progress reporting**: optional `onProgress` callback for `transcribeOffline` reports Whisper, diarization, and alignment phases
+- **Real-time segment streaming**: optional `onSegment` callback for `transcribeOffline` delivers each Whisper segment (start, end, text) as it's produced — enables live transcript preview and time-based loading bars
 - TypeScript-first API with complete type definitions
 
 ## Requirements
@@ -68,7 +73,9 @@ const pipeline = await Pipeline.load({
 });
 
 const audio = loadAudioAsFloat32Array('./audio-16khz-mono.wav');
-const result = await pipeline.transcribe(audio);
+
+// Offline mode — fastest for batch processing
+const result = await pipeline.transcribeOffline(audio);
 
 for (const segment of result.segments) {
   const end = segment.start + segment.duration;
@@ -87,8 +94,12 @@ pipeline.close();
 ```typescript
 class Pipeline {
   static async load(config: ModelConfig): Promise<Pipeline>;
+  async transcribeOffline(audio: Float32Array, onProgress?: (phase: number, progress: number) => void, onSegment?: (start: number, end: number, text: string) => void): Promise<TranscriptionResult>;
   async transcribe(audio: Float32Array): Promise<TranscriptionResult>;
+  setLanguage(language: string): void;
+  setDecodeOptions(options: DecodeOptions): void;
   createSession(): PipelineSession;
+  async setUseCoreml(useCoreml: boolean): Promise<void>;
   close(): void;
   get isClosed(): boolean;
 }
@@ -96,12 +107,85 @@ class Pipeline {
 
 #### `static async load(config: ModelConfig): Promise<Pipeline>`
 
-Validates model paths and initializes native pipeline resources.
+Validates model paths and loads all models (Whisper, CoreML segmentation/embedding, PLDA, and optionally VAD) into a shared cache on a background thread. Models are loaded once and reused across all subsequent `transcribe()`, `transcribeOffline()`, and `createSession()` calls — no redundant loading occurs when switching between modes. Models are freed only when `close()` is called.
+
+#### `async transcribeOffline(audio: Float32Array, onProgress?, onSegment?): Promise<TranscriptionResult>`
+
+Runs Whisper on the **entire** audio buffer in a single `whisper_full()` call, then runs offline diarization and WhisperX-style speaker alignment. This is the fastest mode for batch processing — no streaming infrastructure is involved.
+
+The optional `onProgress` callback receives `(phase, progress)` updates:
+
+| Phase | Value | Meaning |
+| --- | --- | --- |
+| `0` | `0`–`100` | Whisper transcription progress (percentage) |
+| `1` | `0` | Diarization started |
+| `2` | `0` | Speaker alignment started |
+
+```typescript
+const result = await pipeline.transcribeOffline(audio, (phase, progress) => {
+  if (phase === 0) console.log(`Transcribing: ${progress}%`);
+  if (phase === 1) console.log('Running diarization...');
+  if (phase === 2) console.log('Aligning speakers...');
+});
+```
+
+The optional `onSegment` callback receives `(start, end, text)` for each Whisper segment as it's produced during transcription. Times are in seconds. This enables live transcript preview before diarization and alignment complete.
+
+```typescript
+const result = await pipeline.transcribeOffline(audio, undefined, (start, end, text) => {
+  console.log(`[${start.toFixed(2)}-${end.toFixed(2)}] ${text}`);
+});
+```
+
+Both callbacks can be used simultaneously:
+
+```typescript
+const result = await pipeline.transcribeOffline(
+  audio,
+  (phase, progress) => {
+    if (phase === 0) updateProgressBar(progress);
+  },
+  (start, end, text) => {
+    appendToTranscriptPreview(start, end, text);
+  },
+);
+```
 
 #### `async transcribe(audio: Float32Array): Promise<TranscriptionResult>`
 
-Runs one-shot transcription + diarization on the full audio buffer.
+Runs one-shot transcription + diarization using the streaming pipeline internally (pushes 1-second chunks then finalizes).
+
+#### `setLanguage(language: string): void`
 
+Updates the Whisper decode language for subsequent `transcribe()` calls. This is a convenience shorthand for `setDecodeOptions({ language })`.
+
+#### `setDecodeOptions(options: DecodeOptions): void`
+
+Updates one or more Whisper decode options for subsequent `transcribe()` calls. Only the fields you pass are changed; others retain their current values. See `DecodeOptions` for available fields.
+
+
+#### `async setUseCoreml(useCoreml: boolean): Promise<void>`
+
+Switches the Whisper inference backend between GPU-only (`false`) and GPU+CoreML (`true`) at runtime. The method reloads the Whisper context on a background thread with the new `use_coreml` setting. The promise resolves when the new context is ready.
+
+- If the requested mode matches the current mode, returns immediately (no reload).
+- Throws if the pipeline is closed, busy, or models are not loaded.
+- After switching, all subsequent `transcribe()`, `transcribeOffline()`, and streaming session calls use the new backend.
+
+```typescript
+// Start with GPU-only Whisper
+const pipeline = await Pipeline.load({
+  ...modelPaths,
+  useCoreml: false,
+});
+
+// Switch to CoreML-accelerated Whisper at runtime
+await pipeline.setUseCoreml(true);
+const result = await pipeline.transcribeOffline(audio);
+
+// Switch back to GPU-only
+await pipeline.setUseCoreml(false);
+```
 #### `createSession(): PipelineSession`
 
 Creates an independent streaming session for incremental processing.
@@ -120,6 +204,8 @@ Returns `true` after `close()`.
 ```typescript
 class PipelineSession extends EventEmitter {
   async push(audio: Float32Array): Promise<boolean[]>;
+  setLanguage(language: string): void;
+  setDecodeOptions(options: DecodeOptions): void;
   async finalize(): Promise<TranscriptionResult>;
   close(): void;
   get isClosed(): boolean;
@@ -146,6 +232,14 @@ Pushes an arbitrary number of samples into the streaming pipeline.
 - First 10 seconds return an empty array because the pipeline needs a full 10-second window
 - Chunk size is flexible; not restricted to 16,000-sample pushes
 
+#### `setLanguage(language: string): void`
+
+Updates the Whisper decode language on the live streaming session. Takes effect on the next Whisper decode run. Thread-safe — the change is pushed to the C++ pipeline immediately.
+
+#### `setDecodeOptions(options: DecodeOptions): void`
+
+Updates one or more Whisper decode options on the live streaming session. Takes effect on the next Whisper decode run. Thread-safe — changes are pushed to the C++ pipeline immediately. Only the fields you pass are changed; others retain their current values.
+
 #### `async finalize(): Promise<TranscriptionResult>`
 
 Flushes all stages, runs final recluster + alignment, and returns the definitive result.
@@ -284,6 +378,41 @@ export interface ModelConfig {
   suppressNst?: boolean;
 }
 
+export interface DecodeOptions {
+  /** Language code (e.g., 'en', 'zh'). Omit for auto-detect. */
+  language?: string;
+  /** Translate non-English speech to English */
+  translate?: boolean;
+  /** Auto-detect spoken language. Overrides 'language' when true. */
+  detectLanguage?: boolean;
+  /** Number of threads for Whisper inference */
+  nThreads?: number;
+  /** Sampling temperature. 0.0 = greedy deterministic. */
+  temperature?: number;
+  /** Temperature increment for fallback retries */
+  temperatureInc?: number;
+  /** Disable temperature fallback. If true, temperatureInc is ignored. */
+  noFallback?: boolean;
+  /** Beam search size. -1 uses greedy decoding. >1 enables beam search. */
+  beamSize?: number;
+  /** Best-of-N sampling candidates for greedy decoding */
+  bestOf?: number;
+  /** Entropy threshold for decoder fallback */
+  entropyThold?: number;
+  /** Log probability threshold for decoder fallback */
+  logprobThold?: number;
+  /** No-speech probability threshold */
+  noSpeechThold?: number;
+  /** Initial prompt text to condition the decoder */
+  prompt?: string;
+  /** Don't use previous segment as context for next segment */
+  noContext?: boolean;
+  /** Suppress blank outputs at the beginning of segments */
+  suppressBlank?: boolean;
+  /** Suppress non-speech tokens */
+  suppressNst?: boolean;
+}
+
 export interface AlignedSegment {
   /** Global speaker label (e.g., SPEAKER_00). */
   speaker: string;
@@ -306,7 +435,68 @@ export interface TranscriptionResult {
 
 ## Usage Examples
 
-### One-shot transcription
+### Offline transcription (recommended for batch)
+
+```typescript
+import { Pipeline } from 'pyannote-cpp-node';
+
+async function runOffline(audio: Float32Array) {
+  const pipeline = await Pipeline.load({
+    segModelPath: './models/segmentation.gguf',
+    embModelPath: './models/embedding.gguf',
+    pldaPath: './models/plda.gguf',
+    coremlPath: './models/embedding.mlpackage',
+    segCoremlPath: './models/segmentation.mlpackage',
+    whisperModelPath: './models/ggml-large-v3-turbo-q5_0.bin',
+  });
+
+  // Runs Whisper on full audio at once + offline diarization
+  const result = await pipeline.transcribeOffline(audio);
+
+  for (const seg of result.segments) {
+    const end = seg.start + seg.duration;
+    console.log(`[${seg.speaker}] ${seg.start.toFixed(2)}-${end.toFixed(2)} ${seg.text.trim()}`);
+  }
+
+  pipeline.close();
+}
+```
+
+### Offline transcription with progress and live transcript preview
+
+```typescript
+import { Pipeline } from 'pyannote-cpp-node';
+
+async function runOfflineWithCallbacks(audio: Float32Array) {
+  const pipeline = await Pipeline.load({
+    segModelPath: './models/segmentation.gguf',
+    embModelPath: './models/embedding.gguf',
+    pldaPath: './models/plda.gguf',
+    coremlPath: './models/embedding.mlpackage',
+    segCoremlPath: './models/segmentation.mlpackage',
+    whisperModelPath: './models/ggml-large-v3-turbo-q5_0.bin',
+  });
+
+  const result = await pipeline.transcribeOffline(
+    audio,
+    // Progress callback — phase 0 is Whisper (0-100%), phase 1 is diarization, phase 2 is alignment
+    (phase, progress) => {
+      if (phase === 0) updateProgressBar(progress);
+      if (phase === 1) showStatus('Identifying speakers...');
+      if (phase === 2) showStatus('Aligning speakers to transcript...');
+    },
+    // Segment callback — each Whisper segment as it's produced (before diarization)
+    (start, end, text) => {
+      appendToLivePreview(`[${start.toFixed(2)}-${end.toFixed(2)}] ${text}`);
+    },
+  );
+
+  console.log(`Done: ${result.segments.length} speaker-labeled segments`);
+  pipeline.close();
+}
+```
+
+### One-shot transcription (streaming internals)
 
 ```typescript
 import { Pipeline } from 'pyannote-cpp-node';
@@ -321,6 +511,7 @@ async function runOneShot(audio: Float32Array) {
     whisperModelPath: './models/ggml-large-v3-turbo-q5_0.bin',
   });
 
+  // Uses streaming pipeline internally (push 1s chunks + finalize)
   const result = await pipeline.transcribe(audio);
 
   for (const seg of result.segments) {
@@ -419,6 +610,88 @@ const pipeline = await Pipeline.load({
 });
 ```
 
+### Changing language at runtime
+
+```typescript
+import { Pipeline } from 'pyannote-cpp-node';
+
+const pipeline = await Pipeline.load({
+  segModelPath: './models/segmentation.gguf',
+  embModelPath: './models/embedding.gguf',
+  pldaPath: './models/plda.gguf',
+  coremlPath: './models/embedding.mlpackage',
+  segCoremlPath: './models/segmentation.mlpackage',
+  whisperModelPath: './models/ggml-large-v3-turbo-q5_0.bin',
+  language: 'en',
+});
+
+// First transcription in English
+const result1 = await pipeline.transcribe(englishAudio);
+
+// Switch to Korean for the next transcription
+pipeline.setLanguage('ko');
+const result2 = await pipeline.transcribe(koreanAudio);
+
+// Or update multiple decode options at once
+pipeline.setDecodeOptions({
+  language: 'zh',
+  temperature: 0.2,
+  beamSize: 5,
+});
+const result3 = await pipeline.transcribe(chineseAudio);
+
+pipeline.close();
+```
+
+### Switching Whisper backend at runtime
+
+```typescript
+import { Pipeline } from 'pyannote-cpp-node';
+
+// Start with GPU-only Whisper (default)
+const pipeline = await Pipeline.load({
+  segModelPath: './models/segmentation.gguf',
+  embModelPath: './models/embedding.gguf',
+  pldaPath: './models/plda.gguf',
+  coremlPath: './models/embedding.mlpackage',
+  segCoremlPath: './models/segmentation.mlpackage',
+  whisperModelPath: './models/ggml-large-v3-turbo-q5_0.bin',
+  useCoreml: false,
+});
+
+// Switch to CoreML-accelerated Whisper encoder at runtime
+// (requires ggml-large-v3-turbo-q5_0-encoder.mlmodelc next to the GGUF)
+await pipeline.setUseCoreml(true);
+const result1 = await pipeline.transcribeOffline(audio);
+
+// Switch back to GPU-only
+await pipeline.setUseCoreml(false);
+const result2 = await pipeline.transcribeOffline(audio);
+
+pipeline.close();
+```
+
+Streaming sessions also support runtime changes:
+
+```typescript
+const session = pipeline.createSession();
+
+session.on('segments', (segments) => {
+  console.log(segments);
+});
+
+// Push English audio
+await session.push(englishChunk);
+
+// Switch language mid-stream — takes effect on the next Whisper decode
+session.setLanguage('ko');
+await session.push(koreanChunk);
+
+const result = await session.finalize();
+
+session.close();
+```
+
 ## JSON Output Format
 
 The pipeline returns this JSON shape:
@@ -447,7 +720,15 @@ All API methods expect decoded PCM samples; file decoding/resampling is handled
 
 ## Architecture
 
-The integrated pipeline runs in 7 stages:
+### Offline mode (`transcribeOffline`)
+
+1. Single `whisper_full()` call on entire audio
+2. Offline diarization (segmentation → powerset → embeddings → PLDA → AHC → VBx)
+3. WhisperX-style alignment (speaker assignment by maximum segment overlap)
+
+### Streaming mode (`transcribe` / `createSession`)
+
+The streaming pipeline runs in 7 stages:
 
 1. VAD silence filter (optional compression of long silence)
 2. Audio buffer (stream-safe FIFO with timestamp tracking)
@@ -459,8 +740,9 @@ The integrated pipeline runs in 7 stages:
 
 ## Performance
 
+- Offline transcription + diarization: **~12x real-time** (30s audio in 2.5s)
 - Diarization only: **39x real-time**
-- Integrated transcription + diarization: **~14.6x real-time**
+- Integrated streaming transcription + diarization: **~14.6x real-time**
 - 45-minute Korean meeting test (6 speakers): **2713s audio in 186s**
 - Each Whisper segment maps 1:1 to a speaker-labeled segment (no merging)
 - Speaker confusion rate: **2.55%**

package/dist/Pipeline.d.ts

@@ -5,9 +5,11 @@ export declare class Pipeline {
     private constructor();
     static load(config: ModelConfig): Promise<Pipeline>;
     transcribe(audio: Float32Array): Promise<TranscriptionResult>;
+    transcribeOffline(audio: Float32Array, onProgress?: (phase: number, progress: number) => void, onSegment?: (start: number, end: number, text: string) => void): Promise<TranscriptionResult>;
     setLanguage(language: string): void;
     setDecodeOptions(options: DecodeOptions): void;
     createSession(): PipelineSession;
+    setUseCoreml(useCoreml: boolean): Promise<void>;
     close(): void;
     get isClosed(): boolean;
 }

package/dist/Pipeline.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Pipeline.d.ts","sourceRoot":"","sources":["../src/Pipeline.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,WAAW,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAElF,qBAAa,QAAQ;IACnB,OAAO,CAAC,MAAM,CAAsB;IAEpC,OAAO;WAIM,IAAI,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;IAiBnD,UAAU,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAOnE,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKnC,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAK9C,aAAa,IAAI,eAAe;IAWhC,KAAK,IAAI,IAAI;IACb,IAAI,QAAQ,IAAI,OAAO,CAAiC;CACzD"}
+{"version":3,"file":"Pipeline.d.ts","sourceRoot":"","sources":["../src/Pipeline.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,WAAW,EAAE,mBAAmB,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAElF,qBAAa,QAAQ;IACnB,OAAO,CAAC,MAAM,CAAsB;IAEpC,OAAO;WAIM,IAAI,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;IAkBnD,UAAU,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAO7D,iBAAiB,CACrB,KAAK,EAAE,YAAY,EACnB,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,EACtD,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,GAC7D,OAAO,CAAC,mBAAmB,CAAC;IAO/B,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKnC,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAK9C,aAAa,IAAI,eAAe;IAW1B,YAAY,CAAC,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAKrD,KAAK,IAAI,IAAI;IACb,IAAI,QAAQ,IAAI,OAAO,CAAiC;CACzD"}

package/dist/Pipeline.js

@@ -21,6 +21,7 @@ export class Pipeline {
             accessSync(config.vadModelPath);
         const binding = getBinding();
         const native = new binding.PipelineModel(config);
+        await native.loadModels();
         return new Pipeline(native);
     }
     async transcribe(audio) {
@@ -32,6 +33,15 @@ export class Pipeline {
             throw new Error('Audio must not be empty');
         return this.native.transcribe(audio);
     }
+    async transcribeOffline(audio, onProgress, onSegment) {
+        if (this.native.isClosed)
+            throw new Error('Pipeline is closed');
+        if (!(audio instanceof Float32Array))
+            throw new TypeError('Expected Float32Array');
+        if (audio.length === 0)
+            throw new Error('Audio must not be empty');
+        return this.native.transcribeOffline(audio, onProgress, onSegment);
+    }
     setLanguage(language) {
         if (this.native.isClosed)
             throw new Error('Pipeline is closed');
@@ -50,6 +60,11 @@ export class Pipeline {
         session._setNative(nativeSession);
         return session;
     }
+    async setUseCoreml(useCoreml) {
+        if (this.native.isClosed)
+            throw new Error('Pipeline is closed');
+        return this.native.switchWhisperMode(useCoreml);
+    }
     close() { this.native.close(); }
     get isClosed() { return this.native.isClosed; }
 }

package/dist/Pipeline.js.map

@@ -1 +1 @@
-{"version":3,"file":"Pipeline.js","sourceRoot":"","sources":["../src/Pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,UAAU,EAA4B,MAAM,cAAc,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAGvD,MAAM,OAAO,QAAQ;IACX,MAAM,CAAsB;IAEpC,YAAoB,MAA2B;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAmB;QACnC,MAAM,aAAa,GAAG;YACpB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,QAAQ;YACf,MAAM,CAAC,UAAU;YACjB,MAAM,CAAC,aAAa;YACpB,MAAM,CAAC,gBAAgB;SACxB,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,aAAa;YAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,MAAM,CAAC,YAAY;YAAE,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACjD,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAmB;QAClC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,CAAC,KAAK,YAAY,YAAY,CAAC;YAAE,MAAM,IAAI,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACnF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,QAAgB;QAC1B,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,gBAAgB,CAAC,OAAsB;QACrC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,aAAa;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAC7C,CAAC,QAAe,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAC1D,CAAC,KAAmB,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,KAAK,CAAC,CACzD,CAAC;QACF,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAClC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,KAAW,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACtC,IAAI,QAAQ,KAAc,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;CACzD"}
+{"version":3,"file":"Pipeline.js","sourceRoot":"","sources":["../src/Pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,UAAU,EAA4B,MAAM,cAAc,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAGvD,MAAM,OAAO,QAAQ;IACX,MAAM,CAAsB;IAEpC,YAAoB,MAA2B;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAmB;QACnC,MAAM,aAAa,GAAG;YACpB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,YAAY;YACnB,MAAM,CAAC,QAAQ;YACf,MAAM,CAAC,UAAU;YACjB,MAAM,CAAC,aAAa;YACpB,MAAM,CAAC,gBAAgB;SACxB,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,aAAa;YAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,MAAM,CAAC,YAAY;YAAE,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;QAC1B,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAmB;QAClC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,CAAC,KAAK,YAAY,YAAY,CAAC;YAAE,MAAM,IAAI,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACnF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,KAAmB,EACnB,UAAsD,EACtD,SAA8D;QAE9D,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,CAAC,KAAK,YAAY,YAAY,CAAC;YAAE,MAAM,IAAI,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACnF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;IACrE,CAAC;IAED,WAAW,CAAC,QAAgB;QAC1B,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,gBAAgB,CAAC,OAAsB;QACrC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,aAAa;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAC7C,CAAC,QAAe,EAAE,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,EAC1D,CAAC,KAAmB,EAAE,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,KAAK,CAAC,CACzD,CAAC;QACF,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAClC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAkB;QACnC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,KAAW,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACtC,IAAI,QAAQ,KAAc,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;CACzD"}

package/dist/binding.d.ts

@@ -1,11 +1,15 @@
 import type { ModelConfig, TranscriptionResult } from './types.js';
 export interface NativePipelineModel {
     transcribe(audio: Float32Array): Promise<TranscriptionResult>;
+    transcribeOffline(audio: Float32Array, onProgress?: (phase: number, progress: number) => void, onSegment?: (start: number, end: number, text: string) => void): Promise<TranscriptionResult>;
     setLanguage(language: string): void;
     setDecodeOptions(options: Record<string, unknown>): void;
     createSession(segmentsCb: (segments: any[]) => void, audioCb: (audio: Float32Array) => void): NativePipelineSession;
     close(): void;
     isClosed: boolean;
+    loadModels(): Promise<void>;
+    isLoaded: boolean;
+    switchWhisperMode(useCoreml: boolean): Promise<void>;
 }
 export interface NativePipelineSession {
     push(audio: Float32Array): Promise<boolean[]>;

package/dist/binding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"binding.d.ts","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,WAAW,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAInE,MAAM,WAAW,mBAAmB;IAClC,UAAU,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC9D,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACzD,aAAa,CACX,UAAU,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,IAAI,EACrC,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,GACrC,qBAAqB,CAAC;IACzB,KAAK,IAAI,IAAI,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC9C,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACzD,QAAQ,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,KAAK,IAAI,IAAI,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,KAAK,MAAM,EAAE,WAAW,KAAK,mBAAmB,CAAC;IAChE,eAAe,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,qBAAqB,CAAC;CACpE;AAoCD,wBAAgB,UAAU,IAAI,aAAa,CAyB1C"}
+{"version":3,"file":"binding.d.ts","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,WAAW,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAInE,MAAM,WAAW,mBAAmB;IAClC,UAAU,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC9D,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,EAAE,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC7L,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACzD,aAAa,CACX,UAAU,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,KAAK,IAAI,EACrC,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,GACrC,qBAAqB,CAAC;IACzB,KAAK,IAAI,IAAI,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,iBAAiB,CAAC,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACtD;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;IAC9C,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACzD,QAAQ,IAAI,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,KAAK,IAAI,IAAI,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,KAAK,MAAM,EAAE,WAAW,KAAK,mBAAmB,CAAC;IAChE,eAAe,EAAE,KAAK,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,qBAAqB,CAAC;CACpE;AAoCD,wBAAgB,UAAU,IAAI,aAAa,CAyB1C"}

package/dist/binding.js.map

@@ -1 +1 @@
-{"version":3,"file":"binding.js","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAIvC,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AA4B/C,IAAI,aAAa,GAAyB,IAAI,CAAC;AAE/C,SAAS,cAAc;IACrB,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,yBAAyB,OAAO,CAAC,QAAQ,oDAAoD,CAC9F,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC7B,OAAO,iCAAiC,CAAC;IAC3C,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC3B,OAAO,+BAA+B,CAAC;IACzC,CAAC;IAED,MAAM,IAAI,KAAK,CACb,sCAAsC,OAAO,CAAC,IAAI,8CAA8C,CACjG,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAChD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,SAAS,GAAG,KAAgC,CAAC;IACnD,OAAO,CACL,OAAO,SAAS,CAAC,aAAa,KAAK,UAAU;QAC7C,OAAO,SAAS,CAAC,eAAe,KAAK,UAAU,CAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;QAC3B,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,IAAI,KAAK,CACb,iCAAiC,WAAW,gEAAgE,OAAO,EAAE,CACtH,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CACb,sCAAsC,WAAW,6DAA6D,CAC/G,CAAC;IACJ,CAAC;IAED,aAAa,GAAG,MAAM,CAAC;IACvB,OAAO,aAAa,CAAC;AACvB,CAAC"}
+{"version":3,"file":"binding.js","sourceRoot":"","sources":["../src/binding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAIvC,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAgC/C,IAAI,aAAa,GAAyB,IAAI,CAAC;AAE/C,SAAS,cAAc;IACrB,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,yBAAyB,OAAO,CAAC,QAAQ,oDAAoD,CAC9F,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC7B,OAAO,iCAAiC,CAAC;IAC3C,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC3B,OAAO,+BAA+B,CAAC;IACzC,CAAC;IAED,MAAM,IAAI,KAAK,CACb,sCAAsC,OAAO,CAAC,IAAI,8CAA8C,CACjG,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAChD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,SAAS,GAAG,KAAgC,CAAC;IACnD,OAAO,CACL,OAAO,SAAS,CAAC,aAAa,KAAK,UAAU;QAC7C,OAAO,SAAS,CAAC,eAAe,KAAK,UAAU,CAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;QAC3B,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;IAErC,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,IAAI,KAAK,CACb,iCAAiC,WAAW,gEAAgE,OAAO,EAAE,CACtH,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CACb,sCAAsC,WAAW,6DAA6D,CAC/G,CAAC;IACJ,CAAC;IAED,aAAa,GAAG,MAAM,CAAC;IACvB,OAAO,aAAa,CAAC;AACvB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pyannote-cpp-node",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -17,8 +17,8 @@
     "access": "public"
   },
   "optionalDependencies": {
-    "@pyannote-cpp-node/darwin-arm64": "0.3.0",
-    "@pyannote-cpp-node/darwin-x64": "0.3.0"
+    "@pyannote-cpp-node/darwin-arm64": "0.5.0",
+    "@pyannote-cpp-node/darwin-x64": "0.5.0"
   },
   "devDependencies": {
     "typescript": "^5.7.0"