Spaces:
Running
Running
| 'use client'; | |
| import { useEffect, useState } from 'react'; | |
| import { apiClient } from '@/utils/api'; | |
| 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'); | |
| apiClient | |
| .get(`/api/jobs/${jobID}/samples`) | |
| .then(res => res.data) | |
| .then(data => { | |
| console.log('Fetched sample images:', 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 }; | |
| } | |