From bf4007d66acd7414561c58b387e8c5f2577911d9 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Mon, 22 Dec 2025 16:01:16 -0700
Subject: [PATCH] Reset the wizard state after closing with model

---
 .../classification/ClassificationModelWizardDialog.tsx     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/src/components/classification/ClassificationModelWizardDialog.tsx b/web/src/components/classification/ClassificationModelWizardDialog.tsx
index 06bf1f850..0c43b9942 100644
--- a/web/src/components/classification/ClassificationModelWizardDialog.tsx
+++ b/web/src/components/classification/ClassificationModelWizardDialog.tsx
@@ -137,6 +137,11 @@ export default function ClassificationModelWizardDialog({
     onClose();
   };
 
+  const handleSuccessClose = () => {
+    dispatch({ type: "RESET" });
+    onClose();
+  };
+
   return (
     <Dialog
       open={open}
@@ -207,7 +212,7 @@ export default function ClassificationModelWizardDialog({
                 step1Data={wizardState.step1Data}
                 step2Data={wizardState.step2Data}
                 initialData={wizardState.step3Data}
-                onClose={onClose}
+                onClose={handleSuccessClose}
                 onBack={handleBack}
               />
             )}