bug fix

app.py

@@ -149,12 +149,20 @@ def main():
     datepicker = st.date_input("Start of Forecast", datetime.date(2022, 10, 24), min_value=datetime.date(2022, 6, 26) + datetime.timedelta(days = 35), max_value=datetime.date(2023, 6, 26) - datetime.timedelta(days = 30), key = "date")
 
 
-    if st.button("Forecast Sales", type="primary"):
+    try:
+    # check if the key exists in session state
+    _ = st.session_state.pressed
+    except AttributeError:
+        # otherwise set it to false
+        st.session_state.pressed = False
+
+    if st.button("Forecast Sales", type="primary") or session.state.pressed:
         dataloader = prepare_dataset(parameters, df.copy(), st.session_state.rain, st.session_state.temperature, st.session_state.date, rain_mapping)
         preds = predict(model, dataloader, st.session_state.date)
 
         data_plot = adjust_data_for_plot(df.copy(), preds)
         fig, axs = generate_plot(data_plot)
+        st.session_state.pressed = True
         st.pyplot(fig)