diff --git a/.gitignore b/.gitignore index 88eef4b4..f8a644c5 100644 --- a/.gitignore +++ b/.gitignore @@ -129,4 +129,6 @@ dmypy.json .pyre/ *.wav -run_*.sh \ No newline at end of file +run_*.sh +pixi* +.pixi \ No newline at end of file diff --git a/whisper_fastapi_online_server.py b/whisper_fastapi_online_server.py index d55e1046..476fab36 100644 --- a/whisper_fastapi_online_server.py +++ b/whisper_fastapi_online_server.py @@ -22,11 +22,22 @@ def format_time(seconds): return str(timedelta(seconds=int(seconds))) -logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s") +# Configure logging for all modules +logging.basicConfig(level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s") + +# Set root logger level logging.getLogger().setLevel(logging.WARNING) + + + + +# Configure main module logger logger = logging.getLogger(__name__) logger.setLevel(logging.DEBUG) + + + ##### LOAD ARGS ##### parser = argparse.ArgumentParser(description="Whisper FastAPI Online Server") @@ -70,6 +81,13 @@ def format_time(seconds): add_shared_args(parser) args = parser.parse_args() + +# Configure specific loggers for different modules +logging.getLogger("whisper_streaming_custom").setLevel(logging.getLevelName(args.log_level)) +logging.getLogger("diarization").setLevel(logging.getLevelName(args.log_level)) + + + SAMPLE_RATE = 16000 CHANNELS = 1 SAMPLES_PER_SEC = SAMPLE_RATE * int(args.min_chunk_size)