update logging with debug env variable

pull/314/head
Ben Xu 2 months ago
parent 07672f498b
commit 178ffc83bd

@ -1,17 +1,20 @@
from livekit.rtc import VideoStream from livekit.rtc import VideoStream
from livekit.agents import JobContext from livekit.agents import JobContext
from datetime import datetime from datetime import datetime
from livekit.agents.pipeline import VoicePipelineAgent import os
from livekit.rtc import VideoFrame from livekit.rtc import VideoFrame
import asyncio import asyncio
# Define the path to the log file # Define the path to the log file
LOG_FILE_PATH = 'video_processor.txt' LOG_FILE_PATH = 'video_processor.txt'
DEBUG = os.getenv('DEBUG', 'false').lower() == 'true'
def log_message(message: str): def log_message(message: str):
"""Append a message to the log file with a timestamp.""" """Append a message to the log file with a timestamp."""
if not DEBUG:
return
timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S') timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
with open(LOG_FILE_PATH, 'a') as log_file: with open(LOG_FILE_PATH, 'a') as log_file:
log_file.write(f"{timestamp} - {message}\n") log_file.write(f"{timestamp} - {message}\n")

Loading…
Cancel
Save