소스 검색

[1.1] FIx better logging

Penta 9 달 전
부모
커밋
5673b0185d
1개의 변경된 파일과 8개의 추가작업 그리고 3개의 파일을 삭제
  1. 8 3
      app.py

+ 8 - 3
app.py

@@ -1,11 +1,13 @@
-from fastapi import FastAPI, UploadFile, File, WebSocket, WebSocketDisconnect
-from faster_whisper import WhisperModel
 import tempfile
 import tempfile
 import os
 import os
 import numpy as np
 import numpy as np
 import time
 import time
 import logging
 import logging
 import os
 import os
+import sys
+
+from fastapi import FastAPI, UploadFile, File, WebSocket, WebSocketDisconnect
+from faster_whisper import WhisperModel
 
 
 def env_bool(name: str, default: bool = False) -> bool:
 def env_bool(name: str, default: bool = False) -> bool:
     return os.getenv(name, str(default)).lower() in ("1", "true", "yes", "on")
     return os.getenv(name, str(default)).lower() in ("1", "true", "yes", "on")
@@ -15,10 +17,13 @@ VAD_ENABLED = env_bool("WHISPER_VAD", True)
 
 
 logging.basicConfig(
 logging.basicConfig(
     level=getattr(logging, LOG_LEVEL, logging.INFO),
     level=getattr(logging, LOG_LEVEL, logging.INFO),
-    format="%(asctime)s [%(levelname)s] %(message)s"
+    format="%(asctime)s [%(levelname)s] [%(name)s] %(message)s",
+    stream=sys.stdout,
+    force=True
 )
 )
 
 
 logger = logging.getLogger("whisper-ws")
 logger = logging.getLogger("whisper-ws")
+logger.setLevel(getattr(logging, LOG_LEVEL, logging.INFO))
 
 
 logger.info(f"Log level = {LOG_LEVEL}")
 logger.info(f"Log level = {LOG_LEVEL}")
 logger.info(f"VAD enabled = {VAD_ENABLED}")
 logger.info(f"VAD enabled = {VAD_ENABLED}")