From 178ffc83bd83cc1849b523650ee7f1d8d168092b Mon Sep 17 00:00:00 2001 From: Ben Xu Date: Mon, 9 Dec 2024 13:37:36 -0800 Subject: [PATCH] update logging with debug env variable --- software/source/server/livekit/video_processor.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/software/source/server/livekit/video_processor.py b/software/source/server/livekit/video_processor.py index 1dc7951..4167cfa 100644 --- a/software/source/server/livekit/video_processor.py +++ b/software/source/server/livekit/video_processor.py @@ -1,17 +1,20 @@ from livekit.rtc import VideoStream from livekit.agents import JobContext from datetime import datetime -from livekit.agents.pipeline import VoicePipelineAgent - +import os from livekit.rtc import VideoFrame import asyncio # Define the path to the log file LOG_FILE_PATH = 'video_processor.txt' +DEBUG = os.getenv('DEBUG', 'false').lower() == 'true' def log_message(message: str): """Append a message to the log file with a timestamp.""" + if not DEBUG: + return + timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S') with open(LOG_FILE_PATH, 'a') as log_file: log_file.write(f"{timestamp} - {message}\n")