hysts's picture
hysts HF staff
Update
cb08619
#!/usr/bin/env python
from __future__ import annotations
import gradio as gr
from constants import (
HARDWARE_CHOICES,
OWNER_CHOICES,
SDK_CHOICES,
SLEEP_TIME_CHOICES,
STATUS_CHOICES,
VISIBILITY_CHOICES,
)
from demo_list import DemoList, get_df_from_yaml
demo_list = DemoList(get_df_from_yaml("list.yaml"))
def update_status_checkboxes(choices: list[str]) -> list[str]:
if "(ALL)" in choices:
return STATUS_CHOICES
elif "(NONE)" in choices:
return []
else:
return choices
def update_hardware_checkboxes(choices: list[str]) -> list[str]:
if "(ALL)" in choices:
return HARDWARE_CHOICES
elif "(NONE)" in choices:
return []
else:
return choices
def update_sdk_checkboxes(choices: list[str]) -> list[str]:
if "(ALL)" in choices:
return SDK_CHOICES
elif "(NONE)" in choices:
return []
else:
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=True):
status = gr.CheckboxGroup(
label="Status",
choices=["(ALL)", "(NONE)"] + STATUS_CHOICES,
value=STATUS_CHOICES,
type="value",
)
hardware = gr.CheckboxGroup(
label="Hardware",
choices=["(ALL)", "(NONE)"] + HARDWARE_CHOICES,
value=HARDWARE_CHOICES,
type="value",
)
sleep_time = gr.CheckboxGroup(
label="Sleep time",
choices=["(ALL)", "(NONE)"] + SLEEP_TIME_CHOICES,
value=SLEEP_TIME_CHOICES,
type="value",
)
multiple_replicas = gr.Checkbox(label="Multiple replicas", value=False)
sdk = gr.CheckboxGroup(
label="SDK",
choices=["(ALL)", "(NONE)"] + SDK_CHOICES,
value=SDK_CHOICES,
type="value",
)
visibility = gr.CheckboxGroup(
label="Visibility",
choices=VISIBILITY_CHOICES,
value=VISIBILITY_CHOICES,
type="value",
)
owner = gr.CheckboxGroup(
label="Owner",
choices=OWNER_CHOICES,
value=OWNER_CHOICES,
type="value",
)
apply_button = gr.Button("Apply")
df = gr.Dataframe(
value=demo_list.df_prettified,
datatype=demo_list.column_datatype,
type="pandas",
interactive=False,
row_count=(0, "dynamic"),
height=1000,
elem_id="table",
)
status.change(
fn=update_status_checkboxes,
inputs=status,
outputs=status,
queue=False,
show_progress=False,
api_name=False,
)
hardware.change(
fn=update_hardware_checkboxes,
inputs=hardware,
outputs=hardware,
queue=False,
show_progress=False,
api_name=False,
)
sdk.change(
fn=update_sdk_checkboxes,
inputs=sdk,
outputs=sdk,
queue=False,
show_progress=False,
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=demo_list.filter,
inputs=[
status,
hardware,
sleep_time,
multiple_replicas,
sdk,
visibility,
owner,
],
outputs=df,
api_name=False,
)
demo.queue(api_open=False).launch()