Updated Dockerfile for Hugging Face Spaces deployment

Dockerfile

@@ -48,7 +48,9 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 RUN mkdir -p /app/cache/huggingface && \
     mkdir -p /app/cache/triton && \
     mkdir -p /app/cache/torch_extensions && \
-    chmod -R 777 /app/cache
+    mkdir -p /app/public/audio && \
+    chmod -R 777 /app/cache && \
+    chmod -R 777 /app/public/audio
 
 # Set environment variables for cache directories
 ENV HF_HOME=/app/cache/huggingface

app/server.js

@@ -10,8 +10,17 @@ 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, { recursive: true });
+    fs.mkdirSync(uploadsDir);
+}
+if (!fs.existsSync(publicDir)) {
+    fs.mkdirSync(publicDir);
+}
+if (!fs.existsSync(audioDir)) {
+    fs.mkdirSync(audioDir);
 }
 
 const storage = multer.memoryStorage();
@@ -69,9 +78,11 @@ app.post('/save-audio', upload.single('audio'), async (req, res) => {
             if (response.ok) {
                 const result = await response.json();
                 console.log(result);
-                audioPaths.push(result.audio_path);
+                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)}`);
                 sentenceIndex++;
-                res.status(200).json({ audio_path: result.audio_path, translation: result.translation });
+                res.status(200).json({ audio_path: `/audio/${path.basename(generatedAudioPath)}`, translation: result.translation });
             } else {
                 console.error('Failed to process the file via FastAPI');
                 res.status(500).send('Failed to process the file via FastAPI');
@@ -85,7 +96,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(folderPath, 'final_audio.wav');
+    const finalAudioPath = path.join(audioDir, 'final_audio.wav');
     const concatCommand = `ffmpeg -y -i "concat:${audioPaths.join('|')}" -acodec copy ${finalAudioPath}`;
     exec(concatCommand, (concatError, concatStdout, concatStderr) => {
         if (concatError) {
@@ -93,7 +104,7 @@ app.get('/concatenate-audio', (req, res) => {
             return res.status(500).send('Error concatenating audio files');
         }
 
-        res.status(200).json({ audio_path: finalAudioPath });
+        res.status(200).json({ audio_path: `/audio/final_audio.wav` });
     });
 });