Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import pandas as pd
+import datetime
 import plotly.express as px
 
 def split_multi_users(dfs):
@@ -37,7 +38,7 @@ def plot_now():
     fig.update_traces(textposition='inside', insidetextanchor='middle')
     return fig, dfs
 
-def plot_history():
+def plot_history(sample=True, sampling_interval_minutes=180):
     dfh = pd.read_pickle("hf://spaces/pluslab/PLUS_Lab_GPUs/history.pkl.gz", )
     dfh = dfh.fillna("FREE")
     dfh = split_multi_users(dfh)
@@ -45,6 +46,15 @@ def plot_history():
     dfh = dfh.groupby(["polling_timestamp", "username"]).sum()
     dfh = dfh.reset_index()
     dfh = dfh.sort_values(by=["polling_timestamp", "count"], ascending=False)
+    if sample:
+        unique_timestamps = dfh["polling_timestamp"].unique()
+        sampled_timestamps = [unique_timestamps[0]]
+        for i, t in enumerate(unique_timestamps[1:]):
+            diff = sampled_timestamps[-1] - t
+            print (sampled_timestamps[-1], t, diff)
+            if diff > datetime.timedelta(minutes=sampling_interval_minutes):
+                sampled_timestamps.append(t)
+        dfh = dfh[dfh["polling_timestamp"].isin(sampled_timestamps)]
     fig = px.area(dfh, x="polling_timestamp", y="count", color='username', color_discrete_map={"FREE": "black",}, markers=True, line_shape='spline',)
     return fig, dfh
 
@@ -61,7 +71,9 @@ def plot_figs():
 
 demo = gr.Interface(
     fn=plot_figs,
-    inputs = [],
+    inputs = [
+
+    ],
     outputs = [
         gr.Plot(label="GPU Status", elem_classes="plotcss"),
         gr.Dataframe(label="GPU Status Details"),