bug fix

app.py

@@ -145,9 +145,33 @@ def main():
 
     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, datepicker)
+        preds = predict(model, dataloader, st.session_state.date)
         update_plot(df, preds, axs, fig)
 
+    def update_plot2(ax):
+    x = np.linspace(0, 2*np.pi, 100)
+    y = np.sin(x)
+    ax.plot(x, y, label='New Line')
+    ax.legend()
+    st.pyplot()
+
+    # Create an initial plot
+    figs, 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()
+    st.pyplot(figs)
+    
+    # Button to update the plot
+    if st.button('Add Line'):
+        update_plot2(ax)
+
+
+
 if __name__ == '__main__':
     main()