From b3b990b6365dc692011bf8a146006af879cd4db8 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Thu, 23 Oct 2025 10:29:51 -0600
Subject: [PATCH] Add retry button for image generation

---
 .../classification/wizard/Step3ChooseExamples.tsx        | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/web/src/components/classification/wizard/Step3ChooseExamples.tsx b/web/src/components/classification/wizard/Step3ChooseExamples.tsx
index ab269e883..06bb2bbad 100644
--- a/web/src/components/classification/wizard/Step3ChooseExamples.tsx
+++ b/web/src/components/classification/wizard/Step3ChooseExamples.tsx
@@ -367,10 +367,13 @@ export default function Step3ChooseExamples({
             )}
           >
             {!unknownImages || unknownImages.length === 0 ? (
-              <div className="flex h-[40vh] items-center justify-center">
+              <div className="flex h-[40vh] flex-col items-center justify-center gap-4">
                 <p className="text-muted-foreground">
                   {t("wizard.step3.noImages")}
                 </p>
+                <Button onClick={generateExamples} variant="select">
+                  {t("wizard.step3.retryGenerate")}
+                </Button>
               </div>
             ) : allImagesClassified && isProcessing ? (
               <div className="flex h-[40vh] flex-col items-center justify-center gap-4">
@@ -388,9 +391,7 @@ export default function Step3ChooseExamples({
                       key={imageName}
                       className={cn(
                         "aspect-square cursor-pointer overflow-hidden rounded-lg border-2 bg-background transition-all",
-                        isSelected
-                          ? "border-selected ring-2 ring-selected"
-                          : "border-border hover:border-primary",
+                        isSelected && "border-selected ring-2 ring-selected",
                       )}
                       onClick={() => toggleImageSelection(imageName)}
                     >