Update class_diagram_generator.py

class_diagram_generator.py

@@ -208,17 +208,7 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                 'bgcolor': 'white',
                 'pad': '0.5',
                 'nodesep': '1.0',
-                'ranksep': '1.5',
-                'fontname': 'Arial',
-                'dpi': '300'
-            },
-            node_attr={
-                'fontname': 'Arial',
-                'fontsize': '10'
-            },
-            edge_attr={
-                'fontname': 'Arial',
-                'fontsize': '9'
+                'ranksep': '1.5'
             }
         )
         
@@ -234,19 +224,20 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             if not class_name:
                 raise ValueError(f"Invalid class: {cls}")
             
-            class_label = ""
+            class_parts = []
             
+            header = class_name
             if class_type == 'abstract':
-                class_label += "<<abstract>>\\n"
+                header = f"<<abstract>>\\n{class_name}"
             elif class_type == 'interface':
-                class_label += "<<interface>>\\n"
+                header = f"<<interface>>\\n{class_name}"
             elif class_type == 'enum':
-                class_label += "<<enumeration>>\\n"
+                header = f"<<enumeration>>\\n{class_name}"
             
-            class_label += f"{class_name}"
+            class_parts.append(header)
             
             if attributes:
-                class_label += "|"
+                attr_section = ""
                 for attr in attributes:
                     visibility = attr.get('visibility', '+')
                     name = attr.get('name', '')
@@ -257,14 +248,18 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                     if attr_type:
                         attr_line += f" : {attr_type}"
                     if is_static:
-                        attr_line = f"{attr_line} (static)"
-                    class_label += f"{attr_line}\\l"
+                        attr_line += " [static]"
+                    
+                    if attr_section:
+                        attr_section += "\\l"
+                    attr_section += attr_line
+                
+                if attr_section:
+                    attr_section += "\\l"
+                    class_parts.append(attr_section)
             
             if methods:
-                if not attributes:
-                    class_label += "|"
-                else:
-                    class_label += "|"
+                method_section = ""
                 for method in methods:
                     visibility = method.get('visibility', '+')
                     name = method.get('name', '')
@@ -284,18 +279,26 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                     method_line += f") : {return_type}"
                     
                     if is_static:
-                        method_line = f"{method_line} (static)"
+                        method_line += " [static]"
                     if is_abstract:
-                        method_line = f"{method_line} (abstract)"
+                        method_line += " [abstract]"
                     
-                    class_label += f"{method_line}\\l"
+                    if method_section:
+                        method_section += "\\l"
+                    method_section += method_line
+                
+                if method_section:
+                    method_section += "\\l"
+                    class_parts.append(method_section)
+            
+            class_label = "|".join(class_parts)
             
             if class_type == 'interface':
                 style = 'filled,dashed'
-                fillcolor = '#f0f0f0'
+                fillcolor = '#f5f5f5'
             elif class_type == 'abstract':
                 style = 'filled'
-                fillcolor = '#e8e8e8'
+                fillcolor = '#eeeeee'
             else:
                 style = 'filled'
                 fillcolor = 'white'
@@ -306,8 +309,7 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                 shape='record',
                 style=style,
                 fillcolor=fillcolor,
-                color='black',
-                penwidth='1'
+                color='black'
             )
         
         for relationship in relationships:
@@ -321,43 +323,35 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             if not all([from_class, to_class]):
                 raise ValueError(f"Invalid relationship: {relationship}")
             
-            edge_attrs = {
-                'color': 'black',
-                'fontcolor': 'black',
-                'fontsize': '9'
-            }
+            edge_attrs = {'color': 'black'}
             
             if label:
                 edge_attrs['label'] = label
             
-            if multiplicity_from and multiplicity_to:
-                edge_attrs['headlabel'] = multiplicity_to
+            if multiplicity_from:
                 edge_attrs['taillabel'] = multiplicity_from
             
+            if multiplicity_to:
+                edge_attrs['headlabel'] = multiplicity_to
+            
             if rel_type == 'inheritance':
                 edge_attrs['arrowhead'] = 'empty'
-                edge_attrs['arrowsize'] = '1.2'
             elif rel_type == 'composition':
                 edge_attrs['arrowhead'] = 'normal'
                 edge_attrs['arrowtail'] = 'diamond'
                 edge_attrs['dir'] = 'both'
-                edge_attrs['arrowsize'] = '1.0'
             elif rel_type == 'aggregation':
                 edge_attrs['arrowhead'] = 'normal'
                 edge_attrs['arrowtail'] = 'odiamond'
                 edge_attrs['dir'] = 'both'
-                edge_attrs['arrowsize'] = '1.0'
             elif rel_type == 'realization':
                 edge_attrs['arrowhead'] = 'empty'
                 edge_attrs['style'] = 'dashed'
-                edge_attrs['arrowsize'] = '1.2'
             elif rel_type == 'dependency':
                 edge_attrs['arrowhead'] = 'normal'
                 edge_attrs['style'] = 'dashed'
-                edge_attrs['arrowsize'] = '1.0'
             else:
                 edge_attrs['arrowhead'] = 'normal'
-                edge_attrs['arrowsize'] = '1.0'
             
             dot.edge(from_class, to_class, **edge_attrs)