Update app.py

app.py

@@ -1,9 +1,10 @@
+# app.py
 import gradio as gr
 import json
 from graphviz import Digraph
 import os
 from tempfile import NamedTemporaryFile
-from sample_data import COMPLEX_SAMPLE_JSON # ¡Aquí está el import!
+from sample_data import COMPLEX_SAMPLE_JSON 
 
 def generate_concept_map(json_input: str) -> str:
     """
@@ -102,32 +103,48 @@ def generate_concept_map(json_input: str) -> str:
                 fontsize='10'
             )
             
-            # Process subnodes (rectangles with lighter fill)
-            for subnode in node.get('subnodes', []):
-                sub_id = subnode.get('id')
-                sub_label = subnode.get('label')
-                sub_rel = subnode.get('relationship')
-                
-                if not all([sub_id, sub_label, sub_rel]):
-                    raise ValueError(f"Invalid subnode: {subnode}")
+            # Helper function to recursively add subnodes and edges
+            def add_subnodes(parent_id, subnodes_list, fill_color, font_size, edge_color, edge_font_size):
+                for subnode in subnodes_list:
+                    sub_id = subnode.get('id')
+                    sub_label = subnode.get('label')
+                    sub_rel = subnode.get('relationship')
                     
-                dot.node(
-                    sub_id,
-                    sub_label,
-                    shape='box',
-                    style='filled',
-                    fillcolor='#FFA726',
-                    fontcolor='white',
-                    fontsize='10'
-                )
-                
-                dot.edge(
-                    node_id,
-                    sub_id,
-                    label=sub_rel,
-                    color='#E91E63',
-                    fontsize='8'
-                )
+                    if not all([sub_id, sub_label, sub_rel]):
+                        raise ValueError(f"Invalid subnode: {subnode}")
+                        
+                    dot.node(
+                        sub_id,
+                        sub_label,
+                        shape='box',
+                        style='filled',
+                        fillcolor=fill_color,
+                        fontcolor='white',
+                        fontsize=str(font_size)
+                    )
+                    
+                    dot.edge(
+                        parent_id,
+                        sub_id,
+                        label=sub_rel,
+                        color=edge_color,
+                        fontsize=str(edge_font_size)
+                    )
+                    
+                    # Recursively call for deeper levels
+                    if 'subnodes' in subnode:
+                        # Slightly adjust colors/sizes for deeper levels if desired
+                        # For fixed 2 children per parent, you might keep colors consistent per level or vary them.
+                        # Here, I'll slightly adjust font size for consistency with depth.
+                        add_subnodes(sub_id, subnode['subnodes'], 
+                                     '#FFA726' if font_size > 8 else '#FFCC80', # Lighter orange/yellow for deeper levels
+                                     font_size - 1 if font_size > 7 else font_size, 
+                                     '#E91E63' if edge_font_size > 7 else '#FF5252', # Reddish for deeper edges
+                                     edge_font_size - 1 if edge_font_size > 7 else edge_font_size)
+            
+            # Start processing subnodes from the first level
+            add_subnodes(node_id, node.get('subnodes', []), '#FFA726', 10, '#E91E63', 8)
+
 
         # Save to temporary file
         with NamedTemporaryFile(delete=False, suffix='.png') as tmp:
@@ -143,7 +160,7 @@ if __name__ == "__main__":
     demo = gr.Interface(
         fn=generate_concept_map,
         inputs=gr.Textbox(
-            value=COMPLEX_SAMPLE_JSON, 
+            value=COMPLEX_SAMPLE_JSON, # ¡Ahora usa el JSON AI simétrico y complejo!
             placeholder="Paste JSON following the documented format",
             label="Structured JSON Input",
             lines=25
@@ -153,8 +170,8 @@ if __name__ == "__main__":
             type="filepath",
             show_download_button=True
         ),
-        title="Advanced Concept Map Generator",
-        description="Create multi-level concept maps from properly formatted JSON"
+        title="Advanced Concept Map Generator (Symmetric AI)",
+        description="Create symmetric, multi-level concept maps for AI from properly formatted JSON."
     )
     
     demo.launch(