Update app.py

app.py

@@ -3,10 +3,12 @@ import os
 import json
 from utils import (
     get_salesforce_client, get_salesforce_objects, get_object_fields,
-    extract_text_from_pdf, extract_key_value_pairs, map_fields,
+    extract_text_from_pdf, extract_key_value_pairs,
     create_record, attach_pdf, log_failure
 )
 
+from ai_mapper import ai_map_fields  # ✅ NEW: AI-based mapper
+
 # Ensure uploads directory exists
 os.makedirs("uploads", exist_ok=True)
 
@@ -88,7 +90,7 @@ def display_mappings(pdf_paths, object_name):
     extracted_data, error = extract_key_value_pairs(pdf_path)
     if error:
         return f"Error: {error}"
-    mappings, confidence_scores, error = map_fields(extracted_data, fields)
+    mappings, confidence_scores, error = ai_map_fields(extracted_data[0]["keys"], fields)  # ✅ updated
     if error:
         return f"Error: {error}"
     output = ""
@@ -112,7 +114,7 @@ def migrate_to_salesforce(pdf_paths, object_name):
     if error:
         log_failure(pdf_path, object_name, error)
         return f"Error: {error}"
-    mappings, _, error = map_fields(extracted_data, fields)
+    mappings, _, error = ai_map_fields(extracted_data[0]["keys"], fields)  # ✅ updated
     if error:
         log_failure(pdf_path, object_name, error)
         return f"Error: {error}"