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):
+def plot_users_disks(topk_to_show=5, skip_drives=[""]):
     df = datasets.load_dataset(
         "pluslab/PLUS_Lab_GPUs_Data",
         data_files="users_disks.csv",
@@ -239,7 +239,8 @@ def plot_users_disks(topk_to_show=5):
         lambda x: x.replace(".cs.ucla.edu", "")
     )
     df_return = df.copy()
-    df = df[df["path"] != "home"]
+    for skip_drive in skip_drives:
+        df = df[df["path"] != skip_drive]
 
     # Ensure numeric
     df["size_GB"] = pd.to_numeric(df["size_GB"], errors="coerce").fillna(0)
@@ -332,10 +333,11 @@ def plot_figs():
     try:
         fig_gpus, df_gpus = plot_gpus()
         fig_disks, df_disks = plot_disks()
-        fig_users_disks, df_users_disks = plot_users_disks()
+        fig_users_disks, df_users_disks = plot_users_disks(skip_drives=["home"])
+        fig_users_disks_home, _ = plot_users_disks(skip_drives=["local1", "data2"], topk_to_show=3)
     except Exception as e:
         print(e)
-    return fig_gpus, fig_disks, fig_users_disks, df_gpus, df_disks, df_users_disks
+    return fig_gpus, fig_disks, fig_users_disks, fig_users_disks_home, df_gpus, df_disks, df_users_disks
 
 demo = gr.Interface(
     fn=plot_figs,
@@ -345,7 +347,8 @@ demo = gr.Interface(
     outputs = [
         gr.Plot(label="GPU Status", elem_classes="plotcss"),
         gr.Plot(label="Disk Status", elem_classes="plotcss"),
-        gr.Plot(label="Users Disk Status", elem_classes="plotcss"),
+        gr.Plot(label="Users Disk Status (Local1)", elem_classes="plotcss"),
+        gr.Plot(label="Users Disk Status (Home)", elem_classes="plotcss"),
         gr.Dataframe(label="GPU Status Details"),
         gr.Dataframe(label="Disk Status Details"),
         gr.Dataframe(label="Users Disk Status Details"),