Update backend/scripts/generate_template_schema_skeletons.py

backend/scripts/generate_template_schema_skeletons.py

@@ -23,12 +23,15 @@ TABLE_ANCHORS = [
     {"key": "qty", "expected_text": "Qty"},
 ]
 
+# Canonical line-item column set. Keep in sync with the UI (src/App.jsx) and prompts.py.
 TABLE_COLUMNS = [
     {"key": "item_number", "label": "Item Number"},
     {"key": "lot_number", "label": "Lot Number"},
     {"key": "description", "label": "Description"},
     {"key": "qty", "label": "Qty"},
     {"key": "price", "label": "Price"},
+    {"key": "total", "label": "Total"},
+    {"key": "bundle", "label": "Bundle"},
 ]
 
 
@@ -94,6 +97,8 @@ def schema_skeleton(form_id: str) -> Dict[str, Any]:
                     {"key": "description", "label": "Description", "bbox_rel_norm": None},
                     {"key": "qty", "label": "Qty", "bbox_rel_norm": None},
                     {"key": "price", "label": "Price", "bbox_rel_norm": None},
+                    {"key": "total", "label": "Total", "bbox_rel_norm": None},
+                    {"key": "bundle", "label": "Bundle", "bbox_rel_norm": None},
                 ],
                 "table_anchors": [
                     {"key": "item_number", "expected_text": "Item Number", "bbox_norm": None},
@@ -120,9 +125,6 @@ def main() -> None:
         data = json.loads(fp.read_text(encoding="utf-8"))
         template_id = (data.get("template_id") or fp.stem).strip()
 
-        # Only touch your known template IDs if you want:
-        # if not template_id.startswith("T"): continue
-
         # Overwrite or create schema skeleton
         data["schema"] = schema_skeleton(form_id=f"template_{template_id}")
 
@@ -134,4 +136,4 @@ def main() -> None:
 
 
 if __name__ == "__main__":
-    main()
+    main()