RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix issue with parsing image info for sample info on some windows machines.

This commit is contained in:
Jaret Burkett 2025-11-19 08:45:05 -07:00
parent cd607c4902
commit 26e4b71b57
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
ui/src/components/SampleImageViewer.tsx
View File

@ -46,10 +46,21 @@ export default function SampleImageViewer({
const onCancel = useCallback(() => setIsOpen(false), []);
const imgInfo = useMemo(() => {
// handle windows C:\\Apps\\AI-Toolkit\\AI-Toolkit\\output\\LoRA-Name\\samples\\1763563000704__000004000_0.jpg
const ii = { filename: '', step: 0, promptIdx: 0 };
if (imgPath) {
const filename = imgPath.split('/').pop();
if (!filename) return ii;
// handle windows
let filename: string | null = null;
if (imgPath.includes('\\')) {
const parts = imgPath.split('\\');
filename = parts[parts.length - 1];
} else {
filename = imgPath.split('/').pop() || null;
}
if (!filename) {
console.error('Filename could not be determined from imgPath:', imgPath);
return ii;
}
ii.filename = filename;
const parts = filename
.split('.')[0]
@ -58,6 +69,8 @@ export default function SampleImageViewer({
if (parts.length === 3) {
ii.step = parseInt(parts[1]);
ii.promptIdx = parseInt(parts[2]);
} else {
console.error('Unexpected filename format for sample image:', filename);
}
}
return ii;