|
|
@@ -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}")
|