bug fix

app.py

@@ -70,6 +70,7 @@ 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)
@@ -141,37 +142,13 @@ 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")
 
     fig, axs = generate_plot(df.copy())
-    st.pyplot(fig)
+    st.pyplot(fig, key="plot")
 
     if st.button("Forecast Sales", type="primary"):
         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)
         update_plot(df, preds, axs, fig)
 
-    
-    #st.cache_data(allow_output_mutation=True)
-    def create_plot():
-        fig, ax = plt.subplots()
-        x_initial = np.linspace(0, 2*np.pi, 100)
-        y_initial = np.cos(x_initial)
-        ax.plot(x_initial, y_initial, label='Initial Line')
-        ax.set_xlabel('X-axis')
-        ax.set_ylabel('Y-axis')
-        ax.set_title('Plot')
-        ax.legend()
-        return fig
-    
-    fig = create_plot()
-    st.pyplot(fig)
-    
-    if st.button('Add Line'):
-        ax = fig.axes[0]
-        x_new = np.linspace(0, 2*np.pi, 100)
-        y_new = np.sin(x_new)
-        ax.plot(x_new, y_new, label='New Line')
-        ax.legend()
-        st.pyplot(fig)
-
 
 
 if __name__ == '__main__':