Made it easy to add control images to the samples in the UI

2025-07-17 12:00:48 -06:00
parent e25d2feddf
commit 8610c6ed7f
16 changed files with 400 additions and 57 deletions
--- a/ui/src/components/SampleImages.tsx
+++ b/ui/src/components/SampleImages.tsx
@@ -14,7 +14,11 @@ export default function SampleImages({ job }: SampleImagesProps) {
    if (job?.job_config) {
      const jobConfig = JSON.parse(job.job_config) as JobConfig;
      const sampleConfig = jobConfig.config.process[0].sample;
-      return sampleConfig.prompts.length;
+      if (sampleConfig.prompts) {
+        return sampleConfig.prompts.length;
+      } else {
+        return sampleConfig.samples.length;
+      }
    }
    return 10;
  }, [job]);