Added refresh button

app.py

@@ -1,10 +1,16 @@
 import streamlit as st
 
-from components import buildings_view, models_view, performance_view, computation_view
+from components import (
+    buildings_view,
+    models_view,
+    performance_view,
+    computation_view,
+    logos,
+    model_selector,
+    header,
+)
 import utils
 
-st.set_page_config(page_title="Pv Generation Dashboard", layout="wide")
-
 PAGES = [
     "Buildings",
     "Models",
@@ -13,6 +19,9 @@ PAGES = [
 ]
 
 
+st.set_page_config(page_title="Pv Generation Dashboard", layout="wide")
+
+
 @st.cache_data(ttl=86400)
 def fetch_data():
     return utils.get_wandb_data(
@@ -23,52 +32,27 @@ def fetch_data():
     )
 
 
+# Load data
 data = fetch_data()
-models = sorted(data["model"].unique().tolist())
-models_to_plot = set()
-model_groups: dict[str, list[str]] = {}
 
-for model in models:
-    group, model_name = model.split(".", maxsplit=1)
-    if group not in model_groups:
-        model_groups[group] = []
-    model_groups[group].append(model_name)
+# Extract models
+models = sorted(data["model"].unique().tolist())
 
 
 with st.sidebar:
-    left, right = st.columns(
-        2
-    )  # Create two columns within the right column for side-by-side images
-    with left:
-        st.image("./images/ku_leuven_logo.png")
-    with right:
-        st.image("./images/energyville_logo.png")
-
+    logos()
     view = st.selectbox("View", PAGES, index=0)
-
-    st.header("Models to include")
-    left, right = st.columns(2)
-    with left:
-        select_none = st.button("Select None", use_container_width=True)
-        if select_none:
-            for model in models:
-                st.session_state[model] = False
-    with right:
-        select_all = st.button("Select All", use_container_width=True)
-        if select_all:
-            for model in models:
-                st.session_state[model] = True
-
-    for model_group, models in model_groups.items():
-        st.text(model_group)
-        for model_name in models:
-            to_plot = st.checkbox(model_name, value=True, key=f"{model_group}.{model_name}")
-            if to_plot:
-                models_to_plot.add(f"{model_group}.{model_name}")
+    models_to_plot = model_selector(models)
+    st.subheader("Refresh data")
+    refresh = st.button(
+        "Refresh", use_container_width=True, help="Fetch the latest data from W&B"
+    )
+    if refresh:
+        fetch_data.clear()
+        st.rerun()
 
 
-st.title("EnFoBench - PV Generation")
-st.divider()
+header()
 
 if view == "Buildings":
     buildings_view(data)

components.py

@@ -3,6 +3,54 @@ import streamlit as st
 import plotly.express as px
 
 
+def header() -> None:
+    st.title("EnFoBench - PV Generation")
+    st.divider()
+
+
+def logos() -> None:
+    left, right = st.columns(2)
+    with left:
+        st.image("./images/ku_leuven_logo.png")
+    with right:
+        st.image("./images/energyville_logo.png")
+
+
+def model_selector(models: list[str]) -> set[str]:
+    # Group models by their prefix
+    model_groups: dict[str, list[str]] = {}
+    for model in models:
+        group, model_name = model.split(".", maxsplit=1)
+        if group not in model_groups:
+            model_groups[group] = []
+        model_groups[group].append(model_name)
+
+    models_to_plot = set()
+
+    st.header("Models to include")
+    left, right = st.columns(2)
+    with left:
+        select_none = st.button("Select None", use_container_width=True)
+        if select_none:
+            for model in models:
+                st.session_state[model] = False
+    with right:
+        select_all = st.button("Select All", use_container_width=True)
+        if select_all:
+            for model in models:
+                st.session_state[model] = True
+
+    for model_group, models in model_groups.items():
+        st.text(model_group)
+        for model_name in models:
+            to_plot = st.checkbox(
+                model_name, value=True, key=f"{model_group}.{model_name}"
+            )
+            if to_plot:
+                models_to_plot.add(f"{model_group}.{model_name}")
+    return models_to_plot
+
+
 def buildings_view(data):
     buildings = (
         data[
@@ -39,14 +87,14 @@ def buildings_view(data):
                 help="Available training data during the first prediction.",
                 format="%f",
                 min_value=0,
-                max_value=float(buildings['Available history (days)'].max()),
+                max_value=float(buildings["Available history (days)"].max()),
             ),
             "Capacity (kW)": st.column_config.ProgressColumn(
                 "Capacity (kW)",
                 help="Available training data during the first prediction.",
                 format="%f",
                 min_value=0,
-                max_value=float(buildings['Capacity (kW)'].max()),
+                max_value=float(buildings["Capacity (kW)"].max()),
             ),
         },
     )