diff --git a/qim3d/gui/interface.py b/qim3d/gui/interface.py index c1743bf4d8885f3d34637b14daa8793689188946..78830582e052d4a72f303530a48bfc522bca88aa 100644 --- a/qim3d/gui/interface.py +++ b/qim3d/gui/interface.py @@ -1,6 +1,6 @@ from pathlib import Path from abc import abstractmethod, ABC -from os import path +from os import path, listdir import gradio as gr @@ -113,13 +113,7 @@ class InterfaceWithExamples(BaseInterface): self._set_examples_list() def _set_examples_list(self): - examples = [ - "fly_150x256x256.tif", - "cement_128x128x128.tif", - "NT_128x128x128.tif", - "shell_225x128x128.tif", - "bone_128x128x128.tif", - ] - self.img_examples = [] - for example in examples: - self.img_examples.append([path.join(self.qim_dir, "examples", example)]) + valid_sufixes = (".tif", ".tiff", ".h5", ".nii", ".gz", ".dcm", ".DCM", ".vol", ".vgi", ".txrm", ".txm", ".xrm") + examples_folder = path.join(self.qim_dir, 'examples') + self.img_examples = [path.join(examples_folder, example) for example in listdir(examples_folder) if example.endswith(valid_sufixes)] + diff --git a/qim3d/gui/local_thickness.py b/qim3d/gui/local_thickness.py index a4c9b22be35360b61fdae32830b9b6d4cebe93f3..0c76f1aad4181b5e23d5e2e054009075778f71e4 100644 --- a/qim3d/gui/local_thickness.py +++ b/qim3d/gui/local_thickness.py @@ -85,7 +85,7 @@ class Interface(InterfaceWithExamples): def define_interface(self): gr.Markdown( - "Interface for _Fast local thickness in 3D and 2D_ (https://github.com/vedranaa/local-thickness)" + "Interface for _Fast local thickness in 3D_ (https://github.com/vedranaa/local-thickness)" ) with gr.Row():