karma/crab-webapp
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab-webapp.git synced 2025-07-05 06:08:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

formatted file

Browse Source
This commit is contained in:
Karma Riuk
2025-05-07 18:21:13 +02:00
parent 522bcc7ebc
commit 4200b5d857
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/routes/datasets.js
View File

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