add sum to the right, make the heatmap larger

app.py

@@ -4,6 +4,11 @@ import numpy as np
 import base64, pickle, textwrap
 import plotly.express as px
 
+# ---------------------------------------------------
+# Page configuration for wide layout
+# ---------------------------------------------------
+st.set_page_config(page_title="Faculty Heatmap Explorer", layout="wide")
+
 # ---------------------------------------------------
 # Decode embedded data
 # ---------------------------------------------------
@@ -17,7 +22,6 @@ df = pickle.loads(base64.b64decode(df_encoded))
 # Helper: wrap long labels
 # ---------------------------------------------------
 def wrap_labels(labels, width=40):
-    import textwrap
     return ["<br>".join(textwrap.wrap(lbl, width)) for lbl in labels]
 
 # ---------------------------------------------------
@@ -28,7 +32,7 @@ st.sidebar.title("Filters")
 def_opts = sorted(df['Dept Track'].dropna().astype(str).unique())
 def_stand = sorted(df['Standing'].dropna().astype(str).unique())
 
-# Clear filters
+# Clear filters button resets session state
 if st.sidebar.button("Clear All Filters"):
     st.session_state.pop('standing', None)
     st.session_state.pop('dept', None)
@@ -48,13 +52,14 @@ dept_sel = st.sidebar.multiselect(
     key='dept'
 )
 
-# Filter names
+# Filter names based on selected Standing & Dept
 mask = (
-    df['Standing'].astype(str).isin(st.session_state.standing) &
-    df['Dept Track'].astype(str).isin(st.session_state.dept)
+    df['Standing'].astype(str).isin(standing_sel) &
+    df['Dept Track'].astype(str).isin(dept_sel)
 )
 name_opts = sorted(df.loc[mask, 'Name'].astype(str).unique())
 
+# Initialize or prune stored names
 if 'names' not in st.session_state or st.session_state.names is None:
     st.session_state.names = name_opts.copy()
 else:
@@ -72,28 +77,35 @@ name_sel = st.sidebar.multiselect(
 # ---------------------------------------------------
 st.title("Faculty Heatmap Explorer")
 
-# Heatmap
+# Display heatmap (with optional row sums)
 if not name_sel:
     st.warning("No faculty selected — please choose at least one.")
     fig = px.imshow(
-        [[0]], labels={'x':'','y':'','color':'value'},
-        text_auto='.2f', title="No faculty selected"
+        [[0]],
+        labels={'x':'','y':'','color':'value'},
+        text_auto='.2f',
+        title="No faculty selected"
     )
     st.plotly_chart(fig, use_container_width=True)
 else:
-    # Sum & average
+    # Combine matrices
     sum_df = None
     for name in name_sel:
         mat = filled_matrices[name]
         sum_df = mat if sum_df is None else sum_df.add(mat, fill_value=0)
     avg_df = sum_df.div(len(name_sel))
 
-    # Wrap labels
-    wrapped_y = wrap_labels(list(avg_df.index))
-    wrapped_x = list(avg_df.columns)
+    # Add a sum column on the right
+    avg_df_ext = avg_df.copy()
+    avg_df_ext['Sum'] = avg_df_ext.sum(axis=1)
+
+    # Wrap y-axis labels for compact display
+    wrapped_y = wrap_labels(list(avg_df_ext.index))
+    wrapped_x = list(avg_df_ext.columns)
 
+    # Create heatmap
     fig = px.imshow(
-        avg_df.values,
+        avg_df_ext.values,
         x=wrapped_x,
         y=wrapped_y,
         labels={'color':'Avg value'},
@@ -101,6 +113,7 @@ else:
         title=f"Avg Heatmap for {len(name_sel)} Faculty"
     )
     fig.update_yaxes(autorange='reversed')
+    # Set a taller figure height for wide layout
+    fig.update_layout(height=800, margin=dict(l=70, r=70, t=70, b=70))
     st.plotly_chart(fig, use_container_width=True)
 
-# End of app