Update app.py

app.py

@@ -109,7 +109,6 @@ def json_to_markdown(data) -> str:
                     md += f"<p><b>MEAT Rationale:</b> {diag.get('meat_rationale','')}</p>"
                     md += "</details>"
 
-                # --- MODIFICATION: Filter to only show present comorbidities ---
                 if "comorbidities" in diag and diag["comorbidities"]:
                     present_comorbidities = [c for c in diag["comorbidities"] if c.get("is_present")]
                     if present_comorbidities:
@@ -117,7 +116,6 @@ def json_to_markdown(data) -> str:
                         for c in present_comorbidities:
                             md += f"<li>✅ <b>{c.get('condition')}</b><br/><i>{c.get('rationale')}</i></li>"
                         md += "</ul></details>"
-                # --- END OF MODIFICATION ---
 
                 md += "</div>"
         return md
@@ -286,6 +284,12 @@ def pdf_to_iframe(file):
     except Exception as e:
         return f"<p style='color:red;'>Failed to display PDF: {e}</p>"
 
+# --- NEW FUNCTION: To reset the UI when the PDF is cleared ---
+def clear_outputs():
+    initial_md = "<div style='border:2px solid #1e40af; border-radius:12px; padding:15px; background-color:#f0f9ff;'>📄 Upload a PDF and click <b>Run Validation</b> to start.</div>"
+    initial_preview = "<p>Upload a PDF to preview</p>"
+    return initial_md, initial_preview
+# --- END OF NEW FUNCTION ---
 
 # ---------- Gradio UI ----------
 with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
@@ -311,7 +315,10 @@ with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
                 value="<div style='border:2px solid #1e40af; border-radius:12px; padding:15px; background-color:#f0f9ff;'>📄 Upload a PDF and click <b>Run Validation</b> to start.</div>",
             )
 
+    # --- MODIFICATION: Added .clear() event handler ---
     pdf_upload.change(fn=pdf_to_iframe, inputs=pdf_upload, outputs=pdf_preview)
+    pdf_upload.clear(fn=clear_outputs, inputs=[], outputs=[output_md, pdf_preview])
+    # --- END OF MODIFICATION ---
 
     run_btn.click(
         fn=process_pipeline,
@@ -335,8 +342,6 @@ if __name__ == "__main__":
     )
 
 
-
-
     
 # import os
 # import gradio as gr