U wZcT@s0ddlmZddZddZddZdd Zd S)  literal_evalcCs|d}|dkrgSt|S)N languagesnoner)rowrrE/home/osanseviero/Desktop/workspace/repos/models-explorer/language.pymake_lang_listsr cCs t|dS)Nr)len)rrrrlanguage_count sr cCs|dkr||ddk}n2|dkr4||ddk}n|dkrL||ddk}d|j|jdkd f<d|j|jikd f<|d }|d d |d <|jtd d |d <|jtd d |d <||d dk}|d }||ik}t | }|jddd d |d<|||| fS)NNLPmodalitynlpAudioaudio Multimodal multimodalFalserr)axisr rcSstd|dkS)N multilingualr)int)xrrr(z"process_for_lang..r) locrisnasumfillnaapplyr r exploder unique)datar no_lang_countmodels_with_langslangs total_langsrrrprocess_for_lang s$   r'cCsL|dkr,|ddk}|ddk}|||BS|dkrD||ddkS|SdS)NzJust Multilingualrrr zThree or more languagesr)r" lingualitymultilingual_tagmultiple_lang_tagsrrrfilter_multilinguality,s   r,N)astrr r r'r,rrrrs