Add code to render networks obtained from an html file in quarto reports

Author: sayalaruano

vuegen/quarto_reportview.py

@@ -280,18 +280,22 @@ def _generate_plot_content(self, plot, is_report_static, static_dir: str = STATI
                 else:
                     plot_content.append(f"""fig_altair\n```\n""")
             elif plot.plot_type == r.PlotType.INTERACTIVE_NETWORK:
-                G = plot.read_network()
-                num_nodes = G.number_of_nodes()
-                num_edges = G.number_of_edges()
-                plot_content.append(f'**Number of nodes:** {num_nodes}\n')
-                plot_content.append(f'**Number of edges:** {num_edges}\n')
-                
+                network_data = plot.read_network()
                 if is_report_static:
                     plot.save_netwrok_image(G, static_plot_path, "png")
                     plot_content.append(self._generate_image_content(static_plot_path))
                 else:
-                    # Get the Network object
-                    net = plot.create_and_save_pyvis_network(G, html_plot_file)
+                    if isinstance(network_data, str) and network_data.endswith('.html'):
+                        # If network_data is the path to an HTML file, just visualize it
+                        html_plot_file = network_data
+                    else:
+                        num_nodes = network_data.number_of_nodes()
+                        num_edges = network_data.number_of_edges()
+                        plot_content.append(f'**Number of nodes:** {num_nodes}\n')
+                        plot_content.append(f'**Number of edges:** {num_edges}\n')
+                        # Get the Network object
+                        net = plot.create_and_save_pyvis_network(network_data, html_plot_file)
+
                     plot_content.append(self._generate_plot_code(plot, html_plot_file))
             else:
                     self.report.logger.warning(f"Unsupported plot type: {plot.plot_type}")

vuegen/streamlit_reportview.py

@@ -313,21 +313,17 @@ def _generate_plot_content(self, plot, static_dir: str = STATIC_FILES_DIR) -> Li
                 plot_content.append(self._generate_plot_code(plot))
             elif plot.plot_type == r.PlotType.INTERACTIVE_NETWORK:
                 network_data = plot.read_network()
-                
                 if isinstance(network_data, str) and network_data.endswith('.html'):
-                    # If network_data is the path to an HTML file, just visualize the existing HTML file
+                    # If network_data is the path to an HTML file, just visualize it
                     html_plot_file = network_data
                     plot_content.append(f"""with open('{html_plot_file}', 'r') as f:
     html_data = f.read()""")
                 else:
                     # Otherwise, create and save a new pyvis network from the graph
                     html_plot_file = os.path.join(static_dir, f"{plot.title.replace(' ', '_')}.html")
                     net = plot.create_and_save_pyvis_network(network_data, html_plot_file)
-
-                    # Now you can use the existing or newly created HTML file
                     num_nodes = len(net.nodes)
                     num_edges = len(net.edges)
-                    
                     plot_content.append(f"""with open('{html_plot_file}', 'r') as f:
     html_data = f.read()
 st.markdown(f"<p style='text-align: center; color: black;'> <b>Number of nodes:</b> {num_nodes} </p>", unsafe_allow_html=True)