SightLine / src /config.py
LovnishVerma's picture
Upload 42 files
88fe4e7 verified
Raw
History Blame Contribute Delete
1.43 kB
import os
from dataclasses import dataclass
from typing import Dict
@dataclass
class Config:
"""Central configuration for SightLine."""
# Timing
CAPTURE_INTERVAL: float = float(os.getenv("SIGHTLINE_CAPTURE_INTERVAL", "3.0"))
SCENE_THRESHOLD: float = float(os.getenv("SIGHTLINE_SCENE_THRESHOLD", "0.10"))
DEBOUNCE_MS: int = int(os.getenv("SIGHTLINE_DEBOUNCE_MS", "800"))
MAX_DIM: int = int(os.getenv("SIGHTLINE_MAX_DIM", "768"))
HASH_SIZE: int = int(os.getenv("SIGHTLINE_HASH_SIZE", "16"))
# Audio
TTS_TIMEOUT: float = float(os.getenv("SIGHTLINE_TTS_TIMEOUT", "12.0"))
TTS_RATE: str = os.getenv("SIGHTLINE_TTS_RATE", "+8%")
AUDIO_FORMAT: str = os.getenv("SIGHTLINE_AUDIO_FORMAT", "mp3")
MAX_QUEUE_SIZE: int = int(os.getenv("SIGHTLINE_MAX_QUEUE_SIZE", "3"))
# Model
MODEL_NAME: str = os.getenv("SIGHTLINE_MODEL_NAME", "microsoft/Florence-2-base")
# Text generation limits for tasks
MAX_NEW_TOKENS: Dict[str, int] = None
# UI
APP_NAME: str = "SightLine"
APP_VERSION: str = "1.0"
def __post_init__(self):
if self.MAX_NEW_TOKENS is None:
self.MAX_NEW_TOKENS = {
"<CAPTION>": 64,
"<DETAILED_CAPTION>": 120,
"<MORE_DETAILED_CAPTION>": 200,
"<OD>": 256,
"<OCR>": 300,
}
CONFIG = Config()