Update app.py

app.py

@@ -396,8 +396,8 @@ with gr.Blocks(title="ADR Detector", css=custom_css, theme=gr.themes.Soft()) as
             "and key medical entities."
         )
 
+    # ── Row 1: Input + Classification/Severity ────────────────────────────────
     with gr.Row():
-        # ── Left column: input ────────────────────────────────────────────────
         with gr.Column(scale=1):
             gr.Markdown("### Input")
             prob1 = gr.Textbox(
@@ -423,7 +423,6 @@ with gr.Blocks(title="ADR Detector", css=custom_css, theme=gr.themes.Soft()) as
                 inputs=[prob1],
             )
 
-        # ── Right column: outputs ─────────────────────────────────────────────
         with gr.Column(scale=1):
             gr.Markdown("### Classification")
             label = gr.Label(label="Severity Probability")
@@ -431,9 +430,13 @@ with gr.Blocks(title="ADR Detector", css=custom_css, theme=gr.themes.Soft()) as
             gr.Markdown("### Severity Rating")
             severity_out = gr.HTML(label="Severity Rating", elem_classes="output-box")
 
+    # ── Row 2: Medical Entities (left) | SHAP chart (right) ──────────────────
+    with gr.Row():
+        with gr.Column(scale=1):
             gr.Markdown("### Medical Entities")
             htext_out = gr.HTML(label="NER Mapping", elem_classes="output-box")
 
+        with gr.Column(scale=1):
             gr.Markdown("### Model Logic (SHAP)")
             shap_out = gr.HTML(label="Feature Importance", elem_classes="output-box")