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 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",
}