Create app.py

app.py

@@ -0,0 +1,107 @@
+# app.py
+
+import os
+import gradio as gr
+import json
+
+from HCCDiagnosisListEngine import HCCDiagnosisListEngine
+from chartdiagnosischecker import ChartDiagnosisChecker
+from MeatValidatorAgent import MEATValidatorAgent
+
+APP_TITLE = "HCC Chart Validation & MEAT Criteria Extraction"
+
+
+# ---------- Helper: JSON → Markdown ----------
+def json_to_markdown(meat_results: list[dict]) -> str:
+    if not meat_results:
+        return "⚠️ **No results found.**"
+
+    md = "## 🏥 HCC Chart Validation Report\n\n"
+
+    for entry in meat_results:
+        md += f"### 📌 {entry['diagnosis']} ({entry['icd10']})\n"
+        md += f"- **Reference:** {entry.get('reference', 'N/A')}\n"
+        md += f"- **Answer:** {entry.get('answer', 'unknown').upper()}\n"
+        md += f"- **Rationale:** {entry.get('rationale', '')}\n"
+        md += f"- **Clinical Status:** {entry.get('clinical_status', 'N/A')}\n"
+        md += f"- **MEAT Criteria:**\n"
+        meat = entry.get("meat", {})
+        if meat:
+            md += "  - Monitor: ✅\n" if meat.get("monitor") else "  - Monitor: ❌\n"
+            md += "  - Evaluate: ✅\n" if meat.get("evaluate") else "  - Evaluate: ❌\n"
+            md += "  - Assess: ✅\n" if meat.get("assess") else "  - Assess: ❌\n"
+            md += "  - Treat: ✅\n" if meat.get("treat") else "  - Treat: ❌\n"
+        md += f"- **MEAT Rationale:** {entry.get('meat_rationale', '')}\n\n"
+    return md.strip()
+
+
+# ---------- Pipeline ----------
+def process_pipeline(pdf_file, hcc_code, model_version, csv_path="hcc_mapping.csv"):
+    try:
+        if pdf_file is None:
+            return "⚠️ Please upload a patient chart PDF."
+
+        pdf_path = pdf_file.name
+        patient_name = os.path.splitext(os.path.basename(pdf_path))[0]
+        print(f"\n[PROCESSING] {patient_name}")
+
+        # Step 1: HCC Diagnoses
+        diag_engine = HCCDiagnosisListEngine(hcc_code, model_version, csv_path)
+        diagnoses = diag_engine.run()
+
+        # Step 2: Chart Checking
+        checker = ChartDiagnosisChecker(pdf_path)
+        checked = checker.run(diagnoses)
+
+        # Step 3: MEAT Validation
+        meat_validator = MEATValidatorAgent()
+        meat_results = meat_validator.run(checked)
+
+        return json_to_markdown(meat_results)
+
+    except Exception as e:
+        print(f"[ERROR] {e}")
+        return f"⚠️ Error during processing: {e}"
+
+
+# ---------- Gradio Theme ----------
+simple_theme = gr.themes.Soft(
+    primary_hue=gr.themes.colors.blue,
+    secondary_hue=gr.themes.colors.slate,
+    neutral_hue=gr.themes.colors.slate,
+).set(
+    button_primary_background_fill="#1e40af",
+    button_primary_background_fill_hover="#1d4ed8",
+    button_primary_text_color="white",
+    background_fill_primary="white",
+    background_fill_secondary="#f8fafc",
+)
+
+
+# ---------- Gradio UI ----------
+with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
+    gr.HTML(f"<h1 style='text-align:center;color:#1e40af;'>🏥 {APP_TITLE}</h1>")
+    gr.HTML("<p style='text-align:center;color:#64748b;'>Upload a chart, set HCC + model version, and validate MEAT criteria.</p>")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            pdf_upload = gr.File(label="Upload Patient Chart (PDF)", file_types=[".pdf"])
+            hcc_code = gr.Textbox(label="HCC Code (e.g., 12)", placeholder="Enter HCC code")
+            model_version = gr.Dropdown(choices=["V24", "V28"], label="Model Version", value="V24")
+            run_btn = gr.Button("🚀 Run Validation", variant="primary")
+
+        with gr.Column(scale=2):
+            output_md = gr.Markdown(
+                label="Validation Report",
+                value="📄 Upload a PDF and click **Run Validation** to start.",
+            )
+
+    run_btn.click(
+        fn=process_pipeline,
+        inputs=[pdf_upload, hcc_code, model_version],
+        outputs=[output_md],
+    )
+
+
+if __name__ == "__main__":
+    interface.queue().launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))