Add title and category fields to MedicalDocumentAnalysis and update tests

Author: adamrefaey

backend/src/document-processor/services/aws-bedrock.service.spec.ts

@@ -65,6 +65,8 @@ vi.mock('@aws-sdk/client-bedrock-runtime', () => {
                 {
                   type: 'text',
                   text: JSON.stringify({
+                    title: 'Blood Test Results',
+                    category: 'general',
                     keyMedicalTerms: [
                       { term: 'RBC', definition: 'Red Blood Cells' },
                       { term: 'WBC', definition: 'White Blood Cells' },
@@ -126,6 +128,8 @@ describe('AwsBedrockService', () => {
   `;
 
   const mockMedicalAnalysis: MedicalDocumentAnalysis = {
+    title: 'Blood Test Results',
+    category: 'general',
     keyMedicalTerms: [
       { term: 'RBC', definition: 'Red Blood Cells' },
       { term: 'WBC', definition: 'White Blood Cells' },
@@ -309,6 +313,8 @@ describe('AwsBedrockService', () => {
 
       // Test a valid response
       const validResponse: MedicalDocumentAnalysis = {
+        title: 'Test Report',
+        category: 'general',
         keyMedicalTerms: [],
         labValues: [],
         diagnoses: [],

backend/src/document-processor/services/aws-bedrock.service.ts

@@ -12,6 +12,8 @@ import { createHash } from 'crypto';
  * Interface for medical document analysis result
  */
 export interface MedicalDocumentAnalysis {
+  title: string;
+  category: string;
   keyMedicalTerms: Array<{ term: string; definition: string }>;
   labValues: Array<{
     name: string;
@@ -44,15 +46,19 @@ export class AwsBedrockService {
   private readonly medicalAnalysisPrompt = `Please analyze this medical document carefully, with specific attention to medical lab reports.
 
 Look for and extract the following information:
-1. Key medical terms visible in the document with their definitions
-2. Lab test values with their normal ranges and whether they are abnormal (particularly important for blood work, metabolic panels, etc.)
-3. Any diagnoses, findings, or medical observations with details and recommendations
-4. Analyze if this is a medical document (lab report, test result, medical chart, prescription, etc.) and provide confidence level
+1. Document title or main subject based on content
+2. Document category based on organ system focus
+3. Key medical terms visible in the document with their definitions
+4. Lab test values with their normal ranges and whether they are abnormal (particularly important for blood work, metabolic panels, etc.)
+5. Any diagnoses, findings, or medical observations with details and recommendations
+6. Analyze if this is a medical document (lab report, test result, medical chart, prescription, etc.) and provide confidence level
 
 This document may be a lab report showing blood work or other test results, so please pay special attention to tables, numeric values, reference ranges, and medical terminology.
 
 Format the response as a JSON object with the following structure:
 {
+  "title": string,
+  "category": string,
   "keyMedicalTerms": [{"term": string, "definition": string}],
   "labValues": [{"name": string, "value": string, "unit": string, "normalRange": string, "isAbnormal": boolean}],
   "diagnoses": [{"condition": string, "details": string, "recommendations": string}],
@@ -63,6 +69,12 @@ Format the response as a JSON object with the following structure:
   }
 }
 
+For the title field, create a concise title that summarizes what the document is about (e.g., "Complete Blood Count Results", "Liver Function Test", "MRI Report").
+For the category field, you MUST choose exactly one of these three values:
+- "heart" - if the document focuses primarily on cardiac/cardiovascular issues or tests
+- "brain" - if the document focuses primarily on neurological issues or brain-related tests
+- "general" - for all other medical documents, or when the focus spans multiple systems
+
 Set isMedicalReport to true if you see ANY medical content such as lab values, medical terminology, doctor's notes, or prescription information.
 Set confidence between 0 and 1 based on document clarity and how confident you are about the medical nature of the document.
 
@@ -115,6 +127,8 @@ INCORRECT RESPONSE FORMATS (DO NOT DO THESE):
 
 CORRECT FORMAT (DO THIS):
 {
+  "title": "Complete Blood Count Results",
+  "category": "heart",
   "keyMedicalTerms": [
     {"term": "RBC", "definition": "Red blood cells"},
     {"term": "WBC", "definition": "White blood cells"}
@@ -401,6 +415,8 @@ Document text:
     // Check if response has all required properties
     if (
       !response ||
+      typeof response.title !== 'string' ||
+      typeof response.category !== 'string' ||
       !Array.isArray(response.keyMedicalTerms) ||
       !Array.isArray(response.labValues) ||
       !Array.isArray(response.diagnoses) ||

backend/src/document-processor/services/document-processor.service.spec.ts

@@ -35,6 +35,8 @@ describe('DocumentProcessorService', () => {
       };
 
       const medicalAnalysis = {
+        title: 'Test Report',
+        category: 'general',
         keyMedicalTerms: [],
         labValues: [],
         diagnoses: [],
@@ -147,6 +149,8 @@ describe('DocumentProcessorService', () => {
           keyValuePairs: [],
         },
         analysis: {
+          title: 'Document 1 Report',
+          category: 'general',
           keyMedicalTerms: [],
           labValues: [],
           diagnoses: [],
@@ -172,6 +176,8 @@ describe('DocumentProcessorService', () => {
           keyValuePairs: [],
         },
         analysis: {
+          title: 'Document 2 Report',
+          category: 'general',
           keyMedicalTerms: [],
           labValues: [],
           diagnoses: [],

backend/src/document-processor/services/document-processor.service.ts

@@ -164,6 +164,8 @@ export class DocumentProcessorService {
             keyValuePairs: [],
           },
           analysis: {
+            title: 'Failed Document',
+            category: 'general',
             keyMedicalTerms: [],
             labValues: [],
             diagnoses: [],