Updated Dockerfile for Hugging Face Spaces deployment

app/server.js

@@ -10,17 +10,8 @@ const app = express();
 const port = 3000;
 
 const uploadsDir = path.join(__dirname, 'uploads');
-const publicDir = path.join(__dirname, 'public');
-const audioDir = path.join(publicDir, 'audio');
-
 if (!fs.existsSync(uploadsDir)) {
-    fs.mkdirSync(uploadsDir);
-}
-if (!fs.existsSync(publicDir)) {
-    fs.mkdirSync(publicDir);
-}
-if (!fs.existsSync(audioDir)) {
-    fs.mkdirSync(audioDir);
+    fs.mkdirSync(uploadsDir, { recursive: true });
 }
 
 const storage = multer.memoryStorage();
@@ -50,7 +41,6 @@ app.post('/save-audio', upload.single('audio'), async (req, res) => {
     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}`;
@@ -78,11 +68,12 @@ app.post('/save-audio', upload.single('audio'), async (req, res) => {
             if (response.ok) {
                 const result = await response.json();
                 console.log(result);
-                const generatedAudioPath = path.join(audioDir, path.basename(result.audio_path));
-                fs.renameSync(result.audio_path, generatedAudioPath);  // Move the file to the public directory
-                audioPaths.push(`/audio/${path.basename(generatedAudioPath)}`);
+                audioPaths.push(result.audio_path);
                 sentenceIndex++;
-                res.status(200).json({ audio_path: `/audio/${path.basename(generatedAudioPath)}`, translation: result.translation });
+                // Serve the audio file from the public directory
+                const publicAudioPath = path.join(__dirname, 'public', path.basename(result.audio_path));
+                fs.copyFileSync(result.audio_path, publicAudioPath);
+                res.status(200).json({ audio_path: `/public/${path.basename(result.audio_path)}`, translation: result.translation });
             } else {
                 console.error('Failed to process the file via FastAPI');
                 res.status(500).send('Failed to process the file via FastAPI');
@@ -96,7 +87,7 @@ app.post('/save-audio', upload.single('audio'), async (req, res) => {
 
 app.get('/concatenate-audio', (req, res) => {
     const folderPath = path.join(uploadsDir, getNextFolderNumber().toString());
-    const finalAudioPath = path.join(audioDir, 'final_audio.wav');
+    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) {
@@ -104,7 +95,7 @@ app.get('/concatenate-audio', (req, res) => {
             return res.status(500).send('Error concatenating audio files');
         }
 
-        res.status(200).json({ audio_path: `/audio/final_audio.wav` });
+        res.status(200).json({ audio_path: finalAudioPath });
     });
 });