Update class_diagram_generator.py

class_diagram_generator.py

@@ -486,20 +486,17 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
         {
           "from": "Car",
           "to": "Vehicle",
-          "type": "inheritance",
-          "label": "extends"
+          "type": "inheritance"
         },
         {
           "from": "Motorcycle",
           "to": "Vehicle",
-          "type": "inheritance",
-          "label": "extends"
+          "type": "inheritance"
         },
         {
           "from": "Car",
           "to": "Engine",
           "type": "composition",
-          "label": "has",
           "multiplicity_from": "1",
           "multiplicity_to": "1"
         },
@@ -507,7 +504,6 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
           "from": "Motorcycle",
           "to": "Engine",
           "type": "composition",
-          "label": "has",
           "multiplicity_from": "1",
           "multiplicity_to": "1"
         },
@@ -515,7 +511,6 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
           "from": "Car",
           "to": "TransmissionType",
           "type": "association",
-          "label": "uses",
           "multiplicity_from": "1",
           "multiplicity_to": "1"
         },
@@ -523,21 +518,18 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
           "from": "Vehicle",
           "to": "FuelType",
           "type": "association",
-          "label": "uses",
           "multiplicity_from": "1",
           "multiplicity_to": "1"
         },
         {
           "from": "GarageService",
           "to": "VehicleService",
-          "type": "realization",
-          "label": "implements"
+          "type": "realization"
         },
         {
           "from": "GarageService",
           "to": "Vehicle",
           "type": "dependency",
-          "label": "services",
           "multiplicity_from": "1",
           "multiplicity_to": "*"
         }
@@ -556,7 +548,6 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
         if 'classes' not in data:
             raise ValueError("Missing required field: classes")
 
-        # Configuración del diagrama usando la misma estrategia exitosa del mapa conceptual
         dot = graphviz.Digraph(comment='Class Diagram')
         dot.attr(rankdir='TB', bgcolor='white', pad='0.5', nodesep='0.8', ranksep='1.2', splines='ortho')
         dot.attr('node', shape='plaintext', fontname='Arial', fontsize='11')
@@ -574,10 +565,8 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             if not class_name:
                 continue
             
-            # Crear tabla HTML para estructura de clase
             html_label = '<TABLE BORDER="1" CELLBORDER="0" CELLSPACING="0" CELLPADDING="5" BGCOLOR="white">'
             
-            # Header con nombre de clase y estereotipo
             if class_type == 'abstract':
                 html_label += f'<TR><TD ALIGN="CENTER"><B>&lt;&lt;abstract&gt;&gt;<BR/>{class_name}</B></TD></TR>'
             elif class_type == 'interface':
@@ -587,10 +576,8 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             else:
                 html_label += f'<TR><TD ALIGN="CENTER"><B>{class_name}</B></TD></TR>'
             
-            # Línea separadora después del nombre
             html_label += '<HR/>'
             
-            # Sección de atributos
             if attributes:
                 for attr in attributes:
                     visibility = attr.get('visibility', '+')
@@ -602,17 +589,14 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                     if attr_type:
                         line += f" : {attr_type}"
                     if is_static:
-                        line = f"<U>{line}</U>"  # Subrayado para elementos estáticos
+                        line = f"<U>{line}</U>"
                     
                     html_label += f'<TR><TD ALIGN="LEFT">{line}</TD></TR>'
             else:
-                # Espacio vacío si no hay atributos
                 html_label += '<TR><TD ALIGN="LEFT"> </TD></TR>'
             
-            # Línea separadora entre atributos y métodos
             html_label += '<HR/>'
             
-            # Sección de métodos
             if methods:
                 for method in methods:
                     visibility = method.get('visibility', '+')
@@ -633,21 +617,18 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                     line += f") : {return_type}"
                     
                     if is_static:
-                        line = f"<U>{line}</U>"  # Subrayado para métodos estáticos
+                        line = f"<U>{line}</U>"
                     if is_abstract:
-                        line = f"<I>{line}</I>"  # Cursiva para métodos abstractos
+                        line = f"<I>{line}</I>"
                     
                     html_label += f'<TR><TD ALIGN="LEFT">{line}</TD></TR>'
             else:
-                # Espacio vacío si no hay métodos
                 html_label += '<TR><TD ALIGN="LEFT"> </TD></TR>'
             
             html_label += '</TABLE>'
             
-            # Agregar nodo con la tabla HTML
             dot.node(class_name, f'<{html_label}>')
         
-        # Procesar relaciones con líneas rectas
         for relationship in relationships:
             from_class = relationship.get('from')
             to_class = relationship.get('to')
@@ -660,19 +641,15 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                 continue
             
             edge_attrs = {
-                'color': 'black'  # Simplificamos a solo color negro, sin configuraciones conflictivas
+                'color': 'black'
             }
             
-            if label:
-                edge_attrs['label'] = label
-            
             if multiplicity_from:
                 edge_attrs['taillabel'] = multiplicity_from
             
             if multiplicity_to:
                 edge_attrs['headlabel'] = multiplicity_to
             
-            # Configurar estilos de flecha según el tipo de relación
             if rel_type == 'inheritance':
                 edge_attrs['arrowhead'] = 'empty'
                 edge_attrs['color'] = 'black'
@@ -694,13 +671,12 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                 edge_attrs['arrowhead'] = 'normal'
                 edge_attrs['style'] = 'dashed'
                 edge_attrs['color'] = 'black'
-            else:  # association
+            else:
                 edge_attrs['arrowhead'] = 'normal'
                 edge_attrs['color'] = 'black'
             
             dot.edge(from_class, to_class, **edge_attrs)
 
-        # Renderizar el diagrama
         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}"