constants.py

import os

LID_LANGUAGES = {
    "eng_Latn": "english",
    'zho_Hans': "chinese",
    'zho_Hant': "chinese",
    'deu_Latn': "german",
    'spa_Latn': "spanish",
    'rus_Cyrl': "russian",
    'kor_Hang': "korean",
    'fra_Latn': "french",
    'jpn_Jpan': "japanese",
    'por_Latn': "portuguese",
    'tur_Latn': "turkish",
    'pol_Latn': "polish",
    'nld_Latn': "dutch",
    'arb_Arab': "arabic",
    'ita_Latn': "italian",
    'ind_Latn': "indonesian",
    'hin_Deva': "hindi",
    'vie_Latn': "vietnamese",
    'heb_Hebr': "hebrew",
    'ukr_Cyrl': "ukrainian",
    'ell_Grek': "greek",
    'ces_Latn': "czech",
    'ron_Latn': "romanian",
    'pes_Arab': "persian",
}

NEETS_AI_LANGID_MAP={
    'english': 'eng-1', 
    'chinese': 'zho-1', 
    'german': 'deu-1', 
    'spanish': 'spa-1', 
    'russian': 'rus-1', 
    'korean': 'kor-1', 
    'french': 'fra-1', 
    'portuguese': 'por-2', 
    'turkish': 'tur-1', 
    'polish': 'pol-2', 
    'dutch': 'nld-3', 
    'arabic': 'ara-1', 
    'italian': 'ita-1', 
    'indonesian': 'ind-1', 
    'hindi': 'hin-1', 
    'vietnamese': 'vie-1', 
    'hebrew': 'heb-1', 
    'ukrainian': 'ukr-1', 
    'greek': 'ell-1', 
    'czech': 'cze-1', 
    'romanian': 'ron-1', 
    'persian': 'fas-1'
}

AYA_MODEL_NAME = "c4ai-aya-expanse-32b"
BATCH_SIZE = 8
USE_REPLICATE = True
USE_ELVENLABS = True
IMAGE_PER_CONVERSATION_LIMIT = 4
DEFAULT_SYSTEM_PREAMBLE_TOKEN_COUNT = 4433
VISION_MODEL_TOKEN_LIMIT = 16000
VISION_COHERE_MODEL_NAME = os.getenv("AYA_VISION_MODEL_NAME")