File size: 614 Bytes
68f9986 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import logging
import collections
class GradioLogHandler(logging.Handler):
def __init__(self, log_queue):
super().__init__()
self.log_queue = log_queue
self.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
def emit(self, record):
self.log_queue.append(self.format(record))
def setup_gradio_logging():
log_queue = collections.deque(maxlen=1000)
gradio_handler = GradioLogHandler(log_queue)
logging.getLogger().setLevel(logging.INFO)
logging.getLogger().addHandler(gradio_handler)
return log_queue, gradio_handler
|