Feat: Can save diagram as html

Author: milankures

cesnet_tszoo/datasets/cesnet_dataset.py

@@ -1107,10 +1107,20 @@ def summary(self, display_type: Literal["text", "diagram"]) -> None:
             print(self.dataset_config)
         elif display_type == DisplayType.DIAGRAM:
             steps = self.dataset_config._get_summary_steps()
-            css_utils.display_summary_diagram(steps)
+            return css_utils.display_summary_diagram(steps)
         else:
             raise NotImplementedError()
 
+    def save_summary_diagram_as_html(self, path: str):
+        if self.dataset_config is None or not self.dataset_config.is_initialized:
+            raise ValueError("Dataset is not initialized. Please call set_dataset_config_and_initialize() before attempting to save summary diagram.")
+
+        steps = self.dataset_config._get_summary_steps()
+        html = css_utils.get_summary_diagram(steps)
+
+        with open(path, "w", encoding="utf-8") as f:
+            f.write(html)
+
     def get_feature_names(self) -> list[str]:
         """Returns a list of all available feature names in the dataset. """
 

cesnet_tszoo/utils/css_styles/utils.py

@@ -75,7 +75,7 @@ def get_css_body(self) -> str:
         return body
 
 
-def display_summary_diagram(steps: list[SummaryDiagramStep]) -> None:
+def get_summary_diagram(steps: list[SummaryDiagramStep]) -> str:
     styles = __get_css_styles("summary_diagram")
 
     fallback_msg = (
@@ -96,4 +96,11 @@ def display_summary_diagram(steps: list[SummaryDiagramStep]) -> None:
     html.append("</div>")
     html = "".join(html)
 
+    return html
+
+
+def display_summary_diagram(steps: list[SummaryDiagramStep]) -> None:
+
+    html = get_summary_diagram(steps)
+
     display(HTML(html))