Update app.py

app.py

@@ -15,6 +15,18 @@ if uploaded_file is not None:
      date = st.text_input("Write the first date")
      freq = st.text_input("Write the frequency")
      horizons = int(st.text_input("Write the number of horizons"))
+     seasonality = match freq:
+                   case 'Y':
+                        10
+                   case 'Q':
+                        4
+                   case 'M':
+                        12
+                   case 'W':
+                        4
+                   case 'D':
+                        7
+     st.write(f'Seasonality: {seasonality}')              
      if series is not None:
           series_train = pd.DataFrame(
                                         {
@@ -29,12 +41,22 @@ if uploaded_file is not None:
           
           fcst = StatsForecast(
                                series_train, 
-                               models=[(auto_arima, 12), (seasonal_naive, 12)], 
+                               models=[(auto_arima, seasonality), (seasonal_naive, seasonality)], 
                                freq=freq, 
                                n_jobs=1
                               )
           forecasts = fcst.forecast(horizons, level=(80, 95))
           st.write(forecasts)
-          fig.add_scatter(x=forecasts['ds'], y=forecasts['auto_arima_season_length-12_mean'], mode='lines')
-          fig.add_scatter(x=forecasts['ds'], y=forecasts['seasonal_naive_season_length-12'], mode='lines')
+          fig.add_scatter(x=forecasts['ds'], y=forecasts[f'auto_arima_season_length-{seasonality}_mean'], mode='lines')
+          fig.update_traces(
+                             name=f'auto_arima_season_length-{seasonality}_mean',
+                             line=dict(color='blue', width=3),
+                             showlegend=True
+          )
+          fig.add_scatter(x=forecasts['ds'], y=forecasts[f'seasonal_naive_season_length-{seasonality}'], mode='lines')
+          fig.update_traces(
+                             name=f'seasonal_naive_season_length-{seasonality}',
+                             line=dict(color='blue', width=3),
+                             showlegend=True
+          )
           st.plotly_chart(fig)