bug fix

app.py

@@ -73,7 +73,7 @@ def update_plot(df, preds, axs):
 @st.cache_resource
 def generate_plot(df):
     fig, axs = plt.subplots(2, 2, figsize=(8, 6))
-    df[["sales", "pred"]] = df[["sales", "pred"]].replace(0.0, np.nan)
+    df["sales"] = df["sales"].replace(0.0, np.nan)
     # Plot scatter plots for each group
     axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='grey')
     
@@ -142,7 +142,7 @@ 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))
 
-    fig, axs = generate_plot(df)
+    fig, axs = generate_plot(df.copy())
 
     st.pyplot(fig)