58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Helper script to adjust whisper ASR logging levels for debugging.
|
|
Run this to see more detailed ASR logging.
|
|
"""
|
|
|
|
import logging
|
|
import sys
|
|
import os
|
|
|
|
# Add the project root (parent of the voicebot directory) to sys.path so
|
|
# imports like `from shared import ...` work when running this script from
|
|
# inside the `voicebot` container.
|
|
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
if project_root not in sys.path:
|
|
sys.path.append(project_root)
|
|
|
|
from shared.logger import logger
|
|
|
|
|
|
def set_debug_logging():
|
|
"""Set logger to DEBUG level for detailed ASR logging."""
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
# Also set the root logger
|
|
logging.getLogger().setLevel(logging.DEBUG)
|
|
|
|
# Create a more detailed formatter if needed
|
|
formatter = logging.Formatter(
|
|
"%(asctime)s - %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s"
|
|
)
|
|
|
|
# Update all handlers
|
|
for handler in logger.handlers:
|
|
handler.setLevel(logging.DEBUG)
|
|
handler.setFormatter(formatter)
|
|
|
|
logger.info("Debug logging enabled for Whisper ASR")
|
|
|
|
|
|
def set_info_logging():
|
|
"""Set logger back to INFO level."""
|
|
logger.setLevel(logging.INFO)
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
|
|
# Update all handlers
|
|
for handler in logger.handlers:
|
|
handler.setLevel(logging.INFO)
|
|
|
|
logger.info("Info logging enabled for Whisper ASR")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1 and sys.argv[1] == "info":
|
|
set_info_logging()
|
|
else:
|
|
set_debug_logging()
|