From f0eaffacd3e174646d59918be5cc7966bb1d8211 Mon Sep 17 00:00:00 2001 From: Silas Kieser Date: Tue, 21 Jan 2025 14:59:36 +0100 Subject: [PATCH] improve logging in whisper_online.py --- src/whisper_streaming/online_asr.py | 2 +- whisper_online.py | 19 +++++++++++-------- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/src/whisper_streaming/online_asr.py b/src/whisper_streaming/online_asr.py index 207da01..26f6611 100644 --- a/src/whisper_streaming/online_asr.py +++ b/src/whisper_streaming/online_asr.py @@ -151,7 +151,7 @@ class OnlineASRProcessor: """ prompt, non_prompt = self.prompt() - logger.debug(f"PROMPT: {prompt}") + logger.debug(f"PROMPT(previous): {prompt}") logger.debug(f"CONTEXT: {non_prompt}") logger.debug( f"transcribing {len(self.audio_buffer)/self.SAMPLING_RATE:2.2f} seconds from {self.buffer_time_offset:2.2f}" diff --git a/whisper_online.py b/whisper_online.py index 9a61c2b..cd5d005 100644 --- a/whisper_online.py +++ b/whisper_online.py @@ -323,15 +323,18 @@ if __name__ == "__main__": if now is None: now = time.time() - start if o[0] is not None: - print( - "%1.4f %1.0f %1.0f %s" % (now * 1000, o[0] * 1000, o[1] * 1000, o[2]), - file=logfile, - flush=True, - ) - print( - "%1.4f %1.0f %1.0f %s" % (now * 1000, o[0] * 1000, o[1] * 1000, o[2]), - flush=True, + log_string = f"{now*1000:1.0f}, {o[0]*1000:1.0f}-{o[1]*1000:1.0f} ({(now-o[1]):+1.0f}s): {o[2]}" + + logger.debug( + log_string ) + + if logfile is not None: + print( + log_string, + file=logfile, + flush=True, + ) else: # No text, so no output pass