-
Notifications
You must be signed in to change notification settings - Fork 0
/
transcription-service.js
68 lines (61 loc) · 1.67 KB
/
transcription-service.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
const EventEmitter = require("events");
const Speech = require("@google-cloud/speech");
const speech = new Speech.SpeechClient();
const request = {
config: {
encoding: "MULAW",
sampleRateHertz: 8000,
languageCode: "en-US",
},
interimResults: true,
};
class TranscriptionService extends EventEmitter {
constructor() {
super();
this.stream = null;
this.refreshStream = true;
// And make sure we refresh it every 60 seconds
this.refreshTimeout = setTimeout(() => {
console.log("Marking stream");
this.refreshStream = true;
}, 60000);
}
send(payload) {
this.getStream().write(payload);
}
close() {
if (this.stream) {
this.stream.destroy();
clearTimeout(this.refreshTimeout);
}
}
getStream() {
if (this.refreshStream) {
if (this.stream) {
this.stream.destroy();
}
console.log("Creating new stream");
this.stream = speech
.streamingRecognize(request)
.on("error", console.error)
.on("data", (data) => {
const result = data.results[0];
if (result === undefined || result.alternatives[0] === undefined) {
return;
}
if (result.isFinal === true) {
this.emit("transcription", result.alternatives[0].transcript);
if (this.refreshTimeout) {
this.refreshStream = true;
clearTimeout(this.refreshTimeout);
}
} else {
// console.log(`Partial result: ${result.alternatives[0].transcript}`);
}
});
this.refreshStream = false;
}
return this.stream;
}
}
module.exports = TranscriptionService;