From 5ebbed3bd70bc51294113a8ff08af074b2d925ea Mon Sep 17 00:00:00 2001
From: Alex Young <alex@blackkettle.org>
Date: Sun, 14 Apr 2024 14:24:59 +0100
Subject: [PATCH] Turn prints into logging.debug calls in
 whisper_online_server.py

---
 whisper_online_server.py | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/whisper_online_server.py b/whisper_online_server.py
index b2f5120..98022e0 100644
--- a/whisper_online_server.py
+++ b/whisper_online_server.py
@@ -4,17 +4,27 @@ from whisper_online import *
 import sys
 import argparse
 import os
+import logging
+
 parser = argparse.ArgumentParser()
 
 # server options
 parser.add_argument("--host", type=str, default='localhost')
 parser.add_argument("--port", type=int, default=43007)
 
+parser.add_argument("-l", "--log-level", dest="log_level", 
+                    choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'],
+                    help="Set the log level",
+                    default='INFO')
+
 
 # options from whisper_online
 add_shared_args(parser)
 args = parser.parse_args()
 
+if args.log_level:
+    logging.basicConfig(format='whisper-server-%(levelname)s: %(message)s',
+                        level=getattr(logging, args.log_level))
 
 # setting whisper object by args 
 
@@ -24,7 +34,7 @@ size = args.model
 language = args.lan
 
 t = time.time()
-print(f"Loading Whisper {size} model for {language}...",file=sys.stderr,end=" ",flush=True)
+logging.debug(f"Loading Whisper {size} model for {language}...")
 
 if args.backend == "faster-whisper":
     from faster_whisper import WhisperModel
@@ -44,10 +54,10 @@ else:
     tgt_language = language
 
 e = time.time()
-print(f"done. It took {round(e-t,2)} seconds.",file=sys.stderr)
+logging.debug(f"done. It took {round(e-t,2)} seconds.")
 
 if args.vad:
-    print("setting VAD filter",file=sys.stderr)
+    logging.debug("setting VAD filter")
     asr.use_vad()
 
 
@@ -70,9 +80,7 @@ if os.path.exists(demo_audio_path):
     # warm up the ASR, because the very first transcribe takes much more time than the other
     asr.transcribe(a)
 else:
-    print("Whisper is not warmed up",file=sys.stderr)
-
-
+    logging.info("Whisper is not warmed up")
 
 
 ######### Server objects
@@ -80,9 +88,6 @@ else:
 import line_packet
 import socket
 
-import logging
-
-
 class Connection:
     '''it wraps conn object'''
     PACKET_SIZE = 65536
@@ -191,11 +196,6 @@ class ServerProcessor:
 
 
 
-
-# Start logging.
-level = logging.INFO
-logging.basicConfig(level=level, format='whisper-server-%(levelname)s: %(message)s')
-
 # server loop
 
 with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: