from tts_webui.magnet.utils import Timer


def log_function_time(fn):
    def wrapper(*args, **kwargs):
        with Timer():
            return fn(*args, **kwargs)

    return wrapper


def log_generator_time(generator):
    def wrapper(*args, **kwargs):
        with Timer():
            yield from generator(*args, **kwargs)

    return wrapper