IO-3515 add client side polling for now, cost centers.

server/ai/bill-ocr/bill-ocr.js

@@ -111,7 +111,7 @@ async function handleBillOcr(request, response) {
             } else {
                 // Start the Textract job (non-blocking) for multi-page documents
                 console.log('PDF => 2+ pages, processing asynchronously');
-                const jobInfo = await startTextractJob(uploadedFile.buffer);
+                const jobInfo = await startTextractJob(uploadedFile.buffer, { jobid, bodyshopid, partsorderid });
 
                 response.status(202).send({
                     success: true,
@@ -154,13 +154,50 @@ async function handleBillOcrStatus(request, response) {
     }
 
     if (jobStatus.status === 'COMPLETED') {
-        //TODO: This needs to be stored in the redis cache and pulled when it's processed. 
-        //const billForm = await generateBillFormData({ jobid, bodyshopid, partsorderid });
-
+        // Generate billForm on-demand if not already generated
+        let billForm = jobStatus.data?.billForm;
+        
+        if (!billForm && jobStatus.context) {
+            try {
+                console.log('Generating bill form data on-demand...');
+                billForm = await generateBillFormData({ 
+                    processedData: jobStatus.data, 
+                    jobid: jobStatus.context.jobid,
+                    bodyshopid: jobStatus.context.bodyshopid,
+                    partsorderid: jobStatus.context.partsorderid,
+                    req: request // Now we have request context!
+                });
+                
+                // Cache the billForm back to Redis for future requests
+                await setTextractJob({
+                    redisPubClient,
+                    textractJobId,
+                    jobData: {
+                        ...jobStatus,
+                        data: {
+                            ...jobStatus.data,
+                            billForm
+                        }
+                    }
+                });
+            } catch (error) {
+                console.error('Error generating bill form data:', error);
+                response.status(500).send({
+                    status: 'COMPLETED',
+                    error: 'Data processed but failed to generate bill form',
+                    message: error.message,
+                    data: jobStatus.data // Still return the raw processed data
+                });
+                return;
+            }
+        }
+        
         response.status(200).send({
             status: 'COMPLETED',
-            data: jobStatus.data
-            // data: { ...jobStatus.data, billForm }
+            data: {
+                ...jobStatus.data,
+                billForm
+            }
         });
     } else if (jobStatus.status === 'FAILED') {
         response.status(500).send({
@@ -199,7 +236,7 @@ async function processSinglePageDocument(pdfBuffer) {
     };
 }
 
-async function startTextractJob(pdfBuffer) {
+async function startTextractJob(pdfBuffer, context = {}) {
     // Upload PDF to S3 temporarily for Textract async processing
     const s3Bucket = process.env.AWS_AI_BUCKET;
     const snsTopicArn = process.env.AWS_TEXTRACT_SNS_TOPIC_ARN;
@@ -254,7 +291,8 @@ async function startTextractJob(pdfBuffer) {
                 status: 'IN_PROGRESS',
                 s3Key: s3Key,
                 uploadId: uploadId,
-                startedAt: new Date().toISOString()
+                startedAt: new Date().toISOString(),
+                context: context // Store the context for later use
             }
         }
     );
@@ -349,6 +387,7 @@ async function handleTextractNotification(message) {
         // Retrieve the results
         const { processedData, originalResponse } = await retrieveTextractResults(textractJobId);
 
+        // Store the processed data - billForm will be generated on-demand in the status endpoint
         await setTextractJob(
             {
                 redisPubClient,