Update concept_map_generator.py

concept_map_generator.py

@@ -4,7 +4,7 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges
 
-def generate_concept_map(json_input: str) -> str: # Removed base_color parameter
+def generate_concept_map(json_input: str, output_format: str) -> str:
     """
     Generates a concept map from JSON input.
 
@@ -217,10 +217,9 @@ def generate_concept_map(json_input: str) -> str: # Removed base_color parameter
         # 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"