fix(logging) - corrected indentation and setup in global_logger.py

Services/ml_model/global_logger.py

@@ -9,17 +9,17 @@ def __init__(self, log_file=None):
         ch = logging.StreamHandler()
         ch.setLevel(logging.DEBUG)
 
-        # Create file handler if log_file is provided
+        # Create formatter
+        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s')
+        ch.setFormatter(formatter)
+        self.logger.addHandler(ch)
+
+        # Add file handler if specified
         if log_file:
             fh = logging.FileHandler(log_file)
             fh.setLevel(logging.DEBUG)
-            formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s')
             fh.setFormatter(formatter)
             self.logger.addHandler(fh)
-
-        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s')
-        ch.setFormatter(formatter)
-        self.logger.addHandler(ch)
 
     def info(self, message, extra=None):
         if extra is None:

Services/ml_model/test_logger.py

@@ -0,0 +1,15 @@
+from global_logger import global_logger
+
+def test_logger():
+    # Test different log levels
+    global_logger.info("Test info message")
+    global_logger.error("Test error message")
+    global_logger.debug("Test debug message")
+
+    # Test with extra data
+    global_logger.info("Test with data", extra={"data": "test_value"})
+
+    print("Check the log file and console output for messages")
+
+if __name__ == "__main__":
+    test_logger()