fixes weight slider and storytelling mode

app.py

@@ -21,14 +21,14 @@ num_generation = container_param.slider(
 length = container_param.slider(label='Length of the generated sentence',
                                 min_value=1, max_value=100, value=10, step=1)
 if mode == 'Create Statistics':
-    container_mode.write('You selected statistics.')
+    
     num_tests = container_param.slider(
         label='Number of tests', min_value=1, max_value=1000, value=3, step=1)
     reaction_weight_mode = container_param.select_slider(
         "Reaction Weight w:", ["Random", "Fixed"])
     if reaction_weight_mode == "Fixed":
         reaction_weight = container_param.slider(
-            label='reaction_weight w', min_value=0, max_value=1, value=-1, step=.001)
+            label='Reaction Weight w', min_value=0.0, max_value=1.0, value=0.5, step=0.01)
     elif reaction_weight_mode == "Random":
         reaction_weight = -1
     if container_button.button('Analyse'):
@@ -41,9 +41,9 @@ 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)
-
-
-elif mode == 'Create Statistics':
+    else:
+        st.markdown('### You selected statistics. Now set your parameters and click the `Analyse` button.')
+elif mode == 'Play Storytelling':
     container_mode.write('Let\'s play storytelling.')
 
     # # , placeholder="Start writing your story...")
@@ -57,8 +57,8 @@ elif mode == 'Create Statistics':
     if container_button.button('Run'):
         story_till_now, emotion = gen.story(
             story_till_now, num_generation, length)
-        st.write('Story:')
+        st.markdown(f'### Story')
         st.text(story_till_now)
-        st.text(f'Emotion: {emotion}')
+        st.markdown(f'The last sentence has an **Emotion** of {emotion["label"]} with a confidence score of {emotion["score"]}.')
     else:
-        st.write('Write the first sentence and then hit the Run button')
+        st.markdown('### Write the first sentence and then hit the `Run` button')

story_gen.py

@@ -57,7 +57,7 @@ class StoryGenerator:
         for i in range(num_generation):
             last_length = len(story_till_now)
             genreate_robot_sentence = self.generator(story_till_now, max_length=self.get_num_token(story_till_now) +
-                                                     length*i, num_return_sequences=1)
+                                                     length, num_return_sequences=1)
             story_till_now = genreate_robot_sentence[0]['generated_text']
             new_sentence = story_till_now[last_length:]
             emotions = self.classifier(new_sentence)

story_gen_test.py

@@ -29,3 +29,6 @@ ax = sns.violinplot(x="reaction_weight", y="num_reactions", data=data).set_title
 
 gen.stats_df[gen.stats_df.sentence_no==3]
 # %%
+import re
+len(re.findall(r'\w+', 'line ive '))
+# %%