-
Notifications
You must be signed in to change notification settings - Fork 27
/
file.js
executable file
·95 lines (83 loc) · 2.88 KB
/
file.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#! /usr/bin/env node
//
// Copyright 2022-2023 Picovoice Inc.
//
// You may not use this file except in compliance with the license. A copy of the license is located in the "LICENSE"
// file accompanying this source.
//
// Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
// an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
//
"use strict";
const { program } = require("commander");
const fs = require("fs");
const { Leopard, LeopardActivationLimitReached } = require("@picovoice/leopard-node");
program
.requiredOption(
"-a, --access_key <string>",
"AccessKey obtain from the Picovoice Console (https://console.picovoice.ai/)"
)
.requiredOption(
"-i, --input_audio_file_path <string>",
"input audio file"
)
.option(
"-l, --library_file_path <string>",
"absolute path to leopard dynamic library"
)
.option("-m, --model_file_path <string>", "absolute path to leopard model")
.option("-p, --disable_automatic_punctuation", "disable automatic punctuation")
.option("-d, --disable_speaker_diarization", "disable speaker diarization")
.option("-v, --verbose", "verbose mode, prints metadata");
if (process.argv.length < 2) {
program.help();
}
program.parse(process.argv);
function fileDemo() {
let audioPath = program["input_audio_file_path"];
let accessKey = program["access_key"]
let libraryFilePath = program["library_file_path"];
let modelFilePath = program["model_file_path"];
let disableAutomaticPunctuation = program["disable_automatic_punctuation"];
let disableSpeakerDiarization = program["disable_speaker_diarization"];
let verbose = program["verbose"];
let engineInstance = new Leopard(
accessKey,
{
'modelPath': modelFilePath,
'libraryPath': libraryFilePath,
'enableAutomaticPunctuation': !disableAutomaticPunctuation,
'enableDiarization': !disableSpeakerDiarization
}
);
if (!fs.existsSync(audioPath)) {
console.error(`--input_audio_file_path file not found: ${audioPath}`);
return;
}
try {
const res = engineInstance.processFile(audioPath);
console.log(res.transcript);
if (verbose) {
console.table(
res.words.map(word => {
return {
"Word": word.word,
"Start time (s)": word.startSec.toFixed(2),
"End time (s)": word.endSec.toFixed(2),
"Confidence": word.confidence.toFixed(2),
"Speaker Tag": word.speakerTag
};
})
);
}
} catch (err) {
if (err instanceof LeopardActivationLimitReached) {
console.error(`AccessKey '${accessKey}' has reached it's processing limit.`);
} else {
console.error(err);
}
}
engineInstance.release();
}
fileDemo();