Commit 92329ddb authored by Yoann HOUPERT's avatar Yoann HOUPERT

add model id to request

parent b2225d85
Pipeline #11511 passed with stage
in 26 minutes and 54 seconds
...@@ -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
}; };
......
...@@ -32,11 +32,15 @@ const routesFactory = config => { ...@@ -32,11 +32,15 @@ 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.');
} }
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) {
return res.status(interalServerCode).json({module: 'Enhancement', error: 'Error during audio enhancing', info: err.toString()}); return res.status(interalServerCode).json({module: 'Enhancement', error: 'Error during audio enhancing', info: err.toString()});
...@@ -45,7 +49,7 @@ const routesFactory = config => { ...@@ -45,7 +49,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) {
...@@ -58,10 +62,10 @@ const routesFactory = config => { ...@@ -58,10 +62,10 @@ const routesFactory = config => {
} }
const jsonStr = json.transcript.transcription; const jsonStr = json.transcript.transcription;
const transcription = JSON.parse(json.transcript.transcription);
let jsonResponse = {}; let jsonResponse = {};
try { try {
const transcription = JSON.parse(json.transcript.transcription);
jsonResponse = { jsonResponse = {
status: 0, status: 0,
hypotheses: transcription.hypotheses, hypotheses: transcription.hypotheses,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment