Update app.py

app.py

@@ -87,40 +87,41 @@ if __name__ == "__main__":
 
         with gr.Tabs():
             with gr.TabItem("Class Diagram"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        json_input_cd = gr.Textbox(
-                            value=CLASS_DIAGRAM_JSON,
-                            placeholder="Paste JSON following the documented format",
-                            label="JSON Input",
-                            lines=20
-                        )
-                        submit_btn_cd = gr.Button("Generate Class Diagram", variant="primary")
-                    
-                    with gr.Column(scale=2):
-                        output_cd = gr.Image(
-                            label="Generated Diagram",
-                            type="filepath",
-                            show_download_button=True,
-                            height=500
-                        )
-                
-                submit_btn_cd.click(
-                    fn=generate_class_diagram, 
-                    inputs=[json_input_cd, output_format_radio], 
-                    outputs=output_cd
-                )
-                
-                gr.Markdown("## Examples")
-                with gr.Row(elem_classes=["example-images"]):
-                    gr.Image(value="./images/cd1.svg", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
-                    gr.Image(value="./images/cd2.svg", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
-                with gr.Row(elem_classes=["example-images"]):
-                    gr.Image(value="./images/cd3.svg", label="Sample 3", show_label=True, interactive=False, height="auto", width="100%")
-                    gr.Image(value="./images/cd4.svg", label="Sample 4", show_label=True, interactive=False, height="auto", width="100%")
-                with gr.Row(elem_classes=["example-images"]):
-                    gr.Image(value="./images/cd5.svg", label="Sample 5", show_label=True, interactive=False, height="auto", width="100%")
-                    gr.Image(value="./images/cd6.svg", label="Sample 6", show_label=True, interactive=False, height="auto", width="100%")
+               with gr.Row():
+                   with gr.Column(scale=1):
+                       json_input_cd = gr.Textbox(
+                           value=CLASS_DIAGRAM_JSON,
+                           placeholder="Paste JSON following the documented format",
+                           label="JSON Input",
+                           lines=20
+                       )
+                       submit_btn_cd = gr.Button("Generate Class Diagram", variant="primary")
+                   
+                   with gr.Column(scale=2):
+                       output_cd = gr.Image(
+                           label="Generated Diagram",
+                           type="filepath",
+                           show_download_button=True,
+                           height=500
+                       )
+               
+               submit_btn_cd.click(
+                   fn=generate_class_diagram, 
+                   inputs=[json_input_cd, output_format_radio], 
+                   outputs=output_cd
+               )
+               
+               gr.Markdown("## Examples")
+               with gr.Row():
+                   with gr.Column(scale=1):
+                       gr.Image(value="./images/cd1.svg", label="Sample 1", show_label=True, interactive=False, height="auto")
+                       gr.Image(value="./images/cd3.svg", label="Sample 3", show_label=True, interactive=False, height="auto")
+                       gr.Image(value="./images/cd5.svg", label="Sample 5", show_label=True, interactive=False, height="auto")
+                   
+                   with gr.Column(scale=1):
+                       gr.Image(value="./images/cd2.svg", label="Sample 2", show_label=True, interactive=False, height="auto")
+                       gr.Image(value="./images/cd4.svg", label="Sample 4", show_label=True, interactive=False, height="auto")
+                       gr.Image(value="./images/cd6.svg", label="Sample 6", show_label=True, interactive=False, height="auto")
 
             # with gr.TabItem("Entity Relationship"):
             #     with gr.Row():