Corrects first_line input and adds boxplot

app.py

@@ -1,12 +1,18 @@
 import streamlit as st
 from story_gen import StoryGenerator
 import plotly.figure_factory as ff
-import plotly.express as px 
+import plotly.express as px
+
+gen = StoryGenerator()
 
 st.set_page_config(page_title='Storytelling ' +
                    u'\U0001F5BC', page_icon=u'\U0001F5BC', layout="wide")
-
-gen = StoryGenerator()
+if 'count' not in st.session_state or st.session_state.count == 6:
+	  st.session_state.count = 0
+	  st.session_state.chat_history_ids = None
+	  st.session_state.old_response = ''
+else:
+    st.session_state.count += 1
 container_mode = st.sidebar.container()
 container_param = st.sidebar.container()
 container_button = st.sidebar.container()
@@ -32,7 +38,7 @@ if mode == 'Create Statistics':
     elif reaction_weight_mode == "Random":
         reaction_weight = -1
     if container_button.button('Analyse'):
-        gen.get_stats(story_till_now="Hello, I'm a language model,",
+        gen.get_stats(story_till_now=story_till_now,
                       num_generation=num_generation, length=length, reaction_weight=reaction_weight, num_tests=num_tests)
         if len(gen.stories) > 0:
             for si, story in enumerate(gen.stories):
@@ -41,6 +47,8 @@ if mode == 'Create Statistics':
             data=gen.stats_df[gen.stats_df.sentence_no==3]
             fig = px.violin(data_frame=data, x="reaction_weight", y="num_reactions", hover_data=data.columns)
             st.plotly_chart(fig, use_container_width=True)
+            fig2 = px.box(data_frame=data, x="reaction_weight", y="num_reactions", hover_data=data.columns)
+            st.plotly_chart(fig2, use_container_width=True)
     else:
         st.markdown('### You selected statistics. Now set your parameters and click the `Analyse` button.')
 elif mode == 'Play Storytelling':
@@ -59,6 +67,6 @@ elif mode == 'Play Storytelling':
             story_till_now, num_generation, length)
         st.markdown(f'### Story')
         st.text(story_till_now)
-        st.markdown(f'The last sentence has an **Emotion** of {emotion["label"]} with a confidence score of {emotion["score"]}.')
+        st.markdown(f'The last sentence has the "{emotion["label"]}" **Emotion** with a confidence score of {emotion["score"]}.')
     else:
         st.markdown('### Write the first sentence and then hit the `Run` button')