Update synoptic_chart_generator.py

synoptic_chart_generator.py

@@ -4,7 +4,7 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges
 
-def generate_synoptic_chart(json_input: str) -> str: # Removed base_color parameter
+def generate_synoptic_chart(json_input: str, output_format: str) -> str:
     """
     Generates a synoptic chart (horizontal flowchart) from JSON input.
 
@@ -119,9 +119,8 @@ def generate_synoptic_chart(json_input: str) -> str: # Removed base_color parame
             }
         )
         
-        base_color = '#19191a' # Hardcoded base color
+        base_color = '#19191a' 
 
-        # Central node styling (rounded box, dark color)
         dot.node(
             'central',
             data['central_node'],
@@ -132,13 +131,11 @@ def generate_synoptic_chart(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
         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"