diff --git a/whisper_fastapi_online_server.py b/whisper_fastapi_online_server.py index 33391be..3771c4e 100644 --- a/whisper_fastapi_online_server.py +++ b/whisper_fastapi_online_server.py @@ -11,6 +11,65 @@ from fastapi.middleware.cors import CORSMiddleware from src.whisper_streaming.whisper_online import backend_factory, online_factory, add_shared_args + +import logging +import logging.config + +def setup_logging(): + logging_config = { + 'version': 1, + 'disable_existing_loggers': False, + 'formatters': { + 'standard': { + 'format': '%(asctime)s - %(name)s - %(levelname)s - %(message)s', + }, + }, + 'handlers': { + 'console': { + 'level': 'DEBUG', + 'class': 'logging.StreamHandler', + 'formatter': 'standard', + }, + }, + 'root': { + 'handlers': ['console'], + 'level': 'DEBUG', + }, + 'loggers': { + 'uvicorn': { + 'handlers': ['console'], + 'level': 'INFO', + 'propagate': False, + }, + 'uvicorn.error': { + 'level': 'INFO', + }, + 'uvicorn.access': { + 'level': 'INFO', + }, + 'src.whisper_streaming.online_asr': { # Add your specific module here + 'handlers': ['console'], + 'level': 'DEBUG', + 'propagate': False, + }, + 'src.whisper_streaming.whisper_streaming': { # Add your specific module here + 'handlers': ['console'], + 'level': 'DEBUG', + 'propagate': False, + }, + }, + } + + logging.config.dictConfig(logging_config) + +setup_logging() +logger = logging.getLogger(__name__) + + + + + + app = FastAPI() app.add_middleware( CORSMiddleware, @@ -238,5 +297,6 @@ if __name__ == "__main__": import uvicorn uvicorn.run( - "whisper_fastapi_online_server:app", host=args.host, port=args.port, reload=True + "whisper_fastapi_online_server:app", host=args.host, port=args.port, reload=True, + log_level="info" )