import streamlit as st

st.set_page_config(page_title="2023 FS Hackathon")
st.title("Founder's Studio AI Sandbox 🕹️")
expander = st.expander("Click here to close this intro", expanded=True)
expander.write(
    """
    This web app allows you to perform common Natural Language Processing tasks, select a task below to get started.
    These tasks are intended to help you validate your intuition and build a proof of concept for your idea.

    If a task you deem useful is not listed here, feel free to get in touch with Founder's Studio team at francesco.capuano@bain.com. 
    Happy hackathon!
    """
)

st.subheader(":point_down: Use the following drop-down menu to select a task!")

OPTION1="Chat with a file 💬📖"
OPTION2="Text summarization 🔎"
OPTION_N="Other 🤔"

option = st.radio(
    "Please select a task 🤖",
    options=[OPTION1, OPTION2, OPTION_N],
                      )

if option == OPTION1:
    from qa import qa_main 
    with st.container(): 
        qa_main()

elif option == OPTION2:
    from summarization import summarization_main
    with st.container():
        summarization_main()

elif option == OPTION_N:
    from mailing import mailing_main
    with st.container():
        user_suggestion = st.text_input(
            "What other task would you like to perform?", 
            placeholder="Transform meeting transcripts into rainbow-colored unicorns"
        )
        if user_suggestion:
            st.write("""
            Thanks for contributing with your suggestion! We are carefully reviewing every suggestion. 
            If you wish to further discuss your task suggestion, consider reaching out to francesco.capuano@bain.com.
            We will get back to you as soon as possible!
            """)

        mailing_main(
            subject="**NEW TASK SUGGESTION** - Automatic email.",
            body=f"User suggestion\n: {user_suggestion}",
            to_address="francesco.capuano@bain.com"
        )
        st.stop()