fix: Improve module imports

Author: AlejandroFernandezLuces

src/ansys/tools/visualization_interface/backends/pyvista/pyvista.py

@@ -21,6 +21,7 @@
 # SOFTWARE.
 """Provides a wrapper to aid in plotting."""
 from abc import abstractmethod
+import importlib.util
 
 from beartype.typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 import pyvista as pv
@@ -52,6 +53,8 @@
 from ansys.tools.visualization_interface.utils.color import Color
 from ansys.tools.visualization_interface.utils.logger import logger
 
+_HAS_TRAME = importlib.util.find_spec("pyvista.trame") and importlib.util.find_spec("trame.app")
+
 if TYPE_CHECKING:
     import numpy as np
 
@@ -129,8 +132,6 @@ def __init__(
         self._origin_colors = {}
 
         # Enable the use of trame if requested and available
-        from ansys.tools.visualization_interface.backends.pyvista.trame_local import _HAS_TRAME
-
         if self._use_trame and _HAS_TRAME:
             # avoids GUI window popping up
             pv.OFF_SCREEN = True
@@ -462,13 +463,10 @@ def show_plotter(self, screenshot: Optional[str] = None) -> None:
             Path for saving a screenshot of the image that is being represented.
 
         """
-        from ansys.tools.visualization_interface.backends.pyvista.trame_local import (
-            _HAS_TRAME,
-            TrameVisualizer,
-        )
-
         if self._use_trame and _HAS_TRAME:
-
+            from ansys.tools.visualization_interface.backends.pyvista.trame_local import (
+                TrameVisualizer,
+            )
             visualizer = TrameVisualizer()
             visualizer.set_scene(self._pl)
             visualizer.show()

src/ansys/tools/visualization_interface/backends/pyvista/trame_local.py

@@ -20,6 +20,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 """Provides `trame <https://kitware.github.io/trame/index.html>`_ visualizer interface for visualization."""
+
 try:
     from pyvista.trame.ui import plotter_ui
     from trame.app import get_server