from app.specialties.cardiology import CARDIOLOGY_CONFIG
from app.specialties.diabetology import DIABETOLOGY_CONFIG
from app.specialties.general import GENERAL_CONFIG
from app.specialties.mental_health import MENTAL_HEALTH_CONFIG

SPECIALTIES = {
    "cardiology": CARDIOLOGY_CONFIG,
    "diabetology": DIABETOLOGY_CONFIG,
    "general": GENERAL_CONFIG,
    "mental_health": MENTAL_HEALTH_CONFIG,
}


def get_specialty_config(code: str) -> dict:
    return SPECIALTIES.get(
        code,
        GENERAL_CONFIG,
    )