a 3$g @sHddlZddlmZddlmZddlmZeZ e Z e 0eddddZ edZeZWdn1sz0Ye eZWdn1s0Yd d Ze je geeegd d d Zeje egeeegd ddZeje eegeeegd ddZWdn1s20Ye dS)N)HuggingfaceHubSearch) HfFileSystemzHub Dataset IDz$Search for dataset id on Huggingfacedataset)label placeholder search_typemainccs~t|iVt|trJtd|d|d}||}ttj||diVnttj|diVtt j |td d iVdS)Nz datasets/@z /**/*.parquet)choicesvalue)r ) filesystemr) revision_textbox isinstanceintfsglobparquet_file_dropdowngrDropdownoutput_dataframepq ParquetFileread_row_group to_pandas)rrevision parquet_file parquet_filesr-/Users/quentinlhoest/hf/parquet-viewer/app.py_show_input_previews  r)inputsoutputscCst|dddS)Nrrrrr)rrrrshow_input_from_dataset_searchsr$cCst||ddS)Nrr"r#)rrrrrshow_input_from_revision"sr%cCst|||dS)Nr"r#)rrrrrrshow_input_from_parquet_file&sr&)gradiorpyarrow.parquetparquetrgradio_huggingfacehub_searchrhuggingface_hubrrBlocksdemoColumndataset_searchTextboxr rr DataFramerrchanger$r%r&launchrrrrs.      & &   *