Test

Author: Keivan Vosoughi

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

@@ -66,9 +66,16 @@ const Configure: FunctionComponent = () => {
         }
     }, [location, wizardModeType]);
 
+    console.log('wizardModeType', wizardModeType);
+
     useEffect(() => {
+        console.log('useEffect 2');
         if (template_name) {
-            form.setFieldValue('use_case', template_name);
+            setTimeout(() => {
+                console.log('setting template name');
+                form.setFieldValue('use_case', template_name);
+            }, 1000);
+            
         }
     }, [template_name]);
 
@@ -99,10 +106,11 @@ const Configure: FunctionComponent = () => {
 
 
     useEffect(() => {
+        console.log('useEffect 1');
         if (formData && formData?.inference_type === undefined && isEmpty(generate_file_name)) {
             form.setFieldValue('inference_type', ModelProviders.CAII);
             setTimeout(() => {
-                form.setFieldValue('use_case','custom');
+                form.setFieldValue('use_case', template_name ? template_name : 'custom');
             }, 1000);
         }
     }, [formData]);
@@ -259,6 +267,7 @@ const Configure: FunctionComponent = () => {
 
                 {(
                     formData?.workflow_type === WorkflowType.FREE_FORM_DATA_GENERATION || 
+                    formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION ||
                     formData?.use_case === 'custom')  && 
                     <Form.Item
                     noStyle

app/client/src/pages/DataGenerator/utils.ts

@@ -109,12 +109,11 @@ export const getHttpStatusCodeVerb = (statusCode: number) => {
 
 export const getWizardModeType = (location: any) => {
   const pathname = location?.pathname || '';
-  switch (pathname) {
-    case '/data-augmentation':
-      return WizardModeType.DATA_AUGMENTATION;
-    case '/data-generator':
-      return WizardModeType.DATA_GENERATION;
-    default:
-      return null;
+  console.log('pathname', pathname);
+  if (pathname.includes('/data-augmentation')) {
+    return WizardModeType.DATA_AUGMENTATION;
+  } else if (pathname.includes('/data-generator')) {
+    return WizardModeType.DATA_GENERATION;
   }
+  return null;
 }

app/client/src/pages/Home/TemplateCard.tsx

@@ -160,7 +160,7 @@ const TemplateCard: React.FC<Props> = ({ template }) => {
 
 
     return (
-        <StyledCard onClick={() => navigate(`/${Pages.GENERATOR}/${template.id}`)}>
+        <StyledCard onClick={() => navigate(`/${Pages.DATA_AUGMENTATION}/${template.id}`)}>
             <TopSection>
                 <StyledTitle>{template.name}</StyledTitle>
                 <Popover

app/client/src/routes.tsx

@@ -43,8 +43,8 @@ const router = createBrowserRouter([
         loader: async () => null
       },
       { 
-        path: `${Pages.GENERATOR}/:template_name`, 
-        element: <DataGenerator key={Pages.GENERATOR}/>, 
+        path: `${Pages.DATA_AUGMENTATION}/:template_name`, 
+        element: <DataGenerator key={Pages.DATA_AUGMENTATION}/>, 
         errorElement: <ErrorPage />,
         loader: async () => null
       },