config.py

"""
Конфигурация и константы для NER-приложения.
"""

# Лимиты
MAX_CHARS = 2000
MAX_BATCH_ROWS = 100
HISTORY_SIZE = 10

# Доступные модели NER
MODELS = {
    "WikiNEuRal (multilingual)": "Babelscape/wikineural-multilingual-ner",
    "BERT Russian NER": "Gherman/bert-base-NER-Russian"
}

# Цветовая схема для подсветки сущностей
COLOR_MAP = {
    "PER": "#3b82f6",      # Синий — персоны
    "ORG": "#22c55e",      # Зелёный — организации
    "LOC": "#f97316",      # Оранжевый — локации
    "MISC": "#a855f7"      # Фиолетовый — прочее
}

# Описания типов сущностей
ENTITY_LABELS = {
    "PER": "Персона (ФИО)",
    "ORG": "Организация",
    "LOC": "Локация (место)",
    "MISC": "Прочее"
}