Update entity_relationship_generator.py

entity_relationship_generator.py

@@ -840,25 +840,10 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
         dot.attr('node', fontname='Arial', fontsize='10', color='#404040')
         dot.attr('edge', fontname='Arial', fontsize='9', color='#4a4a4a')
         
-        entity_colors = {
-            'strong': '#BEBEBE',
-            'weak': '#FFF9C4'
-        }
-        
-        attribute_colors = {
-            'primary_key': '#A8E6CF',
-            'partial_key': '#FFB3BA',
-            'multivalued': '#B8D4F1',
-            'derived': '#F0F8FF',
-            'composite': '#E8E8E8',
-            'regular': '#D4D4D4'
-        }
-        
-        relationship_colors = {
-            'regular': '#E8E8E8',
-            'identifying': '#B8D4F1',
-            'isa': '#F0F0F0'
-        }
+        entity_color = '#BEBEBE'
+        attribute_color = '#B8D4F1'
+        relationship_color = '#FFF9C4'
+        isa_color = '#A8E6CF'
         
         font_color = 'black'
         
@@ -873,7 +858,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
             if not entity_name:
                 continue
             
-            entity_color = entity_colors.get(entity_type, entity_colors['strong'])
+            entity_color = '#BEBEBE'
             
             if entity_type == 'weak':
                 dot.node(
@@ -909,7 +894,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                 attr_type = attr.get('type', 'regular')
                 
                 attr_id = f"{entity_name}_attr_{i}"
-                attr_color = attribute_colors.get(attr_type, attribute_colors['regular'])
+                attr_color = attribute_color
                 
                 if attr_type == 'primary_key':
                     dot.node(
@@ -1009,7 +994,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                 
                 if parent and children:
                     isa_id = f"isa_{rel_name}"
-                    isa_color = relationship_colors.get('isa', relationship_colors['regular'])
+                    isa_color = isa_color
                     
                     dot.node(
                         isa_id, 
@@ -1031,7 +1016,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                         dot.edge(isa_id, child, color='#4a4a4a', len='2.0')
                 
             elif len(entities_involved) >= 2:
-                rel_color = relationship_colors.get(rel_type, relationship_colors['regular'])
+                rel_color = relationship_color
                 
                 if rel_type == 'identifying':
                     dot.node(
@@ -1065,7 +1050,7 @@ def generate_entity_relationship_diagram(json_input: str, output_format: str) ->
                 for j, attr in enumerate(rel_attributes):
                     attr_name = attr.get('name', '')
                     attr_id = f"{rel_name}_attr_{j}"
-                    attr_color = attribute_colors['regular']
+                    attr_color = attribute_color
                     
                     dot.node(
                         attr_id,