UI Bug fixes and initial windows support

ui/src/app/api/gpu/route.ts

@@ -1,13 +1,18 @@
 import { NextResponse } from 'next/server';
 import { exec } from 'child_process';
 import { promisify } from 'util';
+import os from 'os';
 
 const execAsync = promisify(exec);
 
 export async function GET() {
   try {
+    // Get platform
+    const platform = os.platform();
+    const isWindows = platform === 'win32';
+    
     // Check if nvidia-smi is available
-    const hasNvidiaSmi = await checkNvidiaSmi();
+    const hasNvidiaSmi = await checkNvidiaSmi(isWindows);
 
     if (!hasNvidiaSmi) {
       return NextResponse.json({
@@ -18,7 +23,7 @@ export async function GET() {
     }
 
     // Get GPU stats
-    const gpuStats = await getGpuStats();
+    const gpuStats = await getGpuStats(isWindows);
 
     return NextResponse.json({
       hasNvidiaSmi: true,
@@ -37,20 +42,29 @@ export async function GET() {
   }
 }
 
-async function checkNvidiaSmi(): Promise<boolean> {
+async function checkNvidiaSmi(isWindows: boolean): Promise<boolean> {
   try {
-    await execAsync('which nvidia-smi');
+    if (isWindows) {
+      // Check if nvidia-smi is available on Windows
+      // It's typically located in C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe
+      // but we'll just try to run it directly as it may be in PATH
+      await execAsync('nvidia-smi -L');
+    } else {
+      // Linux/macOS check
+      await execAsync('which nvidia-smi');
+    }
     return true;
   } catch (error) {
     return false;
   }
 }
 
-async function getGpuStats() {
-  // Get detailed GPU information in JSON format including fan speed
-  const { stdout } = await execAsync(
-    'nvidia-smi --query-gpu=index,name,driver_version,temperature.gpu,utilization.gpu,utilization.memory,memory.total,memory.free,memory.used,power.draw,power.limit,clocks.current.graphics,clocks.current.memory,fan.speed --format=csv,noheader,nounits',
-  );
+async function getGpuStats(isWindows: boolean) {
+  // Command is the same for both platforms, but the path might be different
+  const command = 'nvidia-smi --query-gpu=index,name,driver_version,temperature.gpu,utilization.gpu,utilization.memory,memory.total,memory.free,memory.used,power.draw,power.limit,clocks.current.graphics,clocks.current.memory,fan.speed --format=csv,noheader,nounits';
+  
+  // Execute command
+  const { stdout } = await execAsync(command);
 
   // Parse CSV output
   const gpus = stdout
@@ -97,7 +111,7 @@ async function getGpuStats() {
           memory: parseInt(clockMemory),
         },
         fan: {
-          speed: parseInt(fanSpeed), // Fan speed as percentage
+          speed: parseInt(fanSpeed) || 0, // Some GPUs might not report fan speed, default to 0
         },
       };
     });

ui/src/app/api/jobs/[jobID]/start/route.ts

@@ -1,9 +1,10 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { PrismaClient } from '@prisma/client';
-import { TOOLKIT_ROOT, defaultTrainFolder } from '@/paths';
+import { TOOLKIT_ROOT } from '@/paths';
 import { spawn } from 'child_process';
 import path from 'path';
 import fs from 'fs';
+import os from 'os';
 import { getTrainingFolder, getHFToken } from '@/server/settings';
 
 const prisma = new PrismaClient();
@@ -51,9 +52,17 @@ export async function GET(request: NextRequest, { params }: { params: { jobID: s
   let pythonPath = 'python';
   // use .venv or venv if it exists
   if (fs.existsSync(path.join(TOOLKIT_ROOT, '.venv'))) {
-    pythonPath = path.join(TOOLKIT_ROOT, '.venv', 'bin', 'python');
+    if (os.platform() === 'win32') {
+      pythonPath = path.join(TOOLKIT_ROOT, '.venv', 'Scripts', 'python.exe');
+    } else {
+      pythonPath = path.join(TOOLKIT_ROOT, '.venv', 'bin', 'python');
+    }
   } else if (fs.existsSync(path.join(TOOLKIT_ROOT, 'venv'))) {
-    pythonPath = path.join(TOOLKIT_ROOT, 'venv', 'bin', 'python');
+    if (os.platform() === 'win32') {
+      pythonPath = path.join(TOOLKIT_ROOT, 'venv', 'Scripts', 'python.exe');
+    } else {
+      pythonPath = path.join(TOOLKIT_ROOT, 'venv', 'bin', 'python');
+    }
   }
 
   const runFilePath = path.join(TOOLKIT_ROOT, 'run.py');

ui/src/app/api/jobs/route.ts

@@ -6,6 +6,7 @@ const prisma = new PrismaClient();
 export async function GET(request: Request) {
   const { searchParams } = new URL(request.url);
   const id = searchParams.get('id');
+  console.log('ID:', id);
 
   try {
     if (id) {
@@ -18,6 +19,7 @@ export async function GET(request: Request) {
     const jobs = await prisma.job.findMany({
       orderBy: { created_at: 'desc' },
     });
+    console.log('Jobs:', jobs);
     return NextResponse.json({ jobs: jobs });
   } catch (error) {
     console.error(error);