Upd logger module

requirements.txt

@@ -21,5 +21,6 @@ transformers
 torch
 sentencepiece
 sacremoses
+protobuf
 
 # Note: Heavy local LLM dependencies (accelerate, bitsandbytes, etc.) are excluded for cloud mode

utils/augment.py

@@ -5,6 +5,13 @@ import random
 from typing import Dict, Tuple
 import ftfy
 import langid
+import logging
+
+# Module logger
+logger = logging.getLogger("augment")
+if not logger.handlers:
+    logger.setLevel(logging.INFO)
+    logger.addHandler(logging.StreamHandler())
 
 P_EMAIL = re.compile(r"[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,}")
 P_PHONE = re.compile(r"(?:(?:\+?\d{1,3})?[\s-]?)?(?:\(?\d{2,4}\)?[\s-]?)?\d{3,4}[\s-]?\d{3,4}")