| import spaces | |
| import gradio as gr | |
| from hfsearch import HFSearchResult, search, update_filter, update_df | |
| with gr.Blocks(fill_width=True) as demo: | |
| with gr.Column(): | |
| search_result = gr.State(value=HFSearchResult()) | |
| with gr.Group(): | |
| with gr.Row(equal_height=True): | |
| infer_status = gr.Radio(label="Inference status", choices=["warm", "cold", "frozen", "all"], value="warm") | |
| gated_status = gr.Radio(label="Gated status", choices=["gated", "non-gated", "all"], value="non-gated") | |
| appr_status = gr.CheckboxGroup(label="Approval method", choices=["auto", "manual"], value=["auto", "manual"]) | |
| with gr.Accordion("Advanced", open=False): | |
| with gr.Row(equal_height=True): | |
| filter = gr.Textbox(label="Query", value="") | |
| author = gr.Textbox(label="Author", value="") | |
| sort = gr.Radio(label="Sort", choices=["last_modified", "likes", "downloads"], value="likes") | |
| sort_method = gr.Radio(label="Sort method", choices=["ascending order", "descending order"], value="ascending order") | |
| limit = gr.Number(label="Limit", info="If 0, fetches all models", value=1000, step=1, minimum=0, maximum=10000000) | |
| run_button = gr.Button("Search", variant="primary") | |
| with gr.Group(): | |
| with gr.Accordion("Filter", open=False): | |
| hide_item = gr.CheckboxGroup(label="Hide items", choices=[], value=[], visible=False) | |
| with gr.Row(equal_height=True): | |
| filter_item1 = gr.Dropdown(label="Filter item", choices=[""], value="", visible=False) | |
| filter1 = gr.Dropdown(label="Filter", choices=[""], value="", allow_custom_value=True, visible=False) | |
| filter_btn = gr.Button("Apply filter", variant="secondary", visible=False) | |
| result_df = gr.DataFrame(label="Results", type="array", value=[[]], interactive=False) | |
| run_button.click(search, [sort, sort_method, filter, author, infer_status, gated_status, appr_status, limit, search_result], [result_df, hide_item, search_result])\ | |
| .success(update_filter, [filter_item1, search_result], [filter_item1, filter1, filter_btn, search_result], queue=False) | |
| gr.on(triggers=[hide_item.change, filter_btn.click], fn=update_df, inputs=[hide_item, filter_item1, filter1, search_result], | |
| outputs=[result_df, search_result], trigger_mode="once", queue=False, show_api=False) | |
| filter_item1.change(update_filter, [filter_item1, search_result], [filter_item1, filter1, filter_btn, search_result], queue=False, show_api=False) | |
| demo.queue().launch() | |