Update app.py

add Predictive Traffic Map

app.py

@@ -5,6 +5,7 @@ from datasets import load_dataset
 import os
 from src.basic_plot import basic_chart
 from src.map_viz import calling_map_viz
+from src.map_viz_pred import calling_pred_map_viz
 from src.data_ingestion import daily_average
 from src.heatmap import HeatMap
 from src.data_ingestion import remove_previous_view, merge_volumes
@@ -91,6 +92,9 @@ def main():
             st.plotly_chart(heatmap.update_day_bar_chart(day_choice))
 
     else: 
+
+        st.sidebar.markdown("Select Plots to show")
+        checkbox_two_pred = st.sidebar.checkbox('Predictive Traffic Map', value = True)
         
         figs = gen_fig()
         today = get_today()
@@ -114,5 +118,9 @@ def main():
         starter_variables = [x_train, str(d), pred_hour_choice, pred_view_choice]
         st.plotly_chart(predicted_figure(clf, starter_variables, figs))
 
+        if checkbox_two_pred:
+            st.subheader("Predictive Traffic Map")
+            st.pyplot(calling_pred_map_viz(counts_df))
+
 if __name__ == "__main__":
     main()