bug fix

app.py

@@ -70,7 +70,6 @@ def update_plot(df, preds, axs, fig):
     axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     return fig, axs
 
-@st.cache_resource
 def generate_plot(df):
     fig, axs = plt.subplots(2, 2, figsize=(8, 6))
     df["sales"] = df["sales"].replace(0.0, np.nan)
@@ -135,18 +134,17 @@ def main():
     
         ### Experiments
     """)
-    rain = st.radio("Rain Indicator", ('Default', 'Yes', 'No'))
+    rain = st.radio("Rain Indicator", ('Default', 'Yes', 'No'), key = "rain")
     
-    temperature = st.slider('Change in Temperature', min_value=-10.0, max_value=10.0, value=0.0, step=0.25)
+    temperature = st.slider('Change in Temperature', min_value=-10.0, max_value=10.0, value=0.0, step=0.25, key = "temperature")
     
-    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))
+    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")
 
     fig, axs = generate_plot(df.copy())
-
     st.pyplot(fig)
 
     if st.button("Forecast Sales", type="primary"):
-        dataloader = prepare_dataset(parameters, df.copy(), rain, temperature, datepicker, rain_mapping)
+        dataloader = prepare_dataset(parameters, df.copy(), st.session_state.rain, st.session_state.temperature, st.session_state.date, rain_mapping)
         preds = predict(model, dataloader, datepicker)
         update_plot(df, preds, axs, fig)