From ad69b9e251ab9fd08dad2abd7f3a0552e7777d5f Mon Sep 17 00:00:00 2001
From: Keivan Vosoughi <kvosoughi@cloudera.com>
Date: Fri, 30 May 2025 19:08:02 -0700
Subject: [PATCH] File Upload for Seeds Instructions

---
 app/client/src/pages/DataGenerator/Prompt.tsx | 28 ++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/app/client/src/pages/DataGenerator/Prompt.tsx b/app/client/src/pages/DataGenerator/Prompt.tsx
index 931c9ce..334d6f5 100644
--- a/app/client/src/pages/DataGenerator/Prompt.tsx
+++ b/app/client/src/pages/DataGenerator/Prompt.tsx
@@ -12,10 +12,13 @@ import { usefetchTopics, useFetchDefaultSchema, useFetchDefaultPrompt } from '..
 import { MAX_NUM_QUESTION, MIN_SEED_INSTRUCTIONS,  MAX_SEED_INSTRUCTIONS } from './constants'
 import { Usecases, WorkflowType } from './types';
 import { useWizardCtx } from './utils';
-import { useDatasetSize, useGetPromptByUseCase } from './hooks';
+import { fetchFileContent, useDatasetSize, useGetPromptByUseCase } from './hooks';
 import CustomPromptButton from './CustomPromptButton';
 import get from 'lodash/get';
 import TextArea from 'antd/es/input/TextArea';
+import FileSelectorButton from './FileSelectorButton';
+import { useMutation } from '@tanstack/react-query';
+import first from 'lodash/first';
 
 const { Title } = Typography;
 
@@ -97,6 +100,16 @@ const Prompt = () => {
         input_value,
         output_key
     );
+    const mutation = useMutation({
+        mutationFn: fetchFileContent
+    });
+
+    useEffect(() => {   
+        if (!isEmpty(mutation.data)) {
+            setItems(mutation.data)
+        }
+    }, [mutation.data]);
+
 
     useEffect(() => {  
         if (isError) {
@@ -193,6 +206,16 @@ const Prompt = () => {
         setCustomTopic('');
     };
 
+    const onAddFiles = (files: File[]) => {
+        console.log('onAddFiles', files);
+        if (!isEmpty (files)) {
+            const file = first(files);
+            mutation.mutate({
+                path: get(file, '_path'),      
+            });
+        };
+    }
+
     return (
         <Row gutter={[50,0]}>
             <LeftCol span={17}>
@@ -302,6 +325,7 @@ const Prompt = () => {
                             shouldUpdate
                             // validateTrigger='onBlur'
                         >
+                            <Flex>
                             <Select
                                 allowClear
                                 mode="multiple"
@@ -368,6 +392,8 @@ const Prompt = () => {
                                     disabled: selectedTopics?.length === MAX_SEED_INSTRUCTIONS
                                 }))}
                             />
+                            <FileSelectorButton onAddFiles={onAddFiles} workflowType={form.getFieldValue('workflow_type')} />
+                            </Flex>
                         </StyledFormItem>
                         <StyledFormItem
                             name='num_questions'