Update app.py

app.py

@@ -24,6 +24,8 @@ with st.sidebar:
     start_date = st.date_input('Start Date', dt.date(2020, 1, 1))
     end_date = st.date_input('End Date', dt.date.today())
     num_of_epochs = st.number_input('Insert a number', step=1, min_value=2, max_value=100, value=10)
+    n_futures = st.number_input('Length of future (to forecast)', step=1, min_value=10, max_value=300, value=30)
+    n_samples = st.number_input('Number of Simulations (to synthesize)', step=10, min_value=10, max_value=10000, value=50)
     st.markdown(
         f"""
         For webpage, please click [here](https://huggingface.co/spaces/eagle0504/stock-forecast).
@@ -160,4 +162,9 @@ if st.button('Train Models'):
         st.success("See the next point forecasts below:")
         st.write(f"Mean is: {np.mean(next_point_forecasts)}")
         st.write(f"Median is: {np.median(next_point_forecasts)}")
-        st.write(f"Std is: {np.std(next_point_forecasts)}")
+        st.write(f"Std is: {np.std(next_point_forecasts)}")
+
+        # Create MC Simulation
+        mean_return = np.mean(next_point_forecasts)
+        std_dev = np.std(next_point_forecasts)
+        plot_monte_carlo_forecasts(data=data, n_futures=n_futures, n_samples=n_samples, mean_return, std_dev)