adds download button

src/lib.py

@@ -4,8 +4,30 @@ import streamlit as st
 
 from src import StoryGenerator
 
+import xlsxwriter
+import pandas as pd
+import io
+
+
+def create_dowload_button(data, sheet_name='AllData', label="Download data", file_name='data.xlsx'):
+
+    buffer = io.BytesIO()
+    with pd.ExcelWriter(buffer, engine='xlsxwriter') as writer:
+        # Write each dataframe to a different worksheet.
+        data.to_excel(writer, sheet_name=sheet_name)
+
+        # Close the Pandas Excel writer and output the Excel file to the buffer
+        writer.save()
+        st.download_button(
+            label=label,
+            data=buffer,
+            file_name=file_name,
+            mime='application/vnd.ms-excel',
+        )
 
 # @st.cache(allow_output_mutation=True)
+
+
 def initialise_storytelling(gen, container_guide, container_param, container_button):
     gen.initialise_models()
     choices_first_sentence = [
@@ -30,9 +52,9 @@ def initialise_storytelling(gen, container_guide, container_param, container_but
     first_emotion = gen.get_emotion(first_sentence)
 
     length = set_input(container_param,
-                                label='Length of the sentence',
-                                    min_value=1, max_value=100, value=10, step=1,
-              key_slider='length_slider', key_input='length_input',)
+                       label='Length of the sentence',
+                       min_value=1, max_value=100, value=10, step=1,
+                       key_slider='length_slider', key_input='length_input',)
     return first_sentence, first_emotion, length
 
 
@@ -65,4 +87,4 @@ def set_input(container_param,
         step=step,
         key=key_slider,
         on_change=slider2input)
-    return number_input
+    return number_input

src/read_logs.py

@@ -5,7 +5,7 @@ import plotly.express as px
 import streamlit as st
 import xlsxwriter
 from os import listdir
-from .lib import set_input
+from .lib import set_input, create_dowload_button
 from os.path import isfile, join, exists
 import printj
 
@@ -163,12 +163,12 @@ class LogAnalyser:
             elif table_mode == 'Table':
                 st.table(dfs)
                 # st.table(df_reaction_pattern.iloc[story_id-1])
+            create_dowload_button(dfs, sheet_name=f'story_{story_id}', file_name=f'data_story_{story_id}.xlsx')
             # print(dfs.render())
         if table_mode == 'Dataframe':
             st.dataframe(df_reaction_pattern)
         elif table_mode == 'Table':
             st.table(df_reaction_pattern)
-
     # @st.cache
     def dfstyle_color_text_col(self, s):
         result = ['background-color: white']*len(s)

src/test.py

@@ -27,7 +27,14 @@ for i in range(len(s)-1):
 
 g2p
 
+# %%
+# import plotly.express as px
+from plotly.offline import init_notebook_mode, iplot
+import numpy as np
+init_notebook_mode()
 
+x = np.linspace(0, 1)
+iplot([{'x': x, 'y': 1-np.exp(-x)}])
 # # def highlight_greaterthan(s,column):
 # #     is_max = pd.Series(data=False, index=s.index)
 # #     is_max[column] = s.loc[column] >= 1