"""
Configuración global del proyecto.
"""

import torch

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32

# IDs de modelos en Hugging Face
OCR_MODEL_ID = "lightonai/LightOnOCR-2-1B"
TRANSLATION_MODEL_ID = "facebook/nllb-200-distilled-600M"
DESCRIPTION_MODEL_ID = "PlanTL-GOB-ES/gpt2-large-bne"

# Códigos de idioma para NLLB
LANG_CODES = {
    "Spanish": "spa_Latn",
    "English": "eng_Latn",
    "French": "fra_Latn",
    "German": "deu_Latn",
    "Italian": "ita_Latn",
}

# Límites
MAX_DISHES = 20
MAX_QUEUE_SIZE = 5
SERVER_PORT = 7860