Spaces:
Running
Running
| #!/usr/bin/python3 | |
| # -*- coding: utf-8 -*- | |
| import gradio as gr | |
| from project_settings import project_path | |
| def get_fs_tab(): | |
| with gr.TabItem("fs"): | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| fs_filename = gr.Textbox(label="filename", max_lines=10) | |
| fs_file = gr.File(label="file") | |
| # fs_file_dir = gr.Textbox(value="data", label="file_dir") | |
| fs_file_dir = gr.Dropdown(choices=["data/dataset", "data/eval_data"], | |
| value="data/dataset", | |
| label="file_dir") | |
| fs_query = gr.Button("query", variant="primary") | |
| with gr.Column(scale=7): | |
| fs_filelist_dataset_state = gr.State(value=[]) | |
| fs_filelist_dataset = gr.Dataset( | |
| components=[fs_filename, fs_file], | |
| samples=fs_filelist_dataset_state.value, | |
| ) | |
| def when_click_query_files(file_dir: str = "data"): | |
| file_dir = project_path / file_dir | |
| dataset_state = list() | |
| for filename in file_dir.glob("**/*.*"): | |
| if filename.is_dir(): | |
| continue | |
| if filename.stem.startswith("."): | |
| continue | |
| if filename.name.endswith(".py"): | |
| continue | |
| if filename.name.endswith(".raw"): | |
| continue | |
| dataset_state.append(( | |
| filename.relative_to(file_dir).as_posix(), | |
| filename.as_posix(), | |
| )) | |
| dataset = gr.Dataset( | |
| components=[fs_filename, fs_file], | |
| samples=dataset_state, | |
| ) | |
| return dataset_state, dataset | |
| fs_filelist_dataset.click( | |
| fn=lambda x: ( | |
| x[1], x[1] | |
| ), | |
| inputs=[fs_filelist_dataset], | |
| outputs=[fs_filename, fs_file] | |
| ) | |
| fs_query.click( | |
| fn=when_click_query_files, | |
| inputs=[fs_file_dir], | |
| outputs=[fs_filelist_dataset_state, fs_filelist_dataset] | |
| ) | |
| return locals() | |
| if __name__ == "__main__": | |
| with gr.Blocks() as block: | |
| fs_components = get_fs_tab() | |
| block.launch() | |