"""
Gradio Demo 配置文件
农村低压配电网电压异常检测项目 - HuggingFace Spaces 版本
"""

import os
from pathlib import Path

# 路径配置 (HuggingFace Spaces 兼容)
DEMO_DIR = Path(__file__).parent
CODE_DIR = DEMO_DIR  # 在 Spaces 中 demo 就是根目录
PROJECT_DIR = DEMO_DIR

# 模型路径 (使用 HuggingFace Hub)
MODEL_DIR = DEMO_DIR / "models"
BEST_MODEL_PATH = None  # 将从 HuggingFace Hub 下载
MODEL_CONFIG_PATH = None

# 数据路径
DATASET_DIR = DEMO_DIR / "dataset"
RURAL_VOLTAGE_DIR = DATASET_DIR / "RuralVoltage" / "realistic_v2"
PSM_DIR = DATASET_DIR / "PSM"

# 预计算数据路径
PRECOMPUTED_DIR = DEMO_DIR / "precomputed"

# SVG 文件路径
SVG_DIR = DEMO_DIR / "docs" / "model_architectures" / "svg"

# 模型配置
MODEL_CONFIGS = {
    "VoltageTimesNet_v2": {
        "enc_in": 16,
        "c_out": 16,
        "seq_len": 100,
        "d_model": 64,
        "d_ff": 64,
        "e_layers": 2,
        "top_k": 5,
        "num_kernels": 6,
    },
    "TimesNet": {
        "enc_in": 16,
        "c_out": 16,
        "seq_len": 100,
        "d_model": 64,
        "d_ff": 64,
        "e_layers": 2,
        "top_k": 5,
        "num_kernels": 6,
    },
    "DLinear": {
        "enc_in": 16,
        "seq_len": 100,
        "pred_len": 100,
        "individual": False,
    },
}

# 可视化配色方案 (柔和科研风格)
THESIS_COLORS = {
    "primary": "#4878A8",
    "secondary": "#72A86D",
    "accent": "#C4785C",
    "warning": "#D4A84C",
    "neutral": "#808080",
    "light_gray": "#B0B0B0",
    "anomaly": "#E74C3C",
    "normal": "#2ECC71",
}

# 模型对比颜色
MODEL_COLORS = {
    "VoltageTimesNet_v2": "#4878A8",
    "VoltageTimesNet": "#72A86D",
    "TimesNet": "#C4785C",
    "TPATimesNet": "#D4A84C",
    "MTSTimesNet": "#9B59B6",
    "DLinear": "#808080",
}

# Gradio 主题配置
GRADIO_THEME = "soft"

# 推理配置
INFERENCE_CONFIG = {
    "batch_size": 32,
    "device": "cpu",
    "default_threshold": 0.5,
}

# 演示数据配置
DEMO_DATA_CONFIG = {
    "sample_length": 1000,
    "window_size": 100,
    "step_size": 1,
}

# HuggingFace Hub 配置
HF_MODEL_REPO = "Sheldon123z/rural-voltage-detection-models"
HF_DATASET_REPO = "Sheldon123z/rural-voltage-datasets"