Regenerate Issues using PDF Files

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]);
@@ -254,17 +262,32 @@ const Configure: FunctionComponent = () => {
                     </Select>
                 </Form.Item>
                 {(formData?.workflow_type === WorkflowType.SUPERVISED_FINE_TUNING || 
-                 formData?.workflow_type === WorkflowType.FREE_FORM_DATA_GENERATION) && 
+                 formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION) && 
                  <UseCaseSelector form={form} />}
 
                 {(
-                    formData?.workflow_type === WorkflowType.SUPERVISED_FINE_TUNING || 
-                    formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION) && 
+                    formData?.workflow_type === WorkflowType.FREE_FORM_DATA_GENERATION || 
+                    (formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION &&
+                    formData?.use_case === 'custom'))  && 
+                    <Form.Item
+                    noStyle
+                    shouldUpdate={(prevValues, currentValues) =>
+                      prevValues.doc_paths !== currentValues.doc_paths ||
+                      prevValues.use_case !== currentValues.use_case
+                    }
+                  >
+                    {({}) => {
+                           const useCase = form.getFieldValue('use_case');
+                           if (useCase === 'custom') {
+   
+                           }
+                      return (
+                     
                 <Form.Item
                     name='doc_paths'
-                    label='Input File'
+                    label={(useCase === 'custom' && formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION)  ? 'Context' : 'Input File'}
                     labelCol={labelCol}
-                    dependencies={['workflow_type']}
+                    dependencies={['workflow_type', 'use_case]']}
                     shouldUpdate
                     validateTrigger="['onBlur','onChange']"
                     tooltip='Select a file from your project that contains the initial data to be augmented.'
@@ -302,10 +325,10 @@ const Configure: FunctionComponent = () => {
                 >
                     <Flex>
                         <Select placeholder={'Select project files'} mode="multiple" value={selectedFiles || []} onChange={onFilesChange} allowClear/>    
-                        <FileSelectorButton onAddFiles={onAddFiles} workflowType={form.getFieldValue('workflow_type')} allowFileTypes={['pdf', 'docx']}/>
+                        <FileSelectorButton onAddFiles={onAddFiles} workflowType={form.getFieldValue('workflow_type')} allowFileTypes={['pdf', 'docx', 'json']}/>
                     </Flex>
-                </Form.Item>}
-                {formData?.workflow_type === WorkflowType.CUSTOM_DATA_GENERATION && 
+                </Form.Item>)}}</Form.Item>}
+                {formData?.workflow_type === WorkflowType.FREE_FORM_DATA_GENERATION && 
                 <>
                     <Form.Item
                         name='input_key'

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

@@ -149,7 +149,7 @@ const Examples: FunctionComponent = () => {
                     </Space>
                 </StyledTitle>
                 <Flex align='center' gap={15}>       
-                    {workflowType === WorkflowType.FREE_FORM_DATA_GENERATION && 
+                    {(workflowType === WorkflowType.FREE_FORM_DATA_GENERATION || workflowType === WorkflowType.CUSTOM_DATA_GENERATION) && 
                       <>
                         <Form.Item
                             name="example_path"

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

@@ -293,7 +293,7 @@ const Prompt = () => {
 
 
                     </div>
-                    {((workflow_type === WorkflowType.CUSTOM_DATA_GENERATION && !isEmpty(doc_paths)) ||
+                    {((workflow_type === WorkflowType.FREE_FORM_DATA_GENERATION && !isEmpty(doc_paths)) ||
                     (workflow_type === WorkflowType.SUPERVISED_FINE_TUNING && !isEmpty(doc_paths))) && 
                         <StyledFormItem
                             name={'num_questions'}

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;
 }