Enhance logging functionality to support console-only output option

Author: DefinetlyNotAI

v4n1_Trainer.py

@@ -12,13 +12,13 @@
 # ---------------- INIT ----------------
 def init(config: TrainingConfig) -> dict:
     """Initialize static, config-free resources (only once)."""
-    log("Loading GPT-Neo tokenizer/model (static init)...", cfg=config)
+    log("Loading GPT-Neo tokenizer/model (static init)...", cfg=config, only_console=True)
     gpt_tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
     gpt_model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
     if gpt_tokenizer.pad_token is None:
         gpt_tokenizer.pad_token = gpt_tokenizer.eos_token
 
-    log("Loading MiniLM for embeddings (static init)...", cfg=config)
+    log("Loading MiniLM for embeddings (static init)...", cfg=config, only_console=True)
     embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 
     return {

vulnscan/log.py

@@ -3,8 +3,12 @@
 from vulnscan.config import TrainingConfig
 
 
-def log(message: str, cfg: TrainingConfig, silent: bool = False):
+def log(message: str, cfg: TrainingConfig, silent: bool = False, only_console: bool = False):
+    """Log a message to console and log file."""
+    if only_console and silent:
+        return  # nothing to do
     if not silent:
         print(message)
-    with open(cfg.LOG_FILE, "a") as f:
-        f.write(f"{datetime.now()} | {message}\n")
+    if not only_console:
+        with open(cfg.LOG_FILE, "a") as f:
+            f.write(f"{datetime.now()} | {message}\n")