Update entity_relationship_generator.py

entity_relationship_generator.py

@@ -435,10 +435,10 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                         height='0.6'
                     )
                     
-                    # Conectar padre con ISA
+                    # Conectar padre con ISA (sin cardinalidad)
                     dot.edge(parent, rel_name, len='1.5', color='#404040', penwidth='2')
                     
-                    # Conectar ISA con hijos
+                    # Conectar ISA con hijos (sin cardinalidad)
                     for child in children:
                         dot.edge(rel_name, child, len='1.5', color='#404040', penwidth='2')
                 
@@ -456,13 +456,29 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                     height='1.0'
                 )
                 
-                # Conectar entidades con cardinalidades
-                for entity in entities_involved:
-                    cardinality = cardinalities.get(entity, '1')
+                # Conectar entidades con cardinalidades posicionadas correctamente
+                if len(entities_involved) >= 2:
+                    entity1, entity2 = entities_involved[0], entities_involved[1]
+                    card1 = cardinalities.get(entity1, '1')
+                    card2 = cardinalities.get(entity2, '1')
+                    
+                    # Primera entidad -> relación (cardinalidad cerca de la entidad)
                     dot.edge(
-                        entity, 
+                        entity1, 
                         rel_name, 
-                        label=f'  {cardinality}  ',
+                        taillabel=card1,
+                        len='2.0',
+                        color='#404040',
+                        penwidth='2',
+                        fontcolor='#000000',
+                        fontsize='12'
+                    )
+                    
+                    # Relación -> segunda entidad (cardinalidad cerca de la entidad)
+                    dot.edge(
+                        rel_name,
+                        entity2, 
+                        headlabel=card2,
                         len='2.0',
                         color='#404040',
                         penwidth='2',
@@ -484,13 +500,29 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                     height='1.0'
                 )
                 
-                # Conectar entidades con cardinalidades bien posicionadas
-                for entity in entities_involved:
-                    cardinality = cardinalities.get(entity, '1')
+                # Conectar entidades con cardinalidades posicionadas correctamente
+                if len(entities_involved) >= 2:
+                    entity1, entity2 = entities_involved[0], entities_involved[1]
+                    card1 = cardinalities.get(entity1, '1')
+                    card2 = cardinalities.get(entity2, '1')
+                    
+                    # Primera entidad -> relación (cardinalidad cerca de la entidad)
                     dot.edge(
-                        entity, 
+                        entity1, 
                         rel_name, 
-                        label=f'  {cardinality}  ',
+                        taillabel=card1,
+                        len='2.0',
+                        color='#404040',
+                        penwidth='2',
+                        fontcolor='#000000',
+                        fontsize='12'
+                    )
+                    
+                    # Relación -> segunda entidad (cardinalidad cerca de la entidad)
+                    dot.edge(
+                        rel_name,
+                        entity2, 
+                        headlabel=card2,
                         len='2.0',
                         color='#404040',
                         penwidth='2',