import streamlit as st
# from emoji import get_emoji
from emotions import get_emotion, get_sentiment_arc_evaluation
from nltk.tokenize import sent_tokenize
nltk.download('punkt')

st.title("I don't even know what this is yet")

name = st.text_input('Who are you?', 'Right, who am I?')
text = st.text_area('Submit your stories', '''Words and symbols are meant to be here''')
emotion_result = get_emotion(text)
st.write(f"Overall emotion of your story: {emotion_result[0]['label']}")

sents = sent_tokenize(text)
emo_arc = []
for sent in sents:
    emo_arc.append(emotion_result[0]['label'])
sentiment_arc_eval = get_sentiment_arc_evaluation(emo_arc)

st.write(f"Emotional arc of your story: {' - '.join(emo_arc)}")
st.write(sentiment_arc_eval)