Upload train.py with huggingface_hub

train.py

@@ -219,20 +219,21 @@ VARIANT_TEMPLATES = [
 ]
 
 SYMPTOM_VARIANTS = [
-    # Brakes
+    # Brakes — pads/shoes
     {"symptoms": [
         "grinding noise when braking", "squealing when stopping", "brake pedal goes to floor",
         "pulsating brake pedal", "car pulls when braking", "scraping sound when stopping",
         "metal on metal when braking", "grinding from front when stopping",
         "high pitched squeal stopping", "brakes feel spongy",
     ], "subcategories": ["Brake > Disc Pads and Brake Shoes"]},
+    # Brakes — rotors/drums
     {"symptoms": [
         "steering wheel shakes when braking", "vibration when stopping",
         "pulsation in pedal at highway speed", "shimmy when braking",
         "warped rotors vibration", "shudder when applying brakes",
     ], "subcategories": ["Brake > Drums and Rotors"]},
     {"symptoms": ["abs light on", "anti-lock brake warning", "abs fault code", "traction control light and abs light"], "subcategories": ["Brake > ABS Components"]},
-    # Suspension — shocks/struts
+    # Suspension — shocks/struts (expert-expanded)
     {"symptoms": [
         "clunking over bumps", "vehicle bounces excessively", "nose dives when stopping",
         "rough ride", "bouncy ride", "car keeps bouncing after bumps",
@@ -240,33 +241,68 @@ SYMPTOM_VARIANTS = [
         "rear sags under load", "knocking over potholes", "harsh ride quality",
         "suspension feels loose", "banging noise hitting bumps",
         "front end clunk", "noise over railroad tracks",
+        "bang over bumps", "thud from front end", "knock over rough road",
+        "chassis slap over bumps", "double thump speed bumps",
+        "front end bangs dips", "rear clunk over bumps", "trunk area thump bumps",
+        "rattle over small bumps", "suspension noise over potholes",
+        "clunk from front suspension", "rear suspension thump",
+        "noise over expansion joints", "knocks on washboard road",
+        "clunk over driveway apron", "noise from strut area over bumps",
     ], "subcategories": ["Suspension > Shocks and Struts"]},
-    # Suspension — strut mounts
+    # Suspension — strut mounts (expert-expanded)
     {"symptoms": [
         "upper mount plate", "strut mount noise", "top mount bearing worn",
         "upper strut mount", "mount plate clunking", "popping when turning steering wheel stationary",
         "creaking from top of strut tower", "strut plate bearing",
+        "upper strut plate", "mount plate bearing", "strut tower mount",
+        "top mount worn", "upper spring seat", "strut top hat",
+        "mount plate popping", "bearing plate noise turning",
+        "strut mount bearing noisy", "upper mount clunk", "top hat binding",
+        "spring seat cracked", "mount bearing seized",
+        "pop while steering from strut mount", "front mount plate noise",
+        "upper mount separated",
     ], "subcategories": ["Suspension > Shocks and Struts"]},
-    # Suspension — alignment
+    # Suspension — alignment (expert-expanded)
     {"symptoms": [
         "vehicle pulls to one side", "uneven tire wear", "steering wheel off center",
         "vehicle pulls right", "vehicle pulls left", "car drifts right",
         "car drifts left", "steering wheel crooked", "feathering on tires",
         "cupping on tires", "car doesn't track straight", "pulls to the right on highway",
+        "pulls to right", "drifts right on flat road", "steering pulls",
+        "needs alignment pulls", "alignment off pulls right", "tracks right",
+        "wanders on highway", "won't hold straight line",
+        "off center steering wheel", "lane drift", "darting on highway",
     ], "subcategories": ["Suspension > Alignment"]},
+    # Suspension — control arms/bushings (expert-expanded)
     {"symptoms": [
         "clunk when turning", "loose steering feel",
         "control arm bushing worn", "worn control arm bushings",
         "sway bar link rattle", "stabilizer bar noise",
         "ball joint play", "loose front end", "front end clunk turning",
+        "bushing cracked", "bushing deteriorated", "lower control arm play",
+        "LCA bushing torn", "compliance bushing shot",
+        "control arm rear bushing split", "bushing separated",
+        "arm bushing excessive movement", "thrust arm bushing cracked",
+        "front end looseness low speed", "clunk turning and braking",
+        "hydro bushing leaking", "rear compliance bushing torn",
+        "FCA bushing worn", "RCA bushing worn",
     ], "subcategories": ["Suspension > Suspension, Springs and Related Components"]},
-    # Engine
+    # Engine — gaskets
     {"symptoms": ["oil leak from top of engine", "burning oil smell", "oil seeping from valve cover", "wet around intake"], "subcategories": ["Engine > Gaskets and Sealing Systems"]},
+    # Engine — timing/valve train (expert-expanded)
     {"symptoms": [
         "rattling on cold start", "timing noise", "engine ticking",
         "timing chain replacement needed", "timing chain stretched",
         "timing chain rattle", "chain slap on startup", "variable valve timing code",
         "cam phaser noise", "vvt solenoid", "engine timing off",
+        "chain stretched", "timing chain code", "P0016 cam crank correlation",
+        "P0017 cam position", "timing set replacement", "chain guides worn",
+        "cold start chain rattle", "chain noise on startup", "chain slack",
+        "cam crank out of sync", "correlation fault", "chain tensioner weak",
+        "guides broken", "phaser correlation fault",
+        "cam timing over retarded", "cam timing over advanced",
+        "engine rattles first start", "chain kit recommended",
+        "front cover timing repair", "timing chain needed",
     ], "subcategories": ["Engine > Valve Train Components"]},
     # Cooling
     {"symptoms": ["overheating", "temperature gauge high", "coolant on ground", "coolant leak", "steam from hood"], "subcategories": ["Belts and Cooling > Radiators, Coolers and Related Components"]},
@@ -279,71 +315,138 @@ SYMPTOM_VARIANTS = [
     {"symptoms": ["car won't start", "clicking when turning key", "no crank", "slow crank", "starter grinds"], "subcategories": ["Electrical, Charging and Starting > Starter and Related Components"]},
     {"symptoms": ["battery light on", "dim headlights", "electrical issues", "charging system warning"], "subcategories": ["Electrical, Charging and Starting > Alternator / Generator and Related Components"]},
     {"symptoms": ["dead battery", "battery keeps dying", "battery won't hold charge", "needs jump every morning"], "subcategories": ["Electrical, Charging and Starting > Battery and Related Components"]},
-    # Ignition
+    # Ignition — CEL/misfire (expert-expanded)
     {"symptoms": [
         "CEL on", "check engine light", "misfire", "rough idle", "rough running",
         "CEL on rough running", "engine misfiring", "stumbles at idle",
         "rough running check engine light", "cylinder misfire code",
         "P0300 random misfire", "P0301 misfire cylinder 1", "hesitation under acceleration",
-        "spark plugs replacement due", "plugs due for service", "100k spark plug service",
         "tune up needed", "tune-up", "engine tune up",
+        "CEL rough running", "CEL with misfire", "check engine rough idle",
+        "MIL on running rough", "SES light rough idle", "CEL P0300",
+        "CEL multiple misfire", "engine light on stumbling",
+        "CEL on with shaking", "service engine soon rough",
+        "misfiring under load", "flashing CEL misfire",
+        "sputtering with check engine light", "shaking at idle CEL",
+        "rough idle with codes", "stumble and check engine light",
+        "P0301", "P0302", "P0303", "P0304",
+        "random misfire", "multiple cylinder misfire",
+        "bucks and jerks with CEL",
     ], "subcategories": ["Ignition > Secondary Ignition"]},
-    # HVAC — A/C
+    # Ignition — spark plugs maintenance (expert-expanded)
+    {"symptoms": [
+        "spark plugs replacement due", "plugs due for service", "100k spark plug service",
+        "plugs due", "spark plug service", "60k plug change",
+        "100k plug service", "iridium plugs due", "platinum plugs worn",
+        "plug gap too wide", "mileage-based plug replacement", "maintenance plugs",
+        "tune up due", "due for plugs per mileage", "recommend spark plugs",
+        "plugs worn", "electrodes worn", "plugs original",
+        "overdue for spark plugs", "plug service due by maintenance schedule",
+        "replace plugs as maintenance",
+    ], "subcategories": ["Ignition > Secondary Ignition"]},
+    # HVAC — A/C (expert-expanded)
     {"symptoms": [
         "a/c blowing warm", "no cold air", "a/c not working",
         "AC recharge needed blowing warm", "ac not cold enough",
         "ac recharge needed", "a/c weak", "a/c warm on one side",
         "refrigerant low", "ac blows hot", "air conditioning not cooling",
+        "AC recharge", "A/C recharge needed", "freon low",
+        "134a recharge", "1234yf recharge", "ac charge and check",
+        "air not blowing cold", "AC performance check",
+        "AC system evacuation and recharge", "AC evac and recharge",
+        "low refrigerant", "static pressure low", "recharge and leak test",
+        "recover evacuate recharge", "recharge with dye",
+        "not cooling at idle", "vent temp high", "AC weak",
+        "warm air from vents",
     ], "subcategories": ["HVAC > A/C Clutch and Compressor"]},
     {"symptoms": ["no heat", "heater not working", "heat blows cold", "lukewarm heat", "heater weak"], "subcategories": ["HVAC > Air Heater Components"]},
-    # HVAC — blower motor
+    # HVAC — blower motor (expert-expanded)
     {"symptoms": [
         "blower motor not working", "no air from vents", "fan not blowing",
         "blower only works on high", "hvac fan stopped", "no air flow from dash",
         "blower resistor", "fan speed not changing", "only high speed works on fan",
+        "HVAC fan dead", "cabin fan stopped", "no air movement dash",
+        "blower quit working", "heat fan not blowing", "fan motor seized",
+        "blower motor intermittent", "no fan from vents", "blower inop",
+        "interior fan not working", "no blower on any speed",
+        "blower resistor concern", "fan cuts in and out",
+        "no airflow from vents", "HVAC blower intermittent",
     ], "subcategories": ["HVAC > Motors, Core, Case and Related Components"]},
-    # Exhaust
+    # Exhaust — muffler
     {"symptoms": ["loud exhaust", "exhaust leak", "rumbling underneath", "exhaust drone", "holes in muffler"], "subcategories": ["Exhaust > Mufflers"]},
+    # Exhaust — catalytic converter (expert-expanded)
     {"symptoms": [
         "rotten egg smell", "catalytic converter code", "sulfur smell from exhaust",
         "P0420 catalyst efficiency", "P0430 catalyst", "cat efficiency below threshold",
         "rotten egg odor", "exhaust smells like eggs", "failing catalytic converter",
+        "egg smell from exhaust", "sulfur odor driving", "rotten smell tailpipe",
+        "hydrogen sulfide smell", "cat going bad smell", "stinky exhaust",
+        "exhaust smells terrible", "rotten smell after hard acceleration",
+        "bad catalyst smell", "converter smell", "strong sulfur exhaust odor",
+        "rotten egg smell under load", "sulfur smell after highway drive",
     ], "subcategories": ["Exhaust > Catalytic Converter"]},
-    # Steering
+    # Steering — power steering (expert-expanded)
     {"symptoms": [
         "whining when turning", "hard to steer", "power steering whine",
         "whining noise turning wheel", "groan when turning steering wheel",
         "steering stiff", "fluid leaking from steering rack",
         "whine at low speed turns", "moaning sound turning",
         "power steering fluid low", "ps pump noise",
+        "whine at full lock", "pump noise turning", "moan turning wheel",
+        "steering whine low speed", "PS pump whine", "groan turning at idle",
+        "howling when steering", "fluid cavitation noise turning",
+        "power steering noisy", "steering pump moans",
+        "whine when parking", "groan backing out while turning",
+        "steering noise cold", "aerated PS fluid", "low PS fluid noise",
     ], "subcategories": ["Steering > Power Steering Hoses, Pumps, and Related Components"]},
-    # Driveline
+    # Driveline — CV
     {"symptoms": ["clicking when turning", "cv clicking", "clicking noise in turns", "cv axle popping"], "subcategories": ["Driveline and Axles > CV Components"]},
+    # Driveline — wheel bearings (expert-expanded)
     {"symptoms": [
         "humming from wheel area", "wheel noise at speed", "growling noise from front",
         "humming gets louder with speed", "bearing noise",
         "roaring sound from wheel", "drone changes with steering input",
         "wheel hum goes away turning one direction", "rumble from rear axle area",
         "hub bearing noise", "grinding from front wheel",
+        "hum from front wheel", "droning noise road speed", "wheel rumble highway",
+        "noise changes turning left right", "growl from hub area",
+        "bearing drone 40mph plus", "hum louder faster you go",
+        "helicopter noise road speed", "wheel bearing growl",
+        "drone from left front", "drone from right front",
+        "noise unloads on turns", "hub noise",
+        "rumble from corner of vehicle",
     ], "subcategories": ["Driveline and Axles > Wheel Bearings, Seals, and Related Components"]},
-    # Body
+    # Body — mirrors
     {"symptoms": [
         "mirror assembly", "mirror replacement", "side mirror broken",
         "mirror glass cracked", "power mirror not working", "mirror folding mechanism",
         "driver side mirror", "passenger mirror", "rear view mirror fell off",
         "mirror turn signal", "blind spot mirror",
     ], "subcategories": ["Body > Mirrors"]},
-    # Tire — TPMS
+    # Tire — TPMS (expert-expanded)
     {"symptoms": [
         "TPMS light on", "tire pressure light", "tire pressure warning",
         "tpms sensor", "tire monitor light", "low tire pressure warning",
         "tpms fault", "tire pressure monitoring light",
+        "tire light on dash", "TPMS warning", "tire monitor fault",
+        "low tire light stays on after fill", "TPMS sensor battery dead",
+        "tire pressure indicator", "TPMS relearn needed",
+        "tire pressure light flashing", "sensor not detected",
+        "tire pressure monitor inop", "tpms service",
+        "no sensor signal", "relearn tire sensors", "sensor battery low",
     ], "subcategories": ["Tire and Wheel > Sensors"]},
-    # Fluids — ATF
+    # Fluids — ATF (expert-expanded)
     {"symptoms": [
         "ATF flush", "ATF change", "transmission fluid service",
         "trans fluid flush", "automatic transmission fluid",
         "transmission fluid dark", "burnt trans fluid",
+        "ATF exchange", "ATF drain and fill", "automatic transmission fluid service",
+        "transmission service ATF", "ATF dark needs flush",
+        "ATF maintenance", "trans fluid exchange",
+        "transmission fluid service due", "CVT fluid service",
+        "DSG service", "pan drop and filter", "trans service by mileage",
+        "burnt ATF", "fluid discolored", "fluid exchange recommended",
+        "service transmission fluid", "trans fluid change",
     ], "subcategories": ["Oil, Fluids and Chemicals > Functional Fluid, Lubricant, Grease (including Additives)"]},
     # Fluids — oil change with marketing prefix
     {"symptoms": [
@@ -365,6 +468,24 @@ SYMPTOM_TEMPLATES = [
     "Inspect for {symptom}",
     "Diag for {symptom}",
     "Check {symptom}",
+    # Expert-added: DMS/shop shorthand prefixes
+    "Cust states {symptom}",
+    "States {symptom}",
+    "Complains of {symptom}",
+    "Chk and advise {symptom}",
+    "Check and advise {symptom}",
+    "Inspect and advise {symptom}",
+    "Found {symptom}",
+    "Verified {symptom}",
+    "Confirmed {symptom}",
+    "Recommend {symptom}",
+    "Rec {symptom}",
+    "LF {symptom}",
+    "RF {symptom}",
+    "LR {symptom}",
+    "RR {symptom}",
+    "Front {symptom}",
+    "Rear {symptom}",
 ]
 
 COMMON_TYPOS = {
@@ -388,6 +509,7 @@ COMMON_TYPOS = {
 }
 
 HARD_NEGATIVES = [
+    # --- Original hard negatives ---
     {"query": "replace radiator", "positive": "Belts and Cooling > Radiators, Coolers and Related Components", "negative": "Transmission > Radiators, Coolers and Related Components"},
     {"query": "r&r radiator", "positive": "Belts and Cooling > Radiators, Coolers and Related Components", "negative": "Transmission > Radiators, Coolers and Related Components"},
     {"query": "radiator leaking", "positive": "Belts and Cooling > Radiators, Coolers and Related Components", "negative": "Transmission > Radiators, Coolers and Related Components"},
@@ -435,6 +557,33 @@ HARD_NEGATIVES = [
     {"query": "TPMS light on", "positive": "Tire and Wheel > Sensors", "negative": "Ignition > Distributor and Magneto"},
     {"query": "TPMS light on", "positive": "Tire and Wheel > Sensors", "negative": "Driveline and Axles > CV Boots and Related Components"},
     {"query": "tire pressure light on", "positive": "Tire and Wheel > Sensors", "negative": "Driveline and Axles > Wheel Bearings, Seals, and Related Components"},
+    # --- Expert-added hard negatives (with corrected PCAdb names) ---
+    {"query": "CEL rough running", "positive": "Ignition > Secondary Ignition", "negative": "Emission Control > Sensors"},
+    {"query": "CEL rough running", "positive": "Ignition > Secondary Ignition", "negative": "Engine > Valve Train Components"},
+    {"query": "CEL rough running", "positive": "Ignition > Secondary Ignition", "negative": "Air and Fuel Delivery > Fuel Pumps and Related Components"},
+    {"query": "CEL rough running", "positive": "Ignition > Secondary Ignition", "negative": "Air and Fuel Delivery > Filters"},
+    {"query": "Spark plugs replacement due", "positive": "Ignition > Secondary Ignition", "negative": "Ignition > Ignition Coil"},
+    {"query": "Spark plugs replacement due", "positive": "Ignition > Secondary Ignition", "negative": "Engine > Valve Train Components"},
+    {"query": "Spark plugs replacement due", "positive": "Ignition > Secondary Ignition", "negative": "Air and Fuel Delivery > Fuel Injection System and Related Components"},
+    {"query": "AC recharge needed", "positive": "HVAC > A/C Clutch and Compressor", "negative": "HVAC > A/C Condenser and Evaporator"},
+    {"query": "AC recharge needed", "positive": "HVAC > A/C Clutch and Compressor", "negative": "HVAC > Motors, Core, Case and Related Components"},
+    {"query": "blower not working", "positive": "HVAC > Motors, Core, Case and Related Components", "negative": "HVAC > A/C Clutch and Compressor"},
+    {"query": "blower not working", "positive": "HVAC > Motors, Core, Case and Related Components", "negative": "Belts and Cooling > Cooling Fan, Clutch and Motor"},
+    {"query": "TPMS light", "positive": "Tire and Wheel > Sensors", "negative": "Tire and Wheel > Tire"},
+    {"query": "timing chain needed", "positive": "Engine > Valve Train Components", "negative": "Belts and Cooling > Accessory Drive Belt System Components"},
+    {"query": "control arm bushing", "positive": "Suspension > Suspension, Springs and Related Components", "negative": "Suspension > Shocks and Struts"},
+    {"query": "control arm bushing", "positive": "Suspension > Suspension, Springs and Related Components", "negative": "Steering > Steering, Gear and Related Components"},
+    {"query": "clunking over bumps", "positive": "Suspension > Shocks and Struts", "negative": "Suspension > Suspension, Springs and Related Components"},
+    {"query": "clunking over bumps", "positive": "Suspension > Shocks and Struts", "negative": "Steering > Steering, Gear and Related Components"},
+    {"query": "clunking over bumps", "positive": "Suspension > Shocks and Struts", "negative": "Brake > Brake Hydraulics"},
+    {"query": "vehicle pulls right", "positive": "Suspension > Alignment", "negative": "Tire and Wheel > Tire"},
+    {"query": "vehicle pulls right", "positive": "Suspension > Alignment", "negative": "Brake > Brake Hydraulics"},
+    {"query": "vehicle pulls right", "positive": "Suspension > Alignment", "negative": "Suspension > Suspension, Springs and Related Components"},
+    {"query": "whining when turning", "positive": "Steering > Power Steering Hoses, Pumps, and Related Components", "negative": "Belts and Cooling > Accessory Drive Belt System Components"},
+    {"query": "whining when turning", "positive": "Steering > Power Steering Hoses, Pumps, and Related Components", "negative": "Driveline and Axles > CV Components"},
+    {"query": "humming from wheel area", "positive": "Driveline and Axles > Wheel Bearings, Seals, and Related Components", "negative": "Tire and Wheel > Tire"},
+    {"query": "humming from wheel area", "positive": "Driveline and Axles > Wheel Bearings, Seals, and Related Components", "negative": "Brake > Brake Hydraulics"},
+    {"query": "humming from wheel area", "positive": "Driveline and Axles > Wheel Bearings, Seals, and Related Components", "negative": "Driveline and Axles > Differential"},
 ]