config.py

import os

# === Paths ===
BASE_DIR = os.path.dirname(os.path.abspath(__file__))

# Data CSV location
DATA_PATH = os.path.join(BASE_DIR, "data", "synthetic_transactions_samples_5000.csv")

# Saved model and vectorizer paths
MODEL_SAVE_DIR = os.path.join(BASE_DIR, "saved_models")
MODEL_PATH = os.path.join(MODEL_SAVE_DIR, "logreg_model.pkl")
TFIDF_PATH = os.path.join(MODEL_SAVE_DIR, "tfidf_vectorizer.pkl")

# Input column (text)
TEXT_COLUMN = "Sanction_Context"

# Output label columns
LABEL_COLUMNS = [ "Red_Flag_Reason",
    "Maker_Action",
    "Escalation_Level",
    "Risk_Category",
    "Risk_Drivers",
    "Investigation_Outcome"
]