o
    6tiU                     @   s  d dl Z d dlmZmZmZmZmZmZmZm	Z	 e 
dddkr"dndZe 
dddkr.dndZe 
dddkr:dndZe 
d	Zi 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"d#d$d%d&d'd(d)d*d+Zd,d- Zd.d/ Zd0d1 ZdS )2    N)BASE_PROMPTBASE_PROMPT_ARJAIS_CHAT_ARJAIS_CHAT_ENREGION_COUNTRY_PROMPTREGION_COUNTRY_PROMPT_ARREGION_PROMPTREGION_PROMPT_ARCOUNTRYTTrueFREGIONARABIC
MODEL_NAMEEgyptu   مصرMoroccou   المغربAlgeriau   الجزائرLibyau
   ليبياSudanu   السودانTunisiau   تونسJordanu   الأردنLebanonu
   لبنانSyriau
   سوريا	Palestineu   فلسطينYemenu
   اليمنUAEu   الإماراتKSAu   السعوديةGulfu   الخليجLevantu
   الشامzNorth Africau   شمال أفريقياzNile Valleyu   وادي النيلc                 C   sN  | d sdn| d }| d sdn| d }| d   }tr"ts"J dtr,t| }t| }| d }d}tdD ]"}tr@|d | n|d	 | }|d
 |d |    d }||7 }q6trmtrmtratnt}	|	j||||d}
ntr~trst	nt
}	|	j|||d}
ntrtnt}	|	j||d}
trdtv rdtv rtrtj|
d}
|
S tj|
d}
|
S )Ncountry regionfirst_statementzTIf you want to add the country information, you must also add the region informationoptions   arabic_keysenglish_keysz. text
)r   r    r!   choices)r    r!   r(   )r!   r(   jaischat)question)stripr
   r   r   en_ar_countries_regionsranger   r   formatr	   r   r   r   r   r   r   )docr   r    r!   r(   choices_strikey
choice_str
cur_promptdoc_text r7   X/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/arab_culture/utils_mcq.pydoc_to_text1   sN   
r9   c                 C   s   t r| d d S | d d S )Nr"   r$   r%   )r   )r0   r7   r7   r8   doc_to_choicee   s   r:   c                 C   s(   t r| d d n| d d }| }|S )N
answer_keyarabic_answer_keyenglish_answer_key)r   r,   )r0   ansr7   r7   r8   doc_to_targeti   s   
r?   )os"lm_eval.tasks.arab_culture.promptsr   r   r   r   r   r   r   r	   getenvr
   r   r   r   r-   r9   r:   r?   r7   r7   r7   r8   <module>   sX    (
	
4