from src.display.utils import ModelType
from src.leaderboard.read_evals import get_all_strategies
from src.language import lang

def get_model_type_choices():
    """获取模型类型筛选选项，与ModelType枚举兼容"""
    # 生成带符号的模型类型选项
    type_choices = [mt.to_str() for mt in ModelType if mt != ModelType.Unknown]
    # 在开头添加"All"选项
    return ["All"] + type_choices

def get_strategy_choices(results_path):
    """获取策略筛选选项"""
    return get_all_strategies(results_path)

def get_metric_categories():
    """
    获取指标类别分组，按照要求调整：
    - 平稳性和非平稳性合并为一类
    - 同方差和异方差合并为一类
    - 季节特性拆分为季节强度和季节数
    """
    return [
        {
            "name": lang.get("regular_metrics"),
            "prefix": "Regular_",
            "description": lang.get("regular_desc")
        },
        {
            "name": lang.get("stationarity"),
            "prefix": ["Stationary_", "Non_Stationary_"],
            "description": lang.get("stationarity_desc")
        },
        {
            "name": lang.get("trend_strength"),
            "prefix": "Trend_Strength_",
            "description": lang.get("trend_strength_desc")
        },
        {
            "name": lang.get("seasonality_strength"),
            "prefix": "Seasonality_Strength_",
            "description": lang.get("seasonality_strength_desc")
        },
        {
            "name": lang.get("seasonality_count"),
            "prefix": "Seasonality_Count_",
            "description": lang.get("seasonality_count_desc")
        },
        {
            "name": lang.get("volatility"),
            "prefix": "Volatility_",
            "description": lang.get("volatility_desc")
        },
        {
            "name": lang.get("memory"),
            "prefix": "Memory_",
            "description": lang.get("memory_desc")
        },
        {
            "name": lang.get("variance_characteristics"),
            "prefix": ["Homo-Scedasticity_", "Hetero-Scedasticity_"],
            "description": lang.get("variance_characteristics_desc")
        },
        {
            "name": lang.get("outliers"),
            "prefix": "Anomaly_",
            "description": lang.get("outliers_desc")
        }
    ]