Updated Dockerfile for Hugging Face Spaces deployment

Dockerfile

@@ -48,9 +48,7 @@ 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 && \
-    mkdir -p /app/public/audio && \
-    chmod -R 777 /app/cache && \
-    chmod -R 777 /app/public/audio
+    chmod -R 777 /app/cache
 
 # Set environment variables for cache directories
 ENV HF_HOME=/app/cache/huggingface
@@ -73,6 +71,7 @@ COPY . /app
 
 # Create necessary directories for Node.js uploads with proper permissions
 RUN mkdir -p /app/app/uploads && chmod -R 777 /app/app/uploads
+RUN mkdir -p /app/app/public && chmod -R 777 /app/app/public
 
 # Copy wait-for-it script
 COPY wait-for-it.sh /app/wait-for-it.sh

app/server.js

@@ -10,14 +10,20 @@ 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(path.join(__dirname, 'public')));
+app.use(express.static(publicDir));
 app.use(express.json());
 
 const getNextFolderNumber = () => {
@@ -71,7 +77,7 @@ app.post('/save-audio', upload.single('audio'), async (req, res) => {
                 audioPaths.push(result.audio_path);
                 sentenceIndex++;
                 // Serve the audio file from the public directory
-                const publicAudioPath = path.join(__dirname, 'public', path.basename(result.audio_path));
+                const publicAudioPath = path.join(publicDir, 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 {