Spaces:
Sleeping
Sleeping
File size: 4,666 Bytes
aa7cb02 b941e8b aa7cb02 23761b4 aa7cb02 b941e8b aa7cb02 b941e8b aa7cb02 23761b4 aa7cb02 38ba82a e6de97c 38ba82a e6de97c 38ba82a aa7cb02 c276d81 aa7cb02 23761b4 aa7cb02 23761b4 aa7cb02 |
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 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
const express = require('express');
const multer = require('multer');
const path = require('path');
const fs = require('fs');
const { exec } = require('child_process');
const fetch = require('node-fetch');
const FormData = require('form-data');
const app = express();
const port = 3000;
const uploadsDir = path.join(__dirname, 'uploads');
const publicDir = path.join(__dirname, 'public');
if (!fs.existsSync(uploadsDir)) {
fs.mkdirSync(uploadsDir, { recursive: true });
}
if (!fs.existsSync(publicDir)) {
fs.mkdirSync(publicDir, { recursive: true });
}
const storage = multer.memoryStorage();
const upload = multer({ storage: storage });
app.use(express.static(publicDir));
app.use(express.json());
const getNextFolderNumber = () => {
const folders = fs.readdirSync(uploadsDir).filter(file => fs.statSync(path.join(uploadsDir, file)).isDirectory());
const folderNumbers = folders.map(folder => parseInt(folder)).filter(num => !isNaN(num));
return folderNumbers.length > 0 ? Math.max(...folderNumbers) + 1 : 1;
};
let sentenceIndex = 0;
let audioPaths = [];
app.post('/save-audio', upload.single('audio'), async (req, res) => {
const nextFolderNumber = getNextFolderNumber();
const folderPath = path.join(uploadsDir, nextFolderNumber.toString());
if (!fs.existsSync(folderPath)) {
fs.mkdirSync(folderPath, { recursive: true });
}
const rawAudioPath = path.join(folderPath, `audio_${sentenceIndex}.webm`);
const wavAudioPath = path.join(folderPath, `audio_${sentenceIndex}.wav`);
const transcriptionPath = path.join(folderPath, `transcription_${sentenceIndex}.txt`);
fs.writeFileSync(rawAudioPath, req.file.buffer);
fs.writeFileSync(transcriptionPath, req.body.transcript);
const ffmpegCommand = `ffmpeg -i ${rawAudioPath} -ar 44100 -ac 1 ${wavAudioPath}`;
exec(ffmpegCommand, async (error, stdout, stderr) => {
if (error) {
console.error(`Error converting audio to WAV: ${stderr}`);
return res.status(500).send('Error converting audio to WAV');
}
fs.unlinkSync(rawAudioPath);
const formData = new FormData();
formData.append('original_path', fs.createReadStream(wavAudioPath));
formData.append('text', req.body.transcript);
formData.append('lang', 'en');
formData.append('target_lang', 'es');
try {
const response = await fetch('http://localhost:8000/process-audio/', {
method: 'POST',
body: formData,
headers: formData.getHeaders()
});
if (response.ok) {
const result = await response.json();
console.log(result);
audioPaths.push(result.audio_path);
sentenceIndex++;
// Copy the audio file to the public directory
const publicAudioPath = path.join(publicDir, `generated_audio_${sentenceIndex}.wav`);
fs.copyFile(result.audio_path, publicAudioPath, (err) => {
if (err) {
console.error('Error copying audio file to public directory:', err);
return res.status(500).send('Error copying audio file to public directory');
}
res.status(200).json({ audio_path: `generated_audio_${sentenceIndex}.wav`, translation: result.translation });
});
} else {
console.error('Failed to process the file via FastAPI');
res.status(500).send('Failed to process the file via FastAPI');
}
} catch (error) {
console.error('Error calling FastAPI:', error);
res.status(500).send('Error calling FastAPI');
}
});
});
app.get('/download-audio', (req, res) => {
const filePath = path.join(publicDir, req.query.file_path);
res.download(filePath);
});
app.get('/concatenate-audio', (req, res) => {
const folderPath = path.join(uploadsDir, getNextFolderNumber().toString());
const finalAudioPath = path.join(folderPath, 'final_audio.wav');
const concatCommand = `ffmpeg -y -i "concat:${audioPaths.join('|')}" -acodec copy ${finalAudioPath}`;
exec(concatCommand, (concatError, concatStdout, concatStderr) => {
if (concatError) {
console.error(`Error concatenating audio files: ${concatStderr}`);
return res.status(500).send('Error concatenating audio files');
}
res.status(200).json({ audio_path: finalAudioPath });
});
});
app.listen(port, () => {
console.log(`Server running at http://localhost:${port}`);
});
|