commit disable fixes

Author: NirajC-Microsoft

src/frontend/src/components/ChatHistory/ChatHistoryList.test.tsx

@@ -28,6 +28,7 @@ const mockState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading : false,
+  isLoadedSections : []
 };
 
 const renderChatHistoryList = (stateOverride = {}) => {

src/frontend/src/components/ChatHistory/ChatHistoryPanel.test.tsx

@@ -55,6 +55,7 @@ const mockState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading : false,
+  isLoadedSections: []
 };
 
 const mockDispatch = jest.fn();

src/frontend/src/components/DraftCards/SectionCard.test.tsx

@@ -57,6 +57,7 @@ const mockState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading : false,
+  isLoadedSections: []
 }
 
 const renderWithContext = (idx = 0) =>
@@ -210,4 +211,20 @@ describe('SectionCard Component', () => {
       }
     })
   })
+//////
+  it('displays pre-filled section content', async () => {
+    mockState.draftedDocument.sections[0].content = 'Pre-filled content'
+  
+    renderWithContext()
+  
+    await waitFor(() => {
+      expect(screen.getByText('Pre-filled content')).toBeInTheDocument()
+      expect(sectionGenerate).not.toHaveBeenCalled()
+    })
+  })
+
+  ////
+
+  
+
 })

src/frontend/src/components/DraftCards/SectionCard.tsx

@@ -12,6 +12,7 @@ import { useLocation } from 'react-router-dom'
 
 interface SectionCardProps {
   sectionIdx: number
+ // onLoadingChange: (isLoading: boolean) => void
 }
 
 const useStyles = makeStyles({
@@ -111,7 +112,7 @@ const useStyles = makeStyles({
   }
 })
 
-const SectionCard = ({ sectionIdx }: SectionCardProps) => {
+const SectionCard = ({ sectionIdx  }: SectionCardProps) => {
   const location = useLocation()
   const classes = useStyles()
   const [isLoading, setIsLoading] = useState(false)
@@ -157,6 +158,7 @@ const SectionCard = ({ sectionIdx }: SectionCardProps) => {
 
   async function fetchSectionContent(sectionTitle: string, sectionDescription: string , isReqFrom = '') {
     setIsLoading(true)
+    //onLoadingChange(true)
     const sectionGenerateRequest: SectionGenerateRequest = { sectionTitle, sectionDescription }
 
     const response = await sectionGenerate(sectionGenerateRequest)
@@ -192,9 +194,12 @@ const SectionCard = ({ sectionIdx }: SectionCardProps) => {
 
       setCharCount(content.length)
       setIsLoading(false)
-
+      //onLoadingChange(false)
       appStateContext?.dispatch({ type: 'REMOVED_FAILED_SECTION', payload: {section : updatedSection} })
 
+      appStateContext?.dispatch({ type: 'UPDATE_IS_LOADED_SECTIONS', payload: {section : updatedSection} })
+
+
       if(isReqFrom == 'failed')
         appStateContext?.dispatch({ type: 'UPDATE_SECTION_API_REQ_STATUS', payload:  false })
     }

src/frontend/src/components/Sidebar/Sidebar.test.tsx

@@ -55,6 +55,7 @@ const mockState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading : false,
+  isLoadedSections : []
 };
 const mockState2 = {
   isChatHistoryOpen: false,

src/frontend/src/pages/draft/Draft.test.tsx

@@ -175,4 +175,5 @@ describe('Draft Component', () => {
     expect(screen.getByDisplayValue('Sample Draft')).toBeInTheDocument()
   })
 
+ 
 })

src/frontend/src/pages/draft/Draft.tsx

@@ -18,22 +18,50 @@ const Draft = (): JSX.Element => {
   const draftedDocument = appStateContext?.state.draftedDocument
   const sections = draftedDocument?.sections ?? []
 
+  const isLoadedSections = appStateContext?.state.isLoadedSections
+  
+
   const [sectionItems , setSectionItems] = useState<Section[]>([])
   const aiWarningLabel = 'AI-generated content may be incorrect'
 
+
+  const [isExportButtonDisable, setIsExportButtonDisable] = useState<boolean>(false)
+
+  
+  
   // redirect to home page if draftedDocument is empty
 
+  
+
   useEffect(() => {
+    
     sections.forEach((item, index) => {
       setTimeout(() => {
         setSectionItems((prev) => [...prev, item]); 
       }, index * 500); 
     });
+    
+      //appStateContext?.dispatch({ type: 'UPDATE_IS_LOADED_SECTIONS', payload: {section : null, 'act': 'removeAll' } })
+  
   }, []); 
 
   useEffect(()=>{
-    console.log("sectionItems", sectionItems)
-  },[sectionItems])
+    if(isLoadedSections?.length === sections.length)
+    {
+      setIsExportButtonDisable(false);
+    }
+    else{
+      setIsExportButtonDisable(true);
+    }
+
+  },[isLoadedSections])
+
+
+  useEffect(()=>{
+    console.log("is loaded?", isLoadedSections)
+
+    console.log("sectionItems", sectionItems )
+  },[sectionItems, isLoadedSections])
 
   if (!draftedDocument) {
     navigate('/')
@@ -113,6 +141,35 @@ const Draft = (): JSX.Element => {
     return title.replace(/[^a-zA-Z0-9]/g, '')
   }
 
+  // const handleLoadingChange = (isSectionLoading: boolean) => {
+  //   setLoadingCount((prev) => {
+  //     const newCount = isSectionLoading ? prev + 1 : prev - 1
+  //     return newCount
+  //   })
+  // }
+  // useEffect(() => {
+  //   const failedSections = appStateContext?.state?.failedSections ?? []
+  //   if (failedSections.length > 0) {
+      
+  //     setTimeout(() => {
+  //       setIsFailLoading(true);
+  //     }, 5000);
+      
+  //     if(failedSections.length>1)
+  //     {
+  //       setTimeout(() => {
+  //         setLoadingCount(0)
+  //       },5000);
+  //     }
+  //   }
+  //   else
+  //   {
+  //     setTimeout(() => {
+  //     setIsFailLoading(false)
+  //   }, 5000);
+  //   }
+  // }, [appStateContext?.state?.failedSections])
+
   return (
     <Stack className={styles.container}>
       <TitleCard />
@@ -142,6 +199,7 @@ const Draft = (): JSX.Element => {
           onClick={exportToWord}
           aria-label="export document"
           text="Export Document"
+          disabled = {isExportButtonDisable}
         />
       </Stack>
     </Stack>

src/frontend/src/state/AppProvider.tsx

@@ -34,6 +34,7 @@ export interface AppState {
   failedSections : Section[],
   isFailedReqInitiated : boolean,
   isLoading: boolean,
+  isLoadedSections: Section[]
 }
 
 export type Action =
@@ -66,7 +67,8 @@ export type Action =
   | { type: 'REMOVED_FAILED_SECTION'; payload: {section : Section} }
   | { type: 'UPDATE_SECTION_API_REQ_STATUS'; payload: boolean }
   | { type: 'SET_LOADING'; payload: boolean }
-
+  | { type: 'UPDATE_IS_LOADED_SECTIONS'; payload: {section : Section | null ,'act'?: string} }
+  
 const initialState: AppState = {
   isChatHistoryOpen: false,
   chatHistoryLoadingState: ChatHistoryLoadingState.Loading,
@@ -88,6 +90,8 @@ const initialState: AppState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading: false,
+  isLoadedSections: []
+
 }
 
 export const AppStateContext = createContext<

src/frontend/src/state/AppReducer.tsx

@@ -119,6 +119,18 @@ export const appStateReducer = (state: AppState, action: Action): AppState => {
       return { ...state , failedSections : [...tempFailedSections] }
     case 'UPDATE_SECTION_API_REQ_STATUS' : 
       return {...state, isFailedReqInitiated : action.payload}
+    
+    case 'UPDATE_IS_LOADED_SECTIONS' :
+      let tempLoadedSection:any = [];
+      if(action.payload.act === 'removeAll')
+      {
+         tempLoadedSection = [];
+      }else
+      {
+        tempLoadedSection = [...state.isLoadedSections];
+        tempLoadedSection.push(action.payload.section)
+      }
+      return {...state, isLoadedSections: tempLoadedSection}
 
     default:
       return state

src/frontend/src/test/test.utils.tsx

@@ -22,6 +22,7 @@ const defaultMockState = {
   failedSections : [],
   isFailedReqInitiated : false,
   isLoading : false,
+  isLoadedSections : []
 };
 
 // Create a custom render function