update

app.py

@@ -456,6 +456,9 @@ def _run_template_pipeline(file_path, form_hint, file2_path=None):
 
     raw = extract_fields(img_path, source_type)
 
+    confidence = raw.get("_crnn_confidence", {})
+    fields = {k: v for k, v in raw.items() if not k.startswith("_")}
+
     if isinstance(raw, dict) and raw.get('status') == 'error':
         print('[app.py] Blank page detected — skipping extraction')
         raise ValueError('Blank page detected. Please try another file.')
@@ -471,6 +474,26 @@ def _run_template_pipeline(file_path, form_hint, file2_path=None):
             raw = {**raw, **raw2}
 
     fields = _map_template_output(raw, form_hint)
+    confidence = {
+    "registry_no": raw_confidence.get("registry_no", 0.60),
+    "city_municipality": raw_confidence.get("city_municipality", 0.60),
+    "province": raw_confidence.get("province", 0.60),
+    "date_submitted": raw_confidence.get("registration_date", 0.60),
+    "child_first": raw_confidence.get("name_first", 0.60),
+    "child_middle": raw_confidence.get("name_middle", 0.60),
+    "child_last": raw_confidence.get("name_last", 0.60),
+    "sex": raw_confidence.get("sex", 0.60),
+    "dob_day": raw_confidence.get("dob_day", 0.60),
+    "dob_month": raw_confidence.get("dob_month", 0.60),
+    "dob_year": raw_confidence.get("dob_year", 0.60),
+    "pob_city": raw_confidence.get("place_of_birth", 0.60),
+    "mother_first": raw_confidence.get("mother_name", 0.60),
+    "mother_citizenship": raw_confidence.get("mother_citizenship", 0.60),
+    "father_first": raw_confidence.get("father_name", 0.60),
+    "father_citizenship": raw_confidence.get("father_citizenship", 0.60),
+    "parents_marriage_month": raw_confidence.get("marriage_date", 0.60),
+    "parents_marriage_city": raw_confidence.get("marriage_place", 0.60),
+}
     form_class = form_hint if form_hint in ('1A', '2A', '3A', '90') else '1A'
 
     if _bridge is not None:
@@ -522,13 +545,17 @@ def _run_template_pipeline(file_path, form_hint, file2_path=None):
 
             ner_count = sum(1 for v in ner_fields.values() if v)
             print(f'[app.py] NER enriched {ner_count} fields')
-            confidence = {k: mnb_result['confidence'] for k in fields}
+            confidence = {k: confidence.get(k, 0.60) for k in fields}
 
         except Exception as _ner_err:
             print(f'[app.py] NER error (using template only): {_ner_err}')
-            confidence = {k: 0.85 for k in fields}
+            confidence = {}
+
+            for key, value in fields.items():
+                # fallback if no confidence yet
+                confidence[key] = 0.60
     else:
-        confidence = {k: 0.85 for k in fields}
+        confidence = {k: confidence.get(k, 0.60) for k in fields}
 
     non_empty = {k: v for k, v in fields.items() if v}
     print(f'[app.py] form_class={form_class}, {len(non_empty)}/{len(fields)} non-empty fields')

template_matcher.py

@@ -1436,7 +1436,7 @@ def extract_fields(image_path: str, form_type: str = None):
             if final_text:
                 fields[field_name] = final_text
                 crnn_confidences[field_name] = crnn_conf
-
+                
         print(f'[template_matcher] Extracted: {len(fields)}/{len(template)} fields')
         paddle_count = sum(1 for m in debug_methods.values() if m == 'paddle-detect')
         abs_count = sum(1 for m in debug_methods.values() if m == 'absolute')
@@ -1456,7 +1456,10 @@ def extract_fields(image_path: str, form_type: str = None):
         fields['_quality'] = quality
         fields['_corrections'] = corrections
         fields['_crnn_confidence'] = crnn_confidences
-        return fields
+        return {
+            **fields,
+            "_crnn_confidence": crnn_confidences
+        }
 
     except Exception as e:
         print(f'[template_matcher] extract_fields error: {e}')