logging_config.py

import logging
import io

# StringIO buffer to capture logs for streaming
log_buffer = io.StringIO()

logger = logging.getLogger("my_logger")
logger.setLevel(logging.DEBUG)

# log handler that writes to our StringIO buffer
log_handler = logging.StreamHandler(log_buffer)
log_handler.setLevel(logging.DEBUG)

logger.addHandler(log_handler)
# prevent duplicate logs
logger.propagate = False