Skip to content
Commits on Source (2)
...@@ -23,7 +23,7 @@ module.exports = config => { ...@@ -23,7 +23,7 @@ module.exports = config => {
const uuidv1 = require('uuid/v1'); const uuidv1 = require('uuid/v1');
return { return {
transcript: (wavFile, enhancementlCallback) => { transcript: (wavFile, model, enhancementlCallback) => {
let port = config.port; let port = config.port;
if (process.env.OFFLINE_PORT !== undefined) { if (process.env.OFFLINE_PORT !== undefined) {
port = process.env.OFFLINE_PORT; port = process.env.OFFLINE_PORT;
...@@ -45,7 +45,8 @@ module.exports = config => { ...@@ -45,7 +45,8 @@ module.exports = config => {
filename: myUuidString, filename: myUuidString,
contentType: 'audio/x-wav' contentType: 'audio/x-wav'
} }
} },
model
}, },
encoding: null encoding: null
}; };
......
...@@ -33,7 +33,7 @@ const routesFactory = config => { ...@@ -33,7 +33,7 @@ const routesFactory = config => {
const stt = require('../controller/speech-to-text')(config.gstreamer); const stt = require('../controller/speech-to-text')(config.gstreamer);
const offline = require('../controller/offline')(config.offline); const offline = require('../controller/offline')(config.offline);
routes.post('/api/transcript', (req, res) => { routes.post('/api/transcript/:model*?', (req, res) => {
if (!req.body) { if (!req.body) {
return res.status(badRequestCode).send('No files were uploaded.'); return res.status(badRequestCode).send('No files were uploaded.');
} }
...@@ -41,6 +41,9 @@ const routesFactory = config => { ...@@ -41,6 +41,9 @@ const routesFactory = config => {
if (req.get('content-type') === undefined) { if (req.get('content-type') === undefined) {
return res.status(noContentTypeCode).send('No content type given'); return res.status(noContentTypeCode).send('No content type given');
} }
if (req.params.model === undefined) {
req.params.model = 'uc1';
}
enhancer.enhancement(req.body, (err, response, body) => { enhancer.enhancement(req.body, (err, response, body) => {
if (err) { if (err) {
...@@ -50,7 +53,7 @@ const routesFactory = config => { ...@@ -50,7 +53,7 @@ const routesFactory = config => {
} }
if (process.env.IS_OFFLINE === 'true' || (process.env.IS_OFFLINE === undefined && config.isOffline === true)) { if (process.env.IS_OFFLINE === 'true' || (process.env.IS_OFFLINE === undefined && config.isOffline === true)) {
offline.transcript(body, (err, response, body) => { offline.transcript(body, req.params.model, (err, response, body) => {
if (err) { if (err) {
return res.status(interalServerCode).json({module: 'Offline transcription', error: 'Error during audio transcription', info: err.toString()}); return res.status(interalServerCode).json({module: 'Offline transcription', error: 'Error during audio transcription', info: err.toString()});
} else if (response === undefined || body === undefined || response.statusCode !== successCode) { } else if (response === undefined || body === undefined || response.statusCode !== successCode) {
...@@ -64,7 +67,6 @@ const routesFactory = config => { ...@@ -64,7 +67,6 @@ const routesFactory = config => {
const jsonStr = json.transcript.transcription; const jsonStr = json.transcript.transcription;
let jsonResponse = {}; let jsonResponse = {};
try { try {
const transcription = JSON.parse(json.transcript.transcription); const transcription = JSON.parse(json.transcript.transcription);
......