Made it easy to add control images to the samples in the UI

ui/src/app/api/img/[...imagePath]/route.ts

@@ -2,7 +2,7 @@
 import { NextRequest, NextResponse } from 'next/server';
 import fs from 'fs';
 import path from 'path';
-import { getDatasetsRoot, getTrainingFolder } from '@/server/settings';
+import { getDatasetsRoot, getTrainingFolder, getDataRoot } from '@/server/settings';
 
 export async function GET(request: NextRequest, { params }: { params: { imagePath: string } }) {
   const { imagePath } = await params;
@@ -13,8 +13,9 @@ export async function GET(request: NextRequest, { params }: { params: { imagePat
     // Get allowed directories
     const datasetRoot = await getDatasetsRoot();
     const trainingRoot = await getTrainingFolder();
+    const dataRoot = await getDataRoot();
 
-    const allowedDirs = [datasetRoot, trainingRoot];
+    const allowedDirs = [datasetRoot, trainingRoot, dataRoot];
 
     // Security check: Ensure path is in allowed directory
     const isAllowed = allowedDirs.some(allowedDir => filepath.startsWith(allowedDir)) && !filepath.includes('..');

ui/src/app/api/img/upload/route.ts

@@ -0,0 +1,58 @@
+// src/app/api/datasets/upload/route.ts
+import { NextRequest, NextResponse } from 'next/server';
+import { writeFile, mkdir } from 'fs/promises';
+import { join } from 'path';
+import { getDataRoot } from '@/server/settings';
+import {v4 as uuidv4} from 'uuid';
+
+export async function POST(request: NextRequest) {
+  try {
+    const dataRoot = await getDataRoot();
+    if (!dataRoot) {
+      return NextResponse.json({ error: 'Data root path not found' }, { status: 500 });
+    }
+    const imgRoot = join(dataRoot, 'images');
+
+
+    const formData = await request.formData();
+    const files = formData.getAll('files');
+
+    if (!files || files.length === 0) {
+      return NextResponse.json({ error: 'No files provided' }, { status: 400 });
+    }
+
+    // make it recursive if it doesn't exist
+    await mkdir(imgRoot, { recursive: true });
+    const savedFiles = await Promise.all(
+      files.map(async (file: any) => {
+        const bytes = await file.arrayBuffer();
+        const buffer = Buffer.from(bytes);
+
+        const extension = file.name.split('.').pop() || 'jpg';
+
+        // Clean filename and ensure it's unique
+        const fileName = `${uuidv4()}`; // Use UUID for unique file names
+        const filePath = join(imgRoot, `${fileName}.${extension}`);
+
+        await writeFile(filePath, buffer);
+        return filePath;
+      }),
+    );
+
+    return NextResponse.json({
+      message: 'Files uploaded successfully',
+      files: savedFiles,
+    });
+  } catch (error) {
+    console.error('Upload error:', error);
+    return NextResponse.json({ error: 'Error uploading files' }, { status: 500 });
+  }
+}
+
+// Increase payload size limit (default is 4mb)
+export const config = {
+  api: {
+    bodyParser: false,
+    responseLimit: '50mb',
+  },
+};