Update

app.py

@@ -46,15 +46,20 @@ SDK_CHOICES = [
     'streamlit',
     'docker',
 ]
+SLEEP_TIME_CHOICES = list(demo_list.TO_TIME_STR.values())
+SLEEP_TIME_STR_TO_INT = {v: k for k, v in demo_list.TO_TIME_STR.items()}
 OWNER_CHOICES = [WHOAMI, 'other organizations']
 
 
 def update_df(status: list[str], hardware: list[str], sdk: list[str],
-              owner: list[str], multiple_replicas: bool) -> pd.DataFrame:
+              sleep_time: list[str], owner: list[str],
+              multiple_replicas: bool) -> pd.DataFrame:
     df_raw = demo_list.df_raw
     df = demo_list.df
     df = df[(df_raw.status.isin(status)) & (df_raw.hardware.isin(hardware)) &
             (df_raw.sdk.isin(sdk))]
+    sleep_time_int = [SLEEP_TIME_STR_TO_INT[s] for s in sleep_time]
+    df = df[df_raw.sleep_time.isin(sleep_time_int)]
     if multiple_replicas:
         df = df[df_raw.replicas > 1]
     if set(owner) == set(OWNER_CHOICES):
@@ -93,6 +98,15 @@ def update_sdk_checkboxes(choices: list[str]) -> list[str]:
         return choices
 
 
+def update_sleep_time_checkboxes(choices: list[str]) -> list[str]:
+    if '(ALL)' in choices:
+        return SLEEP_TIME_CHOICES
+    elif '(NONE)' in choices:
+        return []
+    else:
+        return choices
+
+
 with gr.Blocks(css='style.css') as demo:
     with gr.Accordion(label='Filter', open=False):
         status = gr.CheckboxGroup(label='Status',
@@ -108,6 +122,11 @@ with gr.Blocks(css='style.css') as demo:
                                choices=['(ALL)', '(NONE)'] + SDK_CHOICES,
                                value=SDK_CHOICES,
                                type='value')
+        sleep_time = gr.CheckboxGroup(label='Sleep time',
+                                      choices=['(ALL)', '(NONE)'] +
+                                      SLEEP_TIME_CHOICES,
+                                      value=SLEEP_TIME_CHOICES,
+                                      type='value')
         owner = gr.CheckboxGroup(label='Owner',
                                  choices=OWNER_CHOICES,
                                  value=OWNER_CHOICES,
@@ -136,9 +155,21 @@ with gr.Blocks(css='style.css') as demo:
                queue=False,
                show_progress=False,
                api_name=False)
-    apply_button.click(
-        fn=update_df,
-        inputs=[status, hardware, sdk, owner, multiple_replicas],
-        outputs=df,
-        api_name=False)
+    sleep_time.change(fn=update_sleep_time_checkboxes,
+                      inputs=sleep_time,
+                      outputs=sleep_time,
+                      queue=False,
+                      show_progress=False,
+                      api_name=False)
+    apply_button.click(fn=update_df,
+                       inputs=[
+                           status,
+                           hardware,
+                           sdk,
+                           sleep_time,
+                           owner,
+                           multiple_replicas,
+                       ],
+                       outputs=df,
+                       api_name=False)
 demo.queue(api_open=False).launch()

demo_list.py

@@ -30,7 +30,7 @@ class DemoList:
     ]
 
     TO_TIME_STR = {
-        -1: '',
+        -1: 'null',
         300: '5 minutes',
         600: '10 minutes',
         900: '15 minutes',