Create concept_map_generator.py

concept_map_generator.py

@@ -0,0 +1,56 @@
+import graphviz
+import json
+from tempfile import NamedTemporaryFile
+import os
+from graph_generator_utils import add_nodes_and_edges
+
+def generate_concept_map(json_input: str) -> str:
+    """
+    Generates a concept map from JSON input.
+    """
+    try:
+        if not json_input.strip():
+            return "Error: Empty input"
+            
+        data = json.loads(json_input)
+        
+        if 'central_node' not in data or 'nodes' not in data:
+            raise ValueError("Missing required fields: central_node or nodes")
+
+        dot = graphviz.Digraph(
+            name='ConceptMap',
+            format='png',
+            graph_attr={
+                'rankdir': 'TB',        # Top-to-Bottom layout (vertical hierarchy)
+                'splines': 'ortho',     # Straight lines
+                'bgcolor': 'white',     # White background
+                'pad': '0.5'            # Padding around the graph
+            }
+        )
+        
+        base_color = '#19191a' # Base color for the central node and gradient
+
+        # Central node styling (rounded box, dark color)
+        dot.node(
+            'central',
+            data['central_node'],
+            shape='box',            # Rectangular shape
+            style='filled,rounded', # Filled and rounded corners
+            fillcolor=base_color,   # Darkest color
+            fontcolor='white',      # White text for dark background
+            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'
+
+    except json.JSONDecodeError:
+        return "Error: Invalid JSON format"
+    except Exception as e:
+        return f"Error: {str(e)}"
+