Samples work in ui now

ui/src/hooks/useJob.tsx

@@ -7,7 +7,7 @@ export default function useJob(jobID: string, reloadInterval: null | number = nu
   const [job, setJob] = useState<Job | null>(null);
   const [status, setStatus] = useState<'idle' | 'loading' | 'success' | 'error'>('idle');
 
-  const refreshJobs = () => {
+  const refreshJob = () => {
     setStatus('loading');
     fetch(`/api/jobs?id=${jobID}`)
       .then(res => res.json())
@@ -23,11 +23,11 @@ export default function useJob(jobID: string, reloadInterval: null | number = nu
   };
 
   useEffect(() => {
-    refreshJobs();
+    refreshJob();
 
     if (reloadInterval) {
       const interval = setInterval(() => {
-        refreshJobs();
+        refreshJob();
       }, reloadInterval);
 
       return () => {
@@ -36,5 +36,5 @@ export default function useJob(jobID: string, reloadInterval: null | number = nu
     }
   }, [jobID]);
 
-  return { job, setJob, status, refreshJobs };
+  return { job, setJob, status, refreshJob };
 }

ui/src/hooks/useSampleImages.tsx

@@ -0,0 +1,41 @@
+'use client';
+
+import { useEffect, useState } from 'react';
+import { Job } from '@prisma/client';
+
+export default function useSampleImages(jobID: string, reloadInterval: null | number = null) {
+  const [sampleImages, setSampleImages] = useState<string[]>([]);
+  const [status, setStatus] = useState<'idle' | 'loading' | 'success' | 'error'>('idle');
+
+  const refreshSampleImages = () => {
+    setStatus('loading');
+    fetch(`/api/jobs/${jobID}/samples`)
+      .then(res => res.json())
+      .then(data => {
+        if (data.samples) {
+          setSampleImages(data.samples);
+        }
+        setStatus('success');
+      })
+      .catch(error => {
+        console.error('Error fetching datasets:', error);
+        setStatus('error');
+      });
+  };
+
+  useEffect(() => {
+    refreshSampleImages();
+
+    if (reloadInterval) {
+      const interval = setInterval(() => {
+        refreshSampleImages();
+      }, reloadInterval);
+
+      return () => {
+        clearInterval(interval);
+      };
+    }
+  }, [jobID]);
+
+  return { sampleImages, setSampleImages, status, refreshSampleImages };
+}