# File 5: utils/logger.py
from torch.utils.tensorboard import SummaryWriter
import datetime

class Logger:
    def __init__(self, log_dir='logs'):
        timestamp = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
        self.writer = SummaryWriter(f"{log_dir}/{timestamp}")
        
    def log_metrics(self, step, metrics):
        for key, value in metrics.items():
            self.writer.add_scalar(key, value, step)
            
    def log_histogram(self, tag, values, step):
        self.writer.add_histogram(tag, values, step)
        
    def close(self):
        self.writer.close()