Trixy7/trixy_core/stt/layers/__init__.py

# -*- coding: utf-8 -*-
"""
Korrektur-Schichten fuer die STT-Nachkorrektur.

Jede Schicht ist optional — fehlende Libraries werden uebersprungen.
"""

from trixy_core.stt.layers.base import CorrectionLayer
from trixy_core.stt.layers.symspell_layer import SymSpellLayer
from trixy_core.stt.layers.phonetic_layer import PhoneticLayer
from trixy_core.stt.layers.kenlm_layer import KenLMLayer
from trixy_core.stt.layers.jamspell_layer import JamSpellLayer
from trixy_core.stt.layers.hunspell_layer import HunspellLayer
from trixy_core.stt.layers.languagetool_layer import LanguageToolLayer

# Registry: Name → Layer-Klasse
LAYER_REGISTRY: dict[str, type[CorrectionLayer]] = {
    "symspell": SymSpellLayer,
    "phonetic": PhoneticLayer,
    "kenlm": KenLMLayer,
    "jamspell": JamSpellLayer,
    "hunspell": HunspellLayer,
    "languagetool": LanguageToolLayer,
}

__all__ = [
    "CorrectionLayer",
    "LAYER_REGISTRY",
    "SymSpellLayer",
    "PhoneticLayer",
    "KenLMLayer",
    "JamSpellLayer",
    "HunspellLayer",
    "LanguageToolLayer",
]