fix: Enhance error handling and add debug message support in report processing

Author: adamrefaey

backend/src/document-processor/controllers/document-processor.controller.ts

@@ -182,20 +182,41 @@ export class DocumentProcessorController {
       this.logger.log(`Started async processing for report: ${reportId}`);
 
       // Get the file from S3
-      const fileBuffer = await this.getFileFromS3(filePath);
+      let fileBuffer;
+      try {
+        fileBuffer = await this.getFileFromS3(filePath);
+        this.logger.log(`Successfully retrieved file from S3 for report: ${reportId}`);
+      } catch (error) {
+        this.logger.error(
+          `Failed to retrieve file from S3 for report ${reportId}: ${error instanceof Error ? error.message : 'Unknown error'}`,
+        );
+        await this.updateReportStatus(reportId, userId, ProcessingStatus.FAILED);
+        return;
+      }
 
       // Process the document
-      const result = await this.documentProcessorService.processDocument(fileBuffer, userId);
+      let result;
+      try {
+        result = await this.documentProcessorService.processDocument(fileBuffer, userId);
+        this.logger.log(`Successfully processed document for report: ${reportId}`);
+      } catch (error) {
+        this.logger.error(
+          `Failed to process document for report ${reportId}: ${error instanceof Error ? error.message : 'Unknown error'}`,
+        );
+        await this.updateReportStatus(reportId, userId, ProcessingStatus.FAILED);
+        return;
+      }
 
       // Fetch the report again to ensure we have the latest version
       const report = await this.reportsService.findOne(reportId, userId);
       if (!report) {
-        throw new Error(`Report ${reportId} not found during async processing`);
+        this.logger.error(`Report ${reportId} not found during async processing`);
+        return;
       }
 
       // Update the report with analysis results
-      report.title = result.analysis.title || 'Untitled Report';
-      report.category = result.analysis.category || 'general';
+      report.title = result.analysis.title;
+      report.category = result.analysis.category;
       report.processingStatus = ProcessingStatus.PROCESSED;
 
       // Extract lab values
@@ -214,25 +235,40 @@ export class DocumentProcessorController {
       this.logger.log(`Completed async processing for report: ${reportId}`);
     } catch (error) {
       // If processing fails, update the report status to indicate failure
-      try {
-        const report = await this.reportsService.findOne(reportId, userId);
-        if (report) {
-          report.processingStatus = ProcessingStatus.FAILED;
-          report.updatedAt = new Date().toISOString();
-          await this.reportsService.updateReport(report);
-        }
-      } catch (updateError: unknown) {
-        this.logger.error(
-          `Failed to update report status after processing error: ${
-            updateError instanceof Error ? updateError.message : 'Unknown error'
-          }`,
-        );
-      }
-
       this.logger.error(
         `Error during async processing for report ${reportId}: ${error instanceof Error ? error.message : 'Unknown error'}`,
       );
-      throw error;
+      await this.updateReportStatus(reportId, userId, ProcessingStatus.FAILED);
+    }
+  }
+
+  /**
+   * Updates a report's processing status
+   * @param reportId - ID of the report to update
+   * @param userId - ID of the user who owns the report
+   * @param status - The new processing status
+   */
+  private async updateReportStatus(
+    reportId: string,
+    userId: string,
+    status: ProcessingStatus,
+    debugMessage: string | undefined = undefined,
+  ): Promise<void> {
+    try {
+      const report = await this.reportsService.findOne(reportId, userId);
+      if (report) {
+        report.processingStatus = status;
+        report.updatedAt = new Date().toISOString();
+        report.debugMessage = debugMessage;
+        await this.reportsService.updateReport(report);
+        this.logger.log(`Updated status of report ${reportId} to ${status}`);
+      }
+    } catch (updateError: unknown) {
+      this.logger.error(
+        `Failed to update report status after processing error: ${
+          updateError instanceof Error ? updateError.message : 'Unknown error'
+        }`,
+      );
     }
   }
 

backend/src/reports/models/report.model.ts

@@ -74,4 +74,7 @@ export class Report {
 
   @ApiProperty({ description: 'Last update timestamp' })
   updatedAt: string;
+
+  @ApiProperty({ description: 'Optional debug message for the report' })
+  debugMessage?: string;
 }

frontend/src/common/models/medicalReport.ts

@@ -58,4 +58,5 @@ export interface MedicalReport {
   fileSize: number;
   createdAt: string; // ISO date string
   updatedAt: string; // ISO date string
+  debugMessage?: string; // Optional debug message for the report
 }