Upload hts_validator.py

hts_validator.py

@@ -88,14 +88,14 @@ SCENARIO_SUMMARIES = {
     "L7": "Aluminum HTS + alum+copper keywords - Alum 232 + 99030133",
     "L8": "Aluminum HTS + all keywords - Alum 232 + 99030133",
     # Level 3: Single HTS - Copper
-    "U1": "Copper HTS, no keyword - Copper + 99030133",
+    "U1": "Copper HTS, no keyword - 99037802 (no copper content)",
     "U2": "Copper HTS + metal keyword - 99030125 (mismatch)",
-    "U3": "Copper HTS + aluminum keyword - 99030125 (mismatch)",
-    "U4": "Copper HTS + copper keyword - Copper tariff",
-    "U5": "Copper HTS + metal+alum keywords - 99030125 (mismatch)",
-    "U6": "Copper HTS + metal+copper keywords - Copper tariff",
-    "U7": "Copper HTS + alum+copper keywords - Copper tariff",
-    "U8": "Copper HTS + all keywords - Copper tariff",
+    "U3": "Copper HTS + aluminum keyword - 99037802 + Alum 232",
+    "U4": "Copper HTS + copper keyword - 99037801 (copper content)",
+    "U5": "Copper HTS + metal+alum keywords - 99037802 + Alum 232",
+    "U6": "Copper HTS + metal+copper keywords - 99037801 (copper content)",
+    "U7": "Copper HTS + alum+copper keywords - FLAG ambiguous",
+    "U8": "Copper HTS + all keywords - FLAG ambiguous",
     # Level 4: No HTS Match
     "N1": "No metal HTS, no keyword - no action",
     "N2": "No metal HTS + metal keyword - 99030125",
@@ -949,17 +949,18 @@ class HTSValidator:
                 )
 
         # H3: Copper Only
+        # New logic: Copper keyword -> 99037801, No copper keyword -> 99037802
         if in_copper and not in_steel and not in_aluminum:
-            if keyword_cat == "K0":  # U1: No keyword
+            if keyword_cat == "K0":  # U1: No keyword -> 99037802 (no copper content)
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U1",
-                    expected_codes=list(COPPER_CODES) + [GENERAL_301_CODE],
-                    forbidden_codes={MISMATCH_CODE},
+                    expected_codes=["99037802"],
+                    forbidden_codes={MISMATCH_CODE, "99037801"},
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K1":  # U2: Metal only
+            elif keyword_cat == "K1":  # U2: Metal only -> mismatch
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U2",
@@ -968,59 +969,61 @@ class HTSValidator:
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K2":  # U3: Aluminum only
+            elif keyword_cat == "K2":  # U3: Aluminum only -> 99037802 + Alum 232
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U3",
-                    expected_codes=[MISMATCH_CODE],
-                    forbidden_codes=ALUMINUM_232_CODES | COPPER_CODES | {GENERAL_301_CODE},
+                    expected_codes=["99037802"] + list(ALUMINUM_232_CODES) + [GENERAL_301_CODE],
+                    forbidden_codes={MISMATCH_CODE, "99037801"},
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K3":  # U4: Copper only
+            elif keyword_cat == "K3":  # U4: Copper only -> 99037801 (copper content)
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U4",
-                    expected_codes=list(COPPER_CODES),
-                    forbidden_codes={MISMATCH_CODE},
+                    expected_codes=["99037801"],
+                    forbidden_codes={MISMATCH_CODE, "99037802"},
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K4":  # U5: Metal + Aluminum
+            elif keyword_cat == "K4":  # U5: Metal + Aluminum -> 99037802 + Alum 232
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U5",
-                    expected_codes=[MISMATCH_CODE],
-                    forbidden_codes=ALL_232_CODES | {GENERAL_301_CODE},
+                    expected_codes=["99037802"] + list(ALUMINUM_232_CODES) + [GENERAL_301_CODE],
+                    forbidden_codes={MISMATCH_CODE, "99037801"},
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K5":  # U6: Metal + Copper
+            elif keyword_cat == "K5":  # U6: Metal + Copper -> 99037801 (copper keyword present)
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U6",
-                    expected_codes=list(COPPER_CODES),
-                    forbidden_codes={MISMATCH_CODE},
+                    expected_codes=["99037801"],
+                    forbidden_codes={MISMATCH_CODE, "99037802"},
                     additional_set=additional_set,
                 **indicators
                 )
-            elif keyword_cat == "K6":  # U7: Aluminum + Copper
+            elif keyword_cat == "K6":  # U7: Aluminum + Copper -> FLAG ambiguous
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U7",
-                    expected_codes=list(COPPER_CODES),
-                    forbidden_codes={MISMATCH_CODE},
+                    expected_codes=[], forbidden_codes=set(),
                     additional_set=additional_set,
-                **indicators
+                    **indicators,
+                    always_flag=True,
+                    flag_reason="Copper HTS with both aluminum and copper keywords - ambiguous"
                 )
-            elif keyword_cat == "K7":  # U8: All three
+            elif keyword_cat == "K7":  # U8: All three -> FLAG ambiguous
                 return self._create_result(
                     entry_number, description, primary_hts, additional_hts,
                     scenario_id="U8",
-                    expected_codes=list(COPPER_CODES),
-                    forbidden_codes={MISMATCH_CODE},
+                    expected_codes=[], forbidden_codes=set(),
                     additional_set=additional_set,
-                **indicators
+                    **indicators,
+                    always_flag=True,
+                    flag_reason="Copper HTS with all keywords - ambiguous"
                 )
 
         # =====================================================================