Update app.py

app.py

@@ -227,16 +227,11 @@ def predict(text: str):
     # ── 4. Inline word-highlight HTML ──
     word_html = _build_highlight_html(tokens, vals_severe)
 
-    # ── 5. SHAP bar/force plot image ──
-    fig = _build_shap_plot(tokens, vals_severe, text)
-
-    # ── 6. NER Pipeline ──
+    # ── 5. NER Pipeline ──
     ner_results = ner_pipe(text)
     ner_html = _build_ner_html(text, ner_results)
 
-    return verdict_html, word_html, fig, ner_html  # add ner_html
-
-    return verdict_html, word_html, fig
+    return verdict_html, word_html, ner_html
 
 # ── HTML word highlights ──────────────────────────────────────────────────────
 def _build_highlight_html(tokens, values):
@@ -354,11 +349,11 @@ def _build_shap_plot(tokens, values, text):
 
 # ── Gradio UI ─────────────────────────────────────────────────────────────────
 EXAMPLES = [
-    ["I felt groggy and dizzy after taking Tylenol."],
-    ["I had a mild headache after the medication."],
-    ["I experienced severe chest pain and difficulty breathing after the injection."],
-    ["My stomach felt slightly upset after the pill."],
-    ["I had an anaphylactic reaction with hives and throat swelling after penicillin."],
+    ["I took ibuprofen for my back pain this morning. About an hour later, I noticed mild stomach discomfort and felt slightly nauseous. The symptoms went away after I ate something."],
+    ["I was prescribed amoxicillin for a sinus infection last week. I developed a mild rash on my arms after the second dose. My doctor said it was a minor allergic reaction and switched me to a different antibiotic."],
+    ["After receiving the contrast dye injection for my CT scan, I immediately felt my throat tightening and had difficulty breathing. I broke out in hives across my chest and face. The medical team administered epinephrine and I was kept under observation for several hours."],
+    ["I am a 67 year old male who started taking metformin for type 2 diabetes last month. I have been experiencing persistent nausea and occasional vomiting since starting the medication. My blood sugar levels have improved but the gastrointestinal side effects are making it difficult to continue."],
+    ["My 8 year old daughter was given penicillin for strep throat. Within 30 minutes she complained of dizziness and her lips began to swell. We rushed her to the emergency room where she was treated for anaphylaxis."],
 ]
 
 CSS = """
@@ -393,28 +388,27 @@ with gr.Blocks(css=CSS, theme=gr.themes.Base()) as demo:
         with gr.Column(scale=1):
             text_input = gr.Textbox(
                 label="Describe your reaction",
-                placeholder="e.g. I felt groggy and dizzy after taking Tylenol.",
-                lines=3,
+                placeholder="e.g. I took ibuprofen for my back pain this morning.",
+                lines=5,
             )
             analyze_btn = gr.Button("🔍 Analyze Reaction", variant="primary")
             gr.Examples(examples=EXAMPLES, inputs=text_input, label="Try an example")
-
         with gr.Column(scale=1):
             verdict_out = gr.HTML(label="Severity Verdict")
 
     gr.Markdown("### 🎨 Word-Level SHAP Highlights")
     highlight_out = gr.HTML(label="Token Highlights")
 
-    gr.Markdown("### 📊 SHAP Token Importance Plot")
-    plot_out = gr.Plot(label="SHAP Bar Chart")
-
     gr.Markdown("### 🏥 Medical Entity Recognition")
     ner_out = gr.HTML(label="Detected Entities")
 
     analyze_btn.click(
         fn=predict,
         inputs=text_input,
-        outputs=[verdict_out, highlight_out, plot_out, ner_out],  # add ner_out
+        outputs=[verdict_out, highlight_out, ner_out],
     )
+
+if __name__ == "__main__":
+    demo.launch()
 if __name__ == "__main__":
     demo.launch()