From 99ba3e5536835b961dfccdff9dbebd2db09d0f5c Mon Sep 17 00:00:00 2001
From: David Grundfest <s233039@dtu.dk>
Date: Thu, 1 Aug 2024 16:00:24 +0200
Subject: [PATCH] Small fix in GUI
---
qim3d/gui/interface.py | 16 +++++-----------
qim3d/gui/local_thickness.py | 2 +-
2 files changed, 6 insertions(+), 12 deletions(-)
diff --git a/qim3d/gui/interface.py b/qim3d/gui/interface.py
index c1743bf4..78830582 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 a4c9b22b..0c76f1aa 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():
--
GitLab