Update app.py

app.py

@@ -5,15 +5,22 @@ from utils import extract_text_from_pdf, parse_pdf_to_dict
 
 from docx.shared import RGBColor
 
-def replace_red_text_with_data(doc_path, data_dict):
-    doc = Document(doc_path)
+from docx import Document
+from docx.shared import RGBColor
+
+def replace_red_text_with_data(word_path, data_dict):
+    doc = Document(word_path)
 
     for para in doc.paragraphs:
-        for run in para.runs:
-            if run.font.color and run.font.color.rgb and run.font.color.rgb == RGBColor(255, 0, 0):
-                text = run.text.strip()
-                if text in data_dict:
-                    run.text = data_dict[text]
+        full_text = para.text
+        for i, run in enumerate(para.runs):
+            if run.font.color and run.font.color.rgb == RGBColor(255, 0, 0):
+                # Search for the key (label) before this red text
+                preceding_text = ''.join(r.text for r in para.runs[:i]).lower()
+                for key in data_dict:
+                    if key.lower() in preceding_text:
+                        run.text = data_dict[key]
+                        break
 
     return doc