Update app.py

app.py

@@ -7,6 +7,10 @@ from statsforecast import StatsForecast
 from statsforecast.models import seasonal_naive, auto_arima
 from utils import calc_seasonality
 
+@st.cache
+def convert_df(df):
+    return df.to_csv().encode('utf-8')
+
 st.title('Forecasting time series')
 
 uploaded_file = st.file_uploader("Choose a CSV file", accept_multiple_files=False)
@@ -48,6 +52,15 @@ if uploaded_file is not None:
             
           st.write(forecasts)
           
+          csv = convert_df(forecasts)
+
+          st.download_button(
+             label="Download data as CSV",
+             data=csv,
+             file_name='forecast.csv',
+             mime='text/csv',
+          )
+                  
           fig_forecast = go.Figure([
                        go.Scatter(
                            name=series,