Update entity_relationship_generator.py

entity_relationship_generator.py

@@ -244,7 +244,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
             if not entity_name:
                 raise ValueError(f"Invalid entity: {entity}")
             
-            entity_label = f"<B>{entity_name}</B>"
+            entity_label = f"{entity_name}"
             
             if attributes:
                 entity_label += "|"
@@ -261,11 +261,11 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                     
                     if attr_type == 'primary_key':
                         if is_multivalued:
-                            primary_keys.append(f"<U>{{{{ {attr_name} }}}}</U>")
+                            primary_keys.append(f"PK: {{ {attr_name} }}")
                         else:
-                            primary_keys.append(f"<U>{attr_name}</U>")
+                            primary_keys.append(f"PK: {attr_name}")
                     elif attr_type == 'foreign_key':
-                        foreign_keys.append(f"<I>{attr_name}</I> (FK)")
+                        foreign_keys.append(f"FK: {attr_name}")
                     else:
                         attr_display = attr_name
                         if is_derived:
@@ -292,7 +292,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
             
             dot.node(
                 entity_name,
-                f"<{entity_label}>",
+                entity_label,
                 shape=shape,
                 style=style,
                 fillcolor=fillcolor,
@@ -326,7 +326,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
             
             dot.node(
                 rel_name,
-                f"<B>{rel_name}</B>",
+                rel_name,
                 shape=rel_shape,
                 style=rel_style,
                 fillcolor=rel_color,