diff --git a/src/viewer/pages/plot_sMRI_vars_study.py b/src/viewer/pages/plot_sMRI_vars_study.py index df34e6c..303546a 100644 --- a/src/viewer/pages/plot_sMRI_vars_study.py +++ b/src/viewer/pages/plot_sMRI_vars_study.py @@ -101,6 +101,9 @@ def panel_incsv() -> None: ) st.session_state.is_updated["csv_plot"] = False + # Show input data + st.dataframe(st.session_state.plot_var["df_data"]) + def panel_rename() -> None: """ diff --git a/src/viewer/pages/prep_sMRI_dicomtonifti.py b/src/viewer/pages/prep_sMRI_dicomtonifti.py index 6941225..7b7e2ae 100644 --- a/src/viewer/pages/prep_sMRI_dicomtonifti.py +++ b/src/viewer/pages/prep_sMRI_dicomtonifti.py @@ -169,6 +169,9 @@ def panel_detect() -> None: icon=":material/thumb_up:", ) + with st.expander('Show dicom metadata', expanded=False): + st.dataframe(st.session_state.df_dicoms) + @st.dialog("DICOM Series Detection") # type:ignore def help_detect_series(): st.markdown( @@ -237,6 +240,11 @@ def panel_extract() -> None: num_nifti = utilio.get_file_count( st.session_state.paths[st.session_state.sel_mod], ".nii.gz" ) + + df_files = utilio.get_file_names( + st.session_state.paths[st.session_state.sel_mod], ".nii.gz" + ) + num_nifti=df_files.shape[0] if num_nifti > 0: st.session_state.flags["dir_nifti"] = True st.session_state.flags[st.session_state.sel_mod] = True @@ -245,6 +253,9 @@ def panel_extract() -> None: icon=":material/thumb_up:", ) + with st.expander('View NIFTI image list'): + st.dataframe(df_files) + @st.dialog("DICOM to Nifti Conversion") # type:ignore def help_extract_nifti(): st.markdown( diff --git a/src/viewer/utils/utils_io.py b/src/viewer/utils/utils_io.py index 06cfa7c..0edfee4 100644 --- a/src/viewer/utils/utils_io.py +++ b/src/viewer/utils/utils_io.py @@ -4,6 +4,7 @@ import zipfile from tkinter import filedialog from typing import Any, BinaryIO, List, Optional +import pandas as pd # https://stackoverflow.com/questions/64719918/how-to-write-streamlit-uploadedfile-to-temporary-in_dir-with-original-filenam # https://gist.github.com/benlansdell/44000c264d1b373c77497c0ea73f0ef2 @@ -118,6 +119,19 @@ def get_file_count(folder_path: str, file_suff: str = "") -> int: count += 1 return count +def get_file_names(folder_path: str, file_suff: str = "") -> pd.DataFrame: + f_names = [] + if os.path.exists(folder_path): + if file_suff == "": + for root, dirs, files in os.walk(folder_path): + f_names.append(files) + else: + for root, dirs, files in os.walk(folder_path): + for file in files: + if file.endswith(file_suff): + f_names.append([file]) + df_out = pd.DataFrame(columns=['FileName'], data=f_names) + return df_out def get_file_list(folder_path: str, file_suff: str = "") -> List: list_nifti: List[str] = []