소스 검색

[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 os
 import numpy as np
 import time
 import logging
 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:
     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(
     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.setLevel(getattr(logging, LOG_LEVEL, logging.INFO))
 
 logger.info(f"Log level = {LOG_LEVEL}")
 logger.info(f"VAD enabled = {VAD_ENABLED}")