Create radial_diagram_generator.py

radial_diagram_generator.py

@@ -0,0 +1,62 @@
+import graphviz
+import json
+from tempfile import NamedTemporaryFile
+import os
+from graph_generator_utils import add_nodes_and_edges
+
+def generate_radial_diagram(json_input: str) -> str:
+    """
+    Generates a radial (center-expanded) diagram 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='RadialDiagram',
+            format='png',
+            engine='neato', # Use 'neato' or 'fdp' for radial/force-directed layout
+            graph_attr={
+                'overlap': 'false', # Prevent node overlap
+                'splines': 'true',  # Smooth splines for edges
+                'bgcolor': 'white', # White background
+                'pad': '0.5',       # Padding around the graph
+                'layout': 'neato'   # Explicitly set layout engine for consistency
+            },
+            node_attr={
+                'fixedsize': 'false' # Allow nodes to resize based on content
+            }
+        )
+        
+        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
+        # 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'
+
+    except json.JSONDecodeError:
+        return "Error: Invalid JSON format"
+    except Exception as e:
+        return f"Error: {str(e)}"
+