import os

API_KEY = os.environ.get("GEMINI_API_KEY")

# Modèle pour des choses très légères
MODEL_LITE_ID = "gemini-3.5-flash-lite"

# Modèle pour identifier visuellement les labels
MODEL_FOR_LABEL_ID = "gemini-3.5-flash-lite"

# Modèle pour des choses normales
MODEL_FLASH_ID = "gemini-3.6-flash"

# Modèle pour des choses dures
MODEL_PRO_ID = "gemini-3.6-flash"
# MODEL_PRO_ID = "gemini-3.1-pro-preview"

PAGE_SPLITTER_KB = {
    "move_left": "<Left>",
    "move_right": "<Right>",
    "confirm_next": "<Return>",
    "rotate_page": "r",
    "rotate_all_pages": "R",
    "rotate_all_files": ",",
    "keep_left": "q",
    "keep_right": "d",
    "keep_as_is": "f",
    "next_page": "s",
    "discard_page": "z",
    "send_end": "a", # Send this page to the end
    "restart_file": "T",
    "arranger": "A", # Call `pdf arranger` software, if available
    "prev_file": "P",
}


PLOTTING_KB = {
    "OK": "<Return>",
    "previous": "p",
    "edit": "e",
    "open pdf": "o",
    "open original pdf": "O",
    "open eval": "i",
}

### Latex templates

    ### Pour la génération d'énoncés

LATEX_BEFORE = r"""\documentclass[varwidth=24.8cm,margin=0.4cm]{standalone}
\usepackage[utf8]{inputenc}
\usepackage[T1]{fontenc}
\usepackage{lmodern}
\usepackage{amsmath, amssymb}
\usepackage{commands}
\usepackage{minted}
\usepackage{graphicx}
\usepackage{enumitem}
\begin{document}
    \begin{minipage}{24.8cm}
"""

LATEX_AFTER = r"""
    \end{minipage}
\end{document}
"""


    ### Pour les annotations sur copies


from string import Template

LATEX_ANOT_BEFORE = Template(
    r"""\documentclass[varwidth=${width_in}in,margin=0.2cm]{standalone}
\usepackage[utf8]{inputenc}
\usepackage[T1]{fontenc}
\usepackage{lmodern}  % Enables arbitrary font scaling
\usepackage{amsmath, amssymb}
\usepackage{mathabx} % larger inline operators.
\usepackage{commands}
%\usepackage{anyfontsize} % replaced by lmodern
\begin{document}
\fontsize{${fontsize}}{${line_spacing}}\selectfont
"""
)

LATEX_ANOT_AFTER = r"""
\end{document}
"""