MODELS = {
    "GPT-4o": "gpt-4o",
    "Gemini 1.5 Pro": "gemini-1.5-pro",
    "Gemini 2.0 Flash": "gemini-2.0-flash",
    "Claude 3.5 Sonnet": "claude-3-5-sonnet-20241022",
    "Claude 3.7 Sonnet": "claude-3-7-sonnet-latest",
}

MODEL_NAME_TO_CHOICE = {
    "gpt-4o": "GPT-4o",
    "gemini-1.5-pro": "Gemini 1.5 Pro",
    "gemini-2.0-flash": "Gemini 2.0 Flash",
    "claude-3-5-sonnet-20241022": "Claude 3.5 Sonnet",
    "claude-3-7-sonnet-latest": "Claude 3.7 Sonnet",
}

MODELS_PATH = {
    "GPT-4o": "gpt_4o",
    "Gemini 1.5 Pro": "gemini_1_5_pro",
    "Gemini 2.0 Flash": "gemini_2_0_flash",
    "Claude 3.5 Sonnet": "claude_3_5_sonnet",
    "Claude 3.7 Sonnet": "claude_3_7_sonnet",
}

BOOKS = {
    # "Іван Нечуйлевицький - Кайдашева Сім'я": "nechuy-levytskyy-ivan-semenovych-kaydasheva-simia.json",
    "Іван Франко - Захар Беркут": "zahar-berkut.json",
}

CLASSES = ["7-A", "7-B", "7-C", "7-D"]

STUDENTS = {
    "7-A": [
        "Гайворонський Владислав",
        "Гордін Георгій",
        "Гуменний Станіслав",
        "Денчик Данило",
        "Кичила Ілля",
        "Климчкук Марія",
        "Коньков Владислав",
        "Молчанов Лев",
        "Нємець Майя",
        "Панасенко Микита",
        "Паславська Марія",
        "Рипаленко Артем",
        "Савченко Роман",
        "Смолкіна Соф'я",
        "Страховецька Софія",
    ],
    "7-B": [
        "Авдєєва Аліса",
        "Агапов Олександр",
        "Горюн Любава",
        "Доротюк Варвара",
        "Дробот Дмитро",
        "Зіоменко Олег",
        "Золотько Варвара",
        "Ільєнко Софія",
        "Книш Ксенія",
        "Кувалдіна Катерина",
        "Лемішев Кирило",
        "Хоменко Валерія",
        "Хорошавіна Софія",
        "Шепеленко Віола",
        "Щербина Герасим",
    ],
    "7-C": [
        "Безрук Дар'я",
        "Бець Вероніка",
        "Головньова Владислава",
        "Данченков Микита",
        "Кадничанський Іосиф",
        "Конопельцева Мілана",
        "Муковоз Гліб",
        "Положій Захар",
        "Поручіков Владислав",
        "Прокіпець Вероніка",
        "Сирота Дмитро",
        "Сомов Максим",
        "Таран Вікторія",
        "Чумачок Софія",
        "Шигалевська Марія",
    ],
    "7-D": [
        "Капустеринська Аліса",
        "Леонідова Мілада",
        "Олійник Давід",
        "Шевченко Олександра",
        "Щербакова Марія",
    ]
}