o
    5i                     @   s6  d Z ddlZddlmZ ddlmZ eejjjZeed  e	dZ
e	dZe	dZe	d	Ze	d
Ze	dZe	dZe	dZdddgddddg dddddddddddddddddddddddddZdddZdd ddd!d"d#Zd$d%d&d'd(d)d*d+d,d-d.d$d%d/Zd0ed1efd2d3ZdS )4z}
Configuration module for the transcription pipeline.
Loads environment variables and defines available models and settings.
    N)Path)load_dotenvz.envR2_ENDPOINT_URL	R2_BUCKETR2_ACCESS_KEY_IDR2_SECRET_ACCESS_KEY
GEMINI_KEYURLSUPABASE_ANONSUPABASE_ADMINgemini-3-pro-previewlowhighTpremium)namethinking_levelssupports_thinking	cost_tiergemini-3-flash-preview)minimalr   mediumr   standardgemini-2.5-progemini-2.5-flashgemini-2.5-flash-litelitegemini-2.0-flashF)r   r   r   r   r   r   )zgemini-flash-latestzgemini-flash-lite-latestg      $@g       @z/tmp/maya3_transcribez./transcriptions)max_segment_duration_secmin_segment_duration_secdefault_modeldefault_thinking_levelwork_dir
output_dirTeluguHindiTamilKannada	MalayalamBengaliGujaratiMarathiPunjabiOdiaEnglish)tehitaknmlbngumrpaorenr#   r$   	model_keyreturnc                 C   sD   | t v rt |  S | tv r| S td|  dtt tt    )z)Resolve model alias to actual model name.zUnknown model: z. Available: )MODEL_ALIASESGEMINI_MODELS
ValueErrorlistkeys)r9    r@   3/home/ubuntu/maya3_transcribe/src/backend/config.pyget_model_namef   s
   (rB   )__doc__ospathlibr   dotenvr   __file__parent_project_rootgetenvr   r   r   r   GEMINI_API_KEYSUPABASE_URLSUPABASE_ANON_KEYSUPABASE_ADMIN_KEYr<   r;   DEFAULT_SETTINGSLANGUAGE_MAPstrrB   r@   r@   r@   rA   <module>   s    







)