File size: 1,430 Bytes
88fe4e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
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()