Add dark mode support to 3D scatter view

Introduced a 'dark_mode' parameter to the 3D scatter view animation, passing it from the callback and applying the appropriate Plotly template based on the mode. This enhances visualization by supporting both dark and light themes.

Author: rookiepeng

view_callbacks/scatter_3d_view.py

@@ -753,7 +753,7 @@ def export_3d_scatter_animation(
         """
         if btn == 0:
             raise PreventUpdate
-    
+
         set_progress(["show"])
 
         if not os.path.exists("data/" + case + "/images"):
@@ -822,6 +822,7 @@ def export_3d_scatter_animation(
                 frame_key=config["slider"],
                 img_list=img_list,
                 colormap=colormap,
+                dark_mode=darkmode,
                 **fig_kwargs
             )
         )

viz/viz.py

@@ -52,6 +52,8 @@
 import numpy as np
 import pandas as pd
 
+import plotly.io as pio
+
 from .graph_data import get_scatter3d_data, get_ref_scatter3d_data
 from .graph_layout import get_scatter3d_layout
 
@@ -323,6 +325,7 @@ def get_animation_data(
     img_list: Optional[List[str]] = None,
     colormap: Optional[str] = None,
     decay: int = 0,
+    dark_mode: bool = True,
     **kwargs: Any,
 ) -> Dict[str, Any]:
     """
@@ -478,6 +481,12 @@ def create_frame_data(frame_idx: int, current_idx: int) -> Dict[str, Any]:
         layout_kwargs["image"] = process_image(img_list[0])
 
     figure_layout = get_scatter3d_layout(**layout_kwargs)
+
+    if dark_mode:
+        figure_layout["template"] = pio.templates["plotly_dark"]
+    else:
+        figure_layout["template"] = pio.templates["plotly"]
+
     figure_layout.update(
         {
             "updatemenus": [