import os

# 运行模式：mock（默认，无模型文件）| onnx（加载 best.onnx 真实推理）
MODEL_MODE = os.getenv("MODEL_MODE", "mock")
MODEL_PATH = os.getenv("MODEL_PATH", "models/best.onnx")
# YOLO 类别（二分类训练基线：healthy/sick；7 类病种扩展后再调整）
MODEL_LABELS = os.getenv("MODEL_LABELS", "healthy,sick").split(",")
MODEL_VERSION = os.getenv("MODEL_VERSION", "mock-2026.08.14" if MODEL_MODE == "mock" else "best.onnx")
# mock 模式返回的固定结果
MOCK_CLASS = os.getenv("MOCK_CLASS", "healthy")
MOCK_CONFIDENCE = float(os.getenv("MOCK_CONFIDENCE", "0.95"))
INTERNAL_API_KEY = os.getenv("INTERNAL_API_KEY", "silk-internal-2026")
ALLOWED_STREAM_HOSTS = [
    item.strip()
    for item in os.getenv("ALLOWED_STREAM_HOSTS", "localhost,127.0.0.1,100.83.103.1").split(",")
    if item.strip()
]
STREAM_TASK_MAX_WORKERS = int(os.getenv("STREAM_TASK_MAX_WORKERS", "2"))
STREAM_TASK_MAX_FRAMES = int(os.getenv("STREAM_TASK_MAX_FRAMES", "10"))