Update app.py

app.py

@@ -121,8 +121,15 @@ def migrate_to_salesforce(pdf_paths, object_name):
     raw_data = {mappings[key]: value for key, value in zip(extracted_data[0]["keys"], extracted_data[0]["values"])}
     data = {}
 
+    id_fields = {"OwnerId", "CreatedById", "LastModifiedById", "AccountId"}
     for field, value in raw_data.items():
-        if isinstance(value, str) and any(sep in value for sep in ["/", "-"]):
+        if field in id_fields:
+            # Only include if it's a valid Salesforce ID
+            if isinstance(value, str) and len(value.strip()) in (15, 18) and value.strip().isalnum():
+                data[field] = value.strip()
+            else:
+                continue
+        elif isinstance(value, str) and any(sep in value for sep in ["/", "-"]):
             try:
                 parsed_date = datetime.strptime(value.strip(), "%m/%d/%y")
                 data[field] = parsed_date.strftime("%Y-%m-%d")