adds plotely violin

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from story_gen import StoryGenerator
+import plotly.figure_factory as ff
+import plotly.express as px 
 
 st.set_page_config(page_title='Storytelling ' +
                    u'\U0001F5BC', page_icon=u'\U0001F5BC', layout="wide")
@@ -34,8 +36,12 @@ if mode == 'Create Statistics':
                       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):
-                st.markdown(f'Story no. {si}:', unsafe_allow_html=False)
+                st.markdown(f'### Story no. {si}:', unsafe_allow_html=False)
                 st.markdown(story, unsafe_allow_html=False)
+            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)
+
 
 elif mode == 'Create Statistics':
     container_mode.write('Let\'s play storytelling.')