update folder creation

ui/src/server/settings.ts

@@ -2,9 +2,32 @@ import { PrismaClient } from '@prisma/client';
 import { defaultDatasetsFolder, defaultDataRoot } from '@/paths';
 import { defaultTrainFolder } from '@/paths';
 import NodeCache from 'node-cache';
+import { usingBrowserDb } from '@/utils/env';
 
 const myCache = new NodeCache();
-const prisma = new PrismaClient();
+
+const isBrowserMode = usingBrowserDb || (process.env.DB_MODE || '').toLowerCase() === 'browser';
+
+const prisma = !isBrowserMode ? new PrismaClient() : null;
+
+const safeSettingLookup = async (key: string) => {
+  if (!prisma) {
+    return null;
+  }
+  try {
+    return await prisma.settings.findFirst({ where: { key } });
+  } catch (error) {
+    console.error(`Prisma settings lookup failed for key ${key}:`, error);
+    return null;
+  }
+};
+
+const envOrDefault = (envValue: string | undefined, fallback: string) => {
+  if (envValue && envValue.trim() !== '') {
+    return envValue.trim();
+  }
+  return fallback;
+};
 
 export const flushCache = () => {
   myCache.flushAll();
@@ -16,14 +39,14 @@ export const getDatasetsRoot = async () => {
   if (datasetsPath) {
     return datasetsPath;
   }
-  let row = await prisma.settings.findFirst({
-    where: {
-      key: 'DATASETS_FOLDER',
-    },
-  });
-  datasetsPath = defaultDatasetsFolder;
-  if (row?.value && row.value !== '') {
-    datasetsPath = row.value;
+  if (isBrowserMode) {
+    datasetsPath = envOrDefault(process.env.DATASETS_FOLDER, defaultDatasetsFolder);
+  } else {
+    const row = await safeSettingLookup('DATASETS_FOLDER');
+    datasetsPath = envOrDefault(process.env.DATASETS_FOLDER, defaultDatasetsFolder);
+    if (row?.value && row.value !== '') {
+      datasetsPath = row.value;
+    }
   }
   myCache.set(key, datasetsPath);
   return datasetsPath as string;
@@ -35,14 +58,14 @@ export const getTrainingFolder = async () => {
   if (trainingRoot) {
     return trainingRoot;
   }
-  let row = await prisma.settings.findFirst({
-    where: {
-      key: key,
-    },
-  });
-  trainingRoot = defaultTrainFolder;
-  if (row?.value && row.value !== '') {
-    trainingRoot = row.value;
+  if (isBrowserMode) {
+    trainingRoot = envOrDefault(process.env.TRAINING_FOLDER, defaultTrainFolder);
+  } else {
+    const row = await safeSettingLookup(key);
+    trainingRoot = envOrDefault(process.env.TRAINING_FOLDER, defaultTrainFolder);
+    if (row?.value && row.value !== '') {
+      trainingRoot = row.value;
+    }
   }
   myCache.set(key, trainingRoot);
   return trainingRoot as string;
@@ -54,14 +77,14 @@ export const getHFToken = async () => {
   if (token) {
     return token;
   }
-  let row = await prisma.settings.findFirst({
-    where: {
-      key: key,
-    },
-  });
-  token = '';
-  if (row?.value && row.value !== '') {
-    token = row.value;
+  if (isBrowserMode) {
+    token = envOrDefault(process.env.HF_TOKEN, '');
+  } else {
+    const row = await safeSettingLookup(key);
+    token = envOrDefault(process.env.HF_TOKEN, '');
+    if (row?.value && row.value !== '') {
+      token = row.value;
+    }
   }
   myCache.set(key, token);
   return token;
@@ -73,14 +96,14 @@ export const getDataRoot = async () => {
   if (dataRoot) {
     return dataRoot;
   }
-  let row = await prisma.settings.findFirst({
-    where: {
-      key: key,
-    },
-  });
-  dataRoot = defaultDataRoot;
-  if (row?.value && row.value !== '') {
-    dataRoot = row.value;
+  if (isBrowserMode) {
+    dataRoot = envOrDefault(process.env.DATA_ROOT, defaultDataRoot);
+  } else {
+    const row = await safeSettingLookup(key);
+    dataRoot = envOrDefault(process.env.DATA_ROOT, defaultDataRoot);
+    if (row?.value && row.value !== '') {
+      dataRoot = row.value;
+    }
   }
   myCache.set(key, dataRoot);
   return dataRoot;