api/config.py

import os

class Settings:
    MODEL_ID: str = os.getenv("MODEL_ID", "IDEA-Research/grounding-dino-base")
    BOX_THRESHOLD: float = float(os.getenv("BOX_THRESHOLD", 0.35))
    TEXT_THRESHOLD: float = float(os.getenv("TEXT_THRESHOLD", 0.25))
    MIN_CONFIDENCE: float = float(os.getenv("MIN_CONFIDENCE", 0.45))
    
    TEXT_LABELS: list[str] = [
        "car",
        "truck",
        "bus",
        "motorcycle",
        "person",
        "ambulance",
        "police vehicle",
        "fire",
        "smoke",
        "tree",
        "water",
        "building",
        "road",
        "debris",
        "damaged vehicle",
        "traffic congestion",
        "collapsed structure",
        "construction barrier"
    ]

settings = Settings()