Multichem commited on
Commit
0a818cb
·
verified ·
1 Parent(s): e6b4d5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -79,6 +79,12 @@ def init_FD_seed_frame():
79
 
80
  return FD_seed
81
 
 
 
 
 
 
 
82
  dk_raw, fd_raw = init_baselines()
83
 
84
  tab1, tab2 = st.tabs(['Data Export', 'Contest Sims'])
@@ -142,7 +148,7 @@ with tab1:
142
  if 'data_export' in st.session_state:
143
  st.download_button(
144
  label="Export optimals set",
145
- data=st.session_state.data_export.to_csv().encode('utf-8'),
146
  file_name='MLB_optimals_export.csv',
147
  mime='text/csv',
148
  )
@@ -169,7 +175,7 @@ with tab1:
169
  if 'data_export' in st.session_state:
170
  st.download_button(
171
  label="Export optimals set",
172
- data=st.session_state.data_export.to_csv().encode('utf-8'),
173
  file_name='MLB_optimals_export.csv',
174
  mime='text/csv',
175
  )
 
79
 
80
  return FD_seed
81
 
82
+ @st.cache_data
83
+ def convert_df_from_parquet(df):
84
+ # IMPORTANT: Cache the conversion to prevent computation on every rerun
85
+ working_file = df.read_parquet()
86
+ return working_file.to_csv().encode('utf-8')
87
+
88
  dk_raw, fd_raw = init_baselines()
89
 
90
  tab1, tab2 = st.tabs(['Data Export', 'Contest Sims'])
 
148
  if 'data_export' in st.session_state:
149
  st.download_button(
150
  label="Export optimals set",
151
+ data=convert_df_from_parquet(st.session_state.data_export),
152
  file_name='MLB_optimals_export.csv',
153
  mime='text/csv',
154
  )
 
175
  if 'data_export' in st.session_state:
176
  st.download_button(
177
  label="Export optimals set",
178
+ data=convert_df_from_parquet(st.session_state.data_export),
179
  file_name='MLB_optimals_export.csv',
180
  mime='text/csv',
181
  )