added route to download comment_gen and

code_refinment

src/routes/datasets.js

@@ -0,0 +1,34 @@
+import { Router } from 'express';
+import { join } from 'path';
+import { getProjectPath } from '../utils/paths.js';
+
+const router = Router();
+
+// Environment variables for paths (all relative to project root)
+const DATA_DIR = process.env.DATA_DIR ? getProjectPath(process.env.DATA_DIR) : getProjectPath('data');
+
+const DATASETS = [
+    "comment_generation",
+    "code_refinement"
+];
+
+router.get('/download/:dataset', async (req, res) => {
+    const { dataset } = req.params;
+    const withContext = req.query.withContext ? JSON.parse(req.query.withContext) : false;
+
+    if (!DATASETS.includes(dataset)) {
+        return res.status(400).json({ error: 'Invalid dataset name' });
+    }
+
+    const fileName = `${dataset}_${withContext ? 'with_context' : 'no_context'}.zip`;
+    const filePath = join(DATA_DIR, fileName);
+
+    try {
+        res.download(filePath);
+    } catch (error) {
+        console.error('Error serving file:', error);
+        res.status(500).json({ error: 'Error serving file' });
+    }
+});
+
+export default router;