import streamlit as st
from src.constants import SETTINGS
from src.utils import load_models, print_results


def main():
    st.title(SETTINGS.title)
    col1, col2 = st.columns(2)

    with col1:
        st.header('Model Selection')
        model_selected = st.radio('Choose model', SETTINGS.models)

    with col2:
        st.header('Input text')
        input_text = col2.text_area(
            'Write or paste any text', 
            value=SETTINGS.sample_text,
            height=150
        )

    if st.button('Analyze', type='primary') and input_text != '':
        pipes = load_models()
        pipe = pipes[model_selected]

        with st.spinner('Analyzing text...'):
            output = pipe(input_text)
        
        st.subheader(f"{SETTINGS.task.replace('-', ' ').capitalize()} results")

        if isinstance(output, dict):
            output = [output]


        for i, result in enumerate(output, 1):
            print_results(result)


if __name__ == '__main__':
    main()