Test

Author: Keivan Vosoughi

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

@@ -93,18 +93,15 @@ const DataGenerator: FunctionComponent<Props> = () => {
         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)
-            });
+            const dataset = mutation?.data?.dataset as any;
             form.setFieldsValue({
                 ...initialData,
-                ...(mutation?.data?.dataset as any)
+                ...dataset,
+                workflow_type: dataset.technique === 'freeform' ? 
+                    WorkflowType.FREE_FORM_DATA_GENERATION : WorkflowType.CUSTOM_DATA_GENERATION   
             });
-            console.log('------> form', form.getFieldsValue());
         }
 
     }, [generate_file_name]);

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

@@ -50,6 +50,7 @@ const StyledContainer = styled.div`
 
 `;
 
+
 const Examples: FunctionComponent = () => {
     const form = Form.useFormInstance();
     const { generate_file_name } = useParams();
@@ -68,9 +69,13 @@ const Examples: FunctionComponent = () => {
     });
 
     useEffect(() => {
-        const useCase = form.getFieldValue('use_case');
-        restore_mutation.mutate(useCase);
-    }, [form.getFieldValue('use_case')]);
+        if (isEmpty(generate_file_name)) {
+            const useCase = form.getFieldValue('use_case');
+            restore_mutation.mutate(useCase);
+        } else {
+            setRecords(form.getFieldValue('examples'));
+        }
+    }, [form.getFieldValue('use_case'), generate_file_name]);
 
 
     useEffect(() => {