Create config.py

config.py

@@ -1,26 +1,29 @@
-# config.py
-
 import os
 
-# Paths
+# === File and Directory Paths ===
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-DATA_PATH = os.path.join(BASE_DIR, "synthetic_transactions_samples_5000.csv")
+DATA_PATH = os.path.join(BASE_DIR, "data", "synthetic_transactions_samples_5000.csv")
 MODEL_DIR = os.path.join(BASE_DIR, "models")
 TFIDF_VECTORIZER_PATH = os.path.join(MODEL_DIR, "tfidf_vectorizer.pkl")
-LR_MODEL_PATH = os.path.join(MODEL_DIR, "logreg_model.pkl")
-XGB_MODEL_PATH = os.path.join(MODEL_DIR, "xgb_model.pkl")
-LGBM_MODEL_PATH = os.path.join(MODEL_DIR, "lgbm_model.pkl")
+VOTING_MODEL_PATH = os.path.join(MODEL_DIR, "voting_classifier.pkl")
+LABEL_ENCODERS_DIR = os.path.join(MODEL_DIR, "label_encoders")
 
-# Columns
+# === Data Column Configuration ===
 TEXT_FIELDS = [
-    "Transaction_Id", "Origin", "Designation", "Keywords", "Name", "SWIFT_Tag", "Currency", "Entity",
-    "Message", "City", "Country", "State", "Hit_Type", "Record_Matching_String", "WatchList_Match_String",
-    "Payment_Sender_Name", "Payment_Reciever_Name", "Swift_Message_Type", "Text_Sanction_Data",
-    "Matched_Sanctioned_Entity", "Red_Flag_Reason", "Risk_Level", "Risk_Score", "CDD_Level", "PEP_Status",
-    "Sanction_Description", "Checker_Notes", "Sanction_Context", "Maker_Action", "Customer_Type", "Industry",
-    "Transaction_Type", "Transaction_Channel", "Geographic_Origin", "Geographic_Destination", "Risk_Category",
-    "Risk_Drivers", "Alert_Status", "Investigation_Outcome", "Source_Of_Funds", "Purpose_Of_Transaction",
-    "Beneficial_Owner"
+    "Transaction_Id", "Origin", "Designation", "Keywords", "Name", "SWIFT_Tag",
+    "Currency", "Entity", "Message", "City", "Country", "State", "Hit_Type",
+    "Record_Matching_String", "WatchList_Match_String", "Payment_Sender_Name",
+    "Payment_Reciever_Name", "Swift_Message_Type", "Text_Sanction_Data",
+    "Matched_Sanctioned_Entity", "Red_Flag_Reason", "Risk_Level", "Risk_Score",
+    "CDD_Level", "PEP_Status", "Sanction_Description", "Checker_Notes",
+    "Sanction_Context", "Maker_Action", "Customer_Type", "Industry",
+    "Transaction_Type", "Transaction_Channel", "Geographic_Origin",
+    "Geographic_Destination", "Risk_Category", "Risk_Drivers",
+    "Alert_Status", "Investigation_Outcome", "Source_Of_Funds",
+    "Purpose_Of_Transaction", "Beneficial_Owner"
 ]
 
-LABEL_COLUMNS = ["Maker_Action", "Escalation_Level", "Risk_Category", "Risk_Drivers", "Investigation_Outcome"]
+LABEL_COLUMNS = [
+    "Maker_Action", "Escalation_Level", "Risk_Category",
+    "Risk_Drivers", "Investigation_Outcome"
+]