Update radial_diagram_generator.py

radial_diagram_generator.py

@@ -4,7 +4,7 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges
 
-def generate_radial_diagram(json_input: str) -> str: # Removed base_color parameter
+def generate_radial_diagram(json_input: str, output_format: str) -> str:
     """
     Generates a radial (center-expanded) diagram from JSON input.
 
@@ -94,7 +94,6 @@ def generate_radial_diagram(json_input: str) -> str: # Removed base_color parame
         
         base_color = '#19191a' # Hardcoded base color
 
-        # Central node styling (rounded box, dark color)
         dot.node(
             'central',
             data['central_node'],
@@ -105,14 +104,11 @@ def generate_radial_diagram(json_input: str) -> str: # Removed base_color parame
             fontsize='16'           # Larger font for central node
         )
         
-        # Add child nodes and edges recursively starting from depth 1
-        # The add_nodes_and_edges function will handle styling consistent with other graphs
         add_nodes_and_edges(dot, 'central', data.get('nodes', []), current_depth=1, base_color=base_color)
 
-        # Save to temporary file
-        with NamedTemporaryFile(delete=False, suffix='.png') as tmp:
-            dot.render(tmp.name, format='png', cleanup=True)
-            return tmp.name + '.png'
+        with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp:
+            dot.render(tmp.name, format=output_format, cleanup=True)
+            return f"{tmp.name}.{output_format}"
 
     except json.JSONDecodeError:
         return "Error: Invalid JSON format"