Update class_diagram_generator.py

class_diagram_generator.py

@@ -222,7 +222,7 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             else:
                 html_label += f'<TR><TD ALIGN="CENTER" BGCOLOR="{header_color}"><B>{class_name}</B></TD></TR>'
             
-            html_label += '<HR/>'
+            html_label += f'<TR><TD BGCOLOR="{body_color}" HEIGHT="5"></TD></TR><HR/><TR><TD BGCOLOR="{body_color}" HEIGHT="5"></TD></TR>'
             
             if attributes:
                 for attr in attributes:
@@ -241,7 +241,7 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
             else:
                 html_label += f'<TR><TD ALIGN="LEFT" BGCOLOR="{body_color}"> </TD></TR>'
             
-            html_label += '<HR/>'
+            html_label += f'<TR><TD BGCOLOR="{body_color}" HEIGHT="5"></TD></TR><HR/><TR><TD BGCOLOR="{body_color}" HEIGHT="5"></TD></TR>'
             
             if methods:
                 for method in methods: