Update app.py

app.py

@@ -72,6 +72,7 @@ def adr_predict(x):
         htext = "<p style='color: black;'>NER processing error.</p>"
         
     label_output = {"Severe Reaction": float(scores[1]), "Non-severe Reaction": float(scores[0])}
+    
     return label_output, local_plot, htext
 
 # FIX: Added !important tags to ensure Gradio's dark mode doesn't override the white background and black text
@@ -86,7 +87,7 @@ with gr.Blocks(title="ADR Detector") as demo:
     with gr.Column(elem_classes="main-header"):
         gr.Markdown("# Adverse Drug Reaction (ADR) Detector")
         gr.Markdown("Analyze clinical text for potential medication-related severity and key medical entities.")
-
+        
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Input")
@@ -106,20 +107,23 @@ with gr.Blocks(title="ADR Detector") as demo:
                 ],
                 inputs=[prob1]
             )
-
+            
         with gr.Column(scale=1):
             gr.Markdown("### Classification")
             label = gr.Label(label="Severity Probability")
             
-            with gr.Tabs():
-                with gr.TabItem("Medical Entities"):
-                    htext = gr.HTML(label="NER Mapping", elem_classes="output-box")
-                with gr.TabItem("Model Logic (SHAP)"):
-                    local_plot = gr.HTML(label='Feature Importance', elem_classes="output-box")
+            # --- TABS REMOVED HERE ---
+            # Both components are now stacked sequentially in the column
+            
+            gr.Markdown("### Medical Entities")
+            htext = gr.HTML(label="NER Mapping", elem_classes="output-box")
+            
+            gr.Markdown("### Model Logic (SHAP)")
+            local_plot = gr.HTML(label='Feature Importance', elem_classes="output-box")
 
     gr.Markdown("---")
     gr.Markdown("Disclaimer: This tool is for research purposes only and does not constitute medical advice.")
-
+    
     submit_btn.click(
         fn=adr_predict,
         inputs=[prob1],