Fix Regenerate Issues

Author: Keivan Vosoughi

app/client/src/pages/DataGenerator/Configure.tsx

@@ -57,10 +57,10 @@ const Configure = () => {
     useEffect(() => {
         if (wizardModeType === WizardModeType.DATA_AUGMENTATION) {
             setWizardModeType(WizardModeType.DATA_AUGMENTATION);
-            form.setFieldValue('workflow_type', 'custom');
+            form.setFieldValue('workflow_type', 'freeform');
         } else {
             setWizardModeType(WizardModeType.DATA_GENERATION);
-            form.setFieldValue('workflow_type', 'freeform');
+            form.setFieldValue('workflow_type', 'custom');
         }
     }, [location, wizardModeType]);
 
@@ -147,6 +147,8 @@ const Configure = () => {
         }
     }
 
+    console.log('Configure --> form', form.getFieldsValue());
+
     return (
         <StepContainer justify='center'>
             <FormContainer vertical>

app/client/src/pages/DataGenerator/DataGenerator.tsx

@@ -115,17 +115,40 @@ const DataGenerator: FunctionComponent<Props> = () => {
 
 
     useEffect(() => {
+        console.log('useEffect generate_file_name', generate_file_name);
         if (generate_file_name && !mutation.data) {
             mutation.mutate(generate_file_name);
         }
+        console.log('mutation.data', mutation.data);
+
         if (mutation.data && mutation?.data?.dataset) {
+            console.log('setting form values', {
+                ...initialData,
+                ...(mutation?.data?.dataset as any)
+            });
             form.setFieldsValue({
                 ...initialData,
                 ...(mutation?.data?.dataset as any)
             });
+            console.log('------> form', form.getFieldsValue());
         }
 
     }, [generate_file_name]);
+
+    useEffect(() => {
+        console.log('useEffect mutation.data', mutation.data);
+        if (mutation.data && mutation?.data?.dataset) {
+            console.log('setting form values', {
+                ...initialData,
+                ...(mutation?.data?.dataset as any)
+            });
+            form.setFieldsValue({
+                ...initialData,
+                ...(mutation?.data?.dataset as any)
+            });
+            console.log('------> form', form.getFieldsValue());
+        }
+    }, [mutation.data]);
 
 
     if (initialData?.technique) {
@@ -175,6 +198,9 @@ const DataGenerator: FunctionComponent<Props> = () => {
 
     const prev = () => setCurrent(Math.max(0, current - 1))
 
+    console.log('formData', formData.current);
+    console.log('form', form.getFieldsValue());
+
     return (
         <WizardCtx.Provider value={{ setIsStepValid }}>
             <Layout style={{ paddingBottom: 45 }}>

app/client/src/pages/DataGenerator/Examples.tsx

@@ -18,6 +18,7 @@ import FreeFormExampleTable from './FreeFormExampleTable';
 import { L } from 'vitest/dist/chunks/reporters.DTtkbAtP.js';
 import Loading from '../Evaluator/Loading';
 import { isEqual, set } from 'lodash';
+import { useParams } from 'react-router-dom';
 
 const { Title, Text } = Typography;
 const Container = styled.div`
@@ -52,9 +53,12 @@ const StyledContainer = styled.div`
 
 const Examples: FunctionComponent = () => {
     const form = Form.useFormInstance();
+    const { template_name, generate_file_name } = useParams();
     const [records, setRecords] = useState<Record<string, string>[]>([]);
     const workflowType = form.getFieldValue('workflow_type');
 
+    console.log('Examples >>> form:', form.getFieldsValue());
+
     const mutation = useMutation({
         mutationFn: fetchFileContent
     });