add optino to show pvnet_gsp_sum

README.md

@@ -73,6 +73,7 @@ With any push to `main`, in order to deploy changes, the `Terraform Cloud` varia
 
 - DB_URL: The database url which will be queried for  forecasts
 - password: The password for accessing the code 
+- SHOW_PVNET_GSP_SUM: Option to show `pvnet_gsp_sum` model or not. This defaults to zero
 
 ## Contributors 
 

src/forecast.py

@@ -14,6 +14,10 @@
 from plots.utils import colour_per_model
 
 
+show_pvnet_gsp_sum = os.getenv("SHOW_PVNET_GSP_SUM", False)
+
+
+
 def forecast_page():
     """Main page for status"""
 
@@ -44,8 +48,11 @@ def forecast_page():
         if location.gsp_id == gsp_id
     ][0]
 
+    models = ["cnn", "National_xg", "pvnet_v2", "blend"]
+    if show_pvnet_gsp_sum:
+        models.append("pvnet_gsp_sum")
     forecast_models = st.sidebar.multiselect(
-        "Select a model", ["cnn", "National_xg", "pvnet_v2", "blend"], ["pvnet_v2"]
+        "Select a model", models, ["pvnet_v2"]
     )
 
     probabilistic_forecasts = [model for model in forecast_models if model in ["National_xg", "pvnet_v2", "blend"]]

src/main.py

@@ -37,6 +37,8 @@
     unsafe_allow_html=True,
 )
 
+show_pvnet_gsp_sum = os.getenv("SHOW_PVNET_GSP_SUM", False)
+
 
 def metric_page():
     # set up title and subheader
@@ -51,7 +53,10 @@ def metric_page():
     use_adjuster = st.sidebar.radio("Use adjuster", [True, False], index=1)
 
     st.sidebar.subheader("Select Forecast Model")
-    model_name = st.sidebar.selectbox("Select", ["cnn", "National_xg", "pvnet_v2"], "pvnet_v2")
+    models = ["cnn", "National_xg", "pvnet_v2"]
+    if show_pvnet_gsp_sum:
+        models.append("pvnet_gsp_sum")
+    model_name = st.sidebar.selectbox("Select", models, "pvnet_v2")
 
     # set up database connection
     url = os.environ["DB_URL"]