'''Test EasyNMT opus-mt

'''

import streamlit as st
from easynmt import EasyNMT
model = EasyNMT('opus-mt')


# ---------- streamlit ----------
# When a user interacts with widgets in the app:
# Streamlit will rerun the code from top to bottom
# ---------- streamlit ----------
st.set_page_config(
    page_title='EasyNMT Testing',
    page_icon='📝',
    layout='wide',
)


with st.container():
    st.markdown('## 📑 Machine Translation')
    st.write('This is a testing of EasyNMT and opus-mt model with Streamlit.')

lang_list = model.get_languages()

b_size = st.slider('Translation quality (Beam size)', 1, 10, 5)

text = ''
submit = ''
target_langs = ''

with st.form(key='nmt'):
    text = st.text_area(
        label='Enter text',
        # placeholder='Enter a sentence. Pāḷi translation is not available now.', 1.0.0 does not have
        help='Auto detect input language 170+.')
    target_langs = st.multiselect(
        'Translate to (can select more than one language)',
        lang_list,
        ['en', 'vi'])
    submit = st.form_submit_button(label='Translate')

if submit and text:
    detected_lang = model.language_detection(text)

    st.write('Dectected input language: ' + detected_lang)

    if not target_langs:
        st.error('Please choose at least 1 target language.')
        st.stop()

    for lang in target_langs:
        try:
            res = model.translate(text, target_lang=lang, beam_size=b_size)
            if res:
                st.success(lang + ' => ' + res)

        except Exception as e:
            st.write(e)