Update radial_diagram_generator.py

radial_diagram_generator.py

@@ -4,68 +4,57 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges
 
-def generate_radial_diagram(json_input: str) -> str:
+def generate_radial_diagram(json_input: str, base_color: str) -> str:
     """
     Generates a radial (center-expanded) diagram from JSON input.
 
     Args:
         json_input (str): A JSON string describing the radial diagram structure.
                           It must follow the Expected JSON Format Example below.
+        base_color (str): The hexadecimal color string (e.g., '#19191a') for the base
+                          color of the nodes, from which a gradient will be generated.
+
+    Returns:
+        str: The filepath to the generated PNG image file.
 
     Expected JSON Format Example:
     {
-      "central_node": "AI Core Concepts & Domains",
+      "central_node": "Making Coffee",
       "nodes": [
         {
-          "id": "foundational_ml",
-          "label": "Foundational ML",
-          "relationship": "builds on",
-          "subnodes": [
-            {"id": "supervised_l", "label": "Supervised Learning", "relationship": "e.g."},
-            {"id": "unsupervised_l", "label": "Unsupervised Learning", "relationship": "e.g."}
-          ]
-        },
-        {
-          "id": "dl_architectures",
-          "label": "Deep Learning Arch.",
-          "relationship": "evolved from",
-          "subnodes": [
-            {"id": "cnns_rad", "label": "CNNs", "relationship": "e.g."},
-            {"id": "rnns_rad", "label": "RNNs", "relationship": "e.g."}
-          ]
-        },
-        {
-          "id": "major_applications",
-          "label": "Major AI Applications",
-          "relationship": "applied in",
-          "subnodes": [
-            {"id": "nlp_rad", "label": "Natural Language Processing", "relationship": "e.g."},
-            {"id": "cv_rad", "label": "Computer Vision", "relationship": "e.g."}
-          ]
-        },
-        {
-          "id": "ethical_concerns",
-          "label": "Ethical AI Concerns",
-          "relationship": "addresses",
+          "id": "coffee_components",
+          "label": "Coffee Components",
+          "relationship": "Involves",
           "subnodes": [
-            {"id": "fairness_rad", "label": "Fairness & Bias", "relationship": "e.g."},
-            {"id": "explainability", "label": "Explainability (XAI)", "relationship": "e.g."}
+            {
+              "id": "equipment",
+              "label": "Equipment Needed",
+              "relationship": "Requires",
+              "subnodes": [
+                {"id": "coffee_maker", "label": "Coffee Maker", "relationship": "e.g."},
+                {"id": "mug", "label": "Mug", "relationship": "For"}
+              ]
+            }
           ]
         },
         {
-          "id": "future_trends",
-          "label": "Future AI Trends",
-          "relationship": "looking at",
+          "id": "coffee_process_enjoyment",
+          "label": "Process & Enjoyment",
+          "relationship": "Leads to",
           "subnodes": [
-            {"id": "agi_future", "label": "AGI Development", "relationship": "e.g."},
-            {"id": "quantum_ai", "label": "Quantum AI", "relationship": "e.g."}
+            {
+              "id": "preparation_steps",
+              "label": "Preparation Steps",
+              "relationship": "Involves",
+              "subnodes": [
+                {"id": "grind_beans", "label": "Grind Beans", "relationship": "Step 1 (if whole)"},
+                {"id": "heat_water", "label": "Heat Water", "relationship": "Step 2"}
+              ]
+            }
           ]
         }
       ]
     }
-
-    Returns:
-        str: The filepath to the generated PNG image file.
     """
     try:
         if not json_input.strip():
@@ -92,7 +81,9 @@ def generate_radial_diagram(json_input: str) -> str:
             }
         )
         
-        base_color = '#19191a' # Base color for the central node and gradient
+        # Ensure base_color is valid, fallback if not
+        if not isinstance(base_color, str) or not base_color.startswith('#') or len(base_color) != 7:
+            base_color = '#19191a' # Fallback to default dark if invalid
 
         # Central node styling (rounded box, dark color)
         dot.node(