fix: logic hardening, data label consistency, and CLI shim creation

Author: Zhe Wang

inspector.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python3
+"""
+FairProp Inspector CLI - Unified Entry Point
+Matches documentation: python inspector.py --check "text"
+"""
+import sys
+import argparse
+import os
+
+# Ensure src is in python path
+sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
+
+from inference.predict import predict
+
+def main():
+    parser = argparse.ArgumentParser(description="FairProp Inspector CLI")
+    parser.add_argument("--check", type=str, required=True, help="Property description to inspect")
+    parser.add_argument("--model", type=str, default="artifacts/model", help="Path to trained model")
+    args = parser.parse_args()
+
+    label, confidence = predict(args.check, args.model)
+    
+    print("-" * 40)
+    print(f"FairProp Inspector Analysis")
+    print("-" * 40)
+    print(f"Input:      {args.check}")
+    print(f"Prediction: {label}")
+    print(f"Confidence: {confidence:.2%}")
+    print("-" * 40)
+
+if __name__ == "__main__":
+    main()

src/generator/generate_data.py

@@ -22,10 +22,10 @@ def generate_synthetic_data(rules: Dict, num_samples: int) -> List[Dict]:
 
     # Mock generation logic for skeleton
     templates = [
-        ("No kids allowed.", "NON_COMPLIANT", "Quiet community available for all residents."),
-        ("Great for families.", "COMPLIANT", "Great for families."),
-        ("Christian only.", "NON_COMPLIANT", "Open to people of all faiths."),
-        ("Walking distance to shops.", "COMPLIANT", "Walking distance to shops.")
+        ("No kids allowed.", 1, "Quiet community available for all residents."),
+        ("Great for families.", 0, "Great for families."),
+        ("Christian only.", 1, "Open to people of all faiths."),
+        ("Walking distance to shops.", 0, "Walking distance to shops.")
     ]
 
     for _ in range(num_samples):

src/trainer/train.py

@@ -65,10 +65,16 @@ def load_data(file_path: str) -> Dataset:
         logger.critical("Failed to parse JSON data.")
         sys.exit(1)
 
-    # Format for HF Dataset: list of dicts -> dict of lists
+    # Label mapping to handle both strings and integers
+    label_map = {"COMPLIANT": 0, "NON_COMPLIANT": 1}
+    
     hf_data = {
         "text": [item["text"] for item in data],
-        "label": [item["label"] for item in data]
+        "label": [
+            item["label"] if isinstance(item["label"], int) 
+            else label_map.get(item["label"], 0) 
+            for item in data
+        ]
     }
     logger.info(f"Successfully loaded [bold green]{len(data)}[/bold green] examples.")
     return Dataset.from_dict(hf_data)