Update app.py

app.py

@@ -227,7 +227,7 @@ def plot_disks(alert_threshold_pct=99.0):
     return fig, df
 
 
-def plot_users_disks(topk_to_show=5, skip_drives=[""]):
+def plot_users_disks(topk_to_show=4, skip_drives=[""]):
     df = datasets.load_dataset(
         "pluslab/PLUS_Lab_GPUs_Data",
         data_files="users_disks.csv",
@@ -303,7 +303,18 @@ def plot_users_disks(topk_to_show=5, skip_drives=[""]):
     # -----------------------------------------------------
 
     # Text labels inside cells
-    text_values = heatmap_df.round(1).astype(str)
+    # text_values = heatmap_df.round(1).astype(str)
+
+    # Abbreviate large numbers (e.g., 14135 -> 14.1k)
+    def format_label(val):
+        if val >= 1000:
+            return f"{val / 1000:.1f}T"
+        elif val > 0:
+            return f"{val:.0f}"
+        return "0"
+
+    # Use .map() for newer pandas, or .applymap() if you are on an older version
+    text_values = heatmap_df.map(format_label)
 
     fig = go.Figure(
         data=go.Heatmap(
@@ -312,7 +323,7 @@ def plot_users_disks(topk_to_show=5, skip_drives=[""]):
             y=heatmap_df.index.tolist(),
             text=text_values.values,
             texttemplate="%{text}",
-            textfont={"size": 12},
+            textfont={"size": 11},
             hovertemplate="path=%{y}<br>user=%{x}<br>size_GB=%{z:.2f}<extra></extra>",
             colorscale="YlOrRd",
         )