add stats & joy/sadness level

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
-# from emoji import get_emoji
 from emotions import get_emotion, get_sentiment_arc_evaluation
-from nltk.tokenize import sent_tokenize
+from stats import get_repetitions
+from nltk.tokenize import sent_tokenize, word_tokenize
 import nltk
 nltk.download('punkt')
 
@@ -9,14 +9,35 @@ 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)
+if text != 'Words and symbols are meant to be here':
+    # emotions
+    emotion_result = get_emotion(text)
+    st.write(f"Overall emotion of your story: {emotion_result[0]['label']}")
+    joy, sad = 0, 0
+    for emo in emotion_result:
+        if emo['label'] == 'joy':
+            if emo['score'] > joy:
+                joy = emo['score']
+        elif emo['label'] == 'love':
+            if emo['score'] > joy:
+                joy = emo['score']
+        elif emo['label'] == 'sadness':
+            sad = emo['score']
+    st.write(f"Joy level: {joy:.2f}%\nSadness level: {sad:.2f}%")
 
-st.write(f"Emotional arc of your story: {' - '.join(emo_arc)}")
-st.write(sentiment_arc_eval)
+    # emotional arcs
+    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)
+
+    # simple statistics
+    words = word_tokenize(text)
+    if len(words) < 5:
+        st.write("Isn't your story a bit too short?")
+    rep_ratio = get_repetitions(words)
+    st.write(f"Repetition ratio: {rep_ratio}")

stats.py

@@ -0,0 +1,4 @@
+def get_repetitions(words):
+    words = [w.lower() for w in words]
+    ratio = len(set(words))/len(words)
+    return ratio