fix: add nodes and node labels

Author: connorferster

src/pynite_plotly/rendering.py

@@ -42,8 +42,8 @@ def __init__(
             spring=3
         ),
         title: str = "Pynite - Simple Finite Element Analysis for Python",
-        height: int = 800,
-        width: int = 800,
+        window_height: int = 800,
+        window_width: int = 800,
     ):
         self.model = model
 
@@ -61,21 +61,16 @@ def __init__(
         self._scalar_bar_text_size = 24
         self.theme = "default"
         self.colors = colors
-        self.line_widths = dict(
-            member=4,
-            loads=2,
-            deformed_member=2,
-            spring=3
-        )
-
+        self.line_widths = line_widths
         self._title = "Pynite - Simple Finite Element Analysis for Python"
-
+        self._window_width = window_width
+        self._window_height = window_height
         self.plotter = None
 
 
         self._layout = default_layout(self._title)
-        self.window_width = width
-        self.window_height = height
+        self._layout.width = window_width
+        self._layout.height = window_height
 
         # self.plotter.set_background('white')  # Setting background color
         # # self.plotter.add_logo_widget('./Resources/Full Logo No Buffer - Transparent.png')
@@ -90,6 +85,7 @@ def __init__(
         # Initialize spring labels
         self._spring_label_points = []
         self._spring_labels = []
+        self._annotations = []
 
     ## Get Figure Size
 
@@ -225,7 +221,6 @@ def render_model(self, reset_camera=True):
         self.update(reset_camera)
 
         # Render the model (code execution will pause here until the user closes the window)
-        self.plotter.show()
         return self.plotter
 
     def screenshot(
@@ -289,8 +284,7 @@ def update(self, reset_camera=True):
         # Clear out the old plot (if any)
         self.plotter = go.Figure()
 
-        layout = default_layout(self._title)
-        self._layout = layout
+        layout = self._layout
 
         ## Set the layout
         self.plotter.layout = self._layout
@@ -1689,6 +1683,8 @@ def add_point_labels(
         point_size: int = 5,
         shape: str = None,
         render_points_as_spheres=False,
+        x_shift=0,
+        y_shift=0
     ):
         assert len(points) == len(labels)
         # if show_points or render_points_as_spheres or shape:
@@ -1703,17 +1699,14 @@ def add_point_labels(
             label = labels[idx]
             x, y, z = point
             text_label = f"{start_bold_tag}{label}{end_bold_tag}"
-            # annotations.append(
-            #     dict(
-            #         x=x, 
-            #         y=y, 
-            #         z=z, 
-            #         text=text_label,
-            #         font=dict(
-            #             color=self.colors['point_label_text']
-            #         ),
-            #     )
-            # )
+            if render_points_as_spheres:
+                ...
+                self.plotter.add_trace(
+                    go.Scatter3d(
+                        x=[x], y=[y], z=[z], marker=dict(color=point_color, size=self.annotation_size/30)
+                    )
+                )
+            # print(f"Annotation: {text_label=} | {point=}")
             annotations.append(
                 dict(
                     x=x, 
@@ -1722,15 +1715,15 @@ def add_point_labels(
                     text=text_label,
                     showarrow=False,
                     font=dict(
-                        color=self.colors['point_label_text'],
+                        color=text_color,
                         size=16
                     ),
-                    yshift=5,
-                    xshift=5
+                    yshift=y_shift,
+                    xshift=x_shift,
                 )
             )
-
-        self.plotter.update_layout(scene=dict(annotations=annotations))
+        self._annotations += annotations
+        self.plotter.update_layout(scene=dict(annotations=self._annotations))
         # TODO: Show points