Update app.py

app.py

@@ -1,3 +1,4 @@
+# # app.py
 # app.py
 
 import os
@@ -10,47 +11,48 @@ from MeatValidatorAgent import MEATValidatorAgent
 
 APP_TITLE = "Risk Adjustment (HCC Chart Validation)"
 CSV_PATH = "hcc_mapping.csv"
+SAMPLE_PDF = "sample_patient_chart.pdf"  # Place a sample PDF in the same folder
+
 
 # ---------- Helper: JSON → Markdown ----------
 def json_to_markdown(meat_results: list[dict]) -> str:
     if not meat_results:
-        return "⚠️ **No results found.**"
+        return "<div style='color:orange; font-weight:bold;'>⚠️ No results found.</div>"
 
-    md = "## 🏥 HCC Chart Validation Report\n\n"
+    md = "<div style='border:2px solid #1e40af; border-radius:12px; padding:15px; background-color:#f0f9ff;'>"
+    md += "<h2 style='color:#1e40af;'>🩺 HCC Chart Validation Report</h2>"
 
     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"
+        md += f"<h3 style='color:#1d4ed8;'>📌 {entry['diagnosis']} ({entry['icd10']})</h3>"
+        md += f"<p><b>Reference:</b> {entry.get('reference', 'N/A')}<br>"
+        md += f"<b>Answer:</b> {entry.get('answer', 'unknown').upper()}<br>"
+        md += f"<b>Rationale:</b> {entry.get('rationale', '')}<br>"
+        md += f"<b>Clinical Status:</b> {entry.get('clinical_status', 'N/A')}</p>"
+        md += "<b>MEAT Criteria:</b><ul>"
         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()
-
+        md += f"<li>Monitor: {'✅' if meat.get('monitor') else '❌'}</li>"
+        md += f"<li>Evaluate: {'✅' if meat.get('evaluate') else '❌'}</li>"
+        md += f"<li>Assess: {'✅' if meat.get('assess') else '❌'}</li>"
+        md += f"<li>Treat: {'✅' if meat.get('treat') else '❌'}</li>"
+        md += "</ul>"
+        md += f"<b>MEAT Rationale:</b> {entry.get('meat_rationale', '')}<br><br>"
+    md += "</div>"
+    return md
 
 
 # ---------- Pipeline ----------
 def process_pipeline(pdf_file, hcc_code, model_version, csv_path=CSV_PATH):
     try:
         if pdf_file is None:
-            return "⚠️ Please upload a patient chart PDF."
+            return "<div style='color:orange; font-weight:bold;'>⚠️ Please upload a patient chart PDF.</div>"
 
         if not hcc_code or str(hcc_code).strip() == "":
-            return "⚠️ Please enter a valid HCC Code before running validation."
+            return "<div style='color:orange; font-weight:bold;'>⚠️ Please enter a valid HCC Code before running validation.</div>"
 
         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()
@@ -59,10 +61,10 @@ def process_pipeline(pdf_file, hcc_code, model_version, csv_path=CSV_PATH):
         checker = ChartDiagnosisChecker(pdf_path)
         checked = checker.run(diagnoses)
 
-        # ✅ New condition: only proceed if at least one "yes"
+        # ✅ Only proceed if at least one "yes"
         has_positive = any(entry.get("answer", "").lower() == "yes" for entry in checked)
         if not has_positive:
-            return f"**❌ Not found diagnosis in the chart related to HCC code {hcc_code}.**"
+            return f"<b>❌ Not found diagnosis in the chart related to HCC code {hcc_code}.</b>"
 
         # Step 3: MEAT Validation
         meat_validator = MEATValidatorAgent()
@@ -72,9 +74,7 @@ def process_pipeline(pdf_file, hcc_code, model_version, csv_path=CSV_PATH):
 
     except Exception as e:
         print(f"[ERROR] {e}")
-        return f"⚠️ Error during processing: {e}"
-
-
+        return f"<div style='color:red; font-weight:bold;'>⚠️ Error during processing: {e}</div>"
 
 
 # ---------- Gradio Theme ----------
@@ -93,8 +93,13 @@ simple_theme = gr.themes.Soft(
 
 # ---------- 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>")
+    gr.HTML(f"""
+        <h1 style='text-align:center;color:#1e40af;'>🩺 {APP_TITLE}</h1>
+        <p style='text-align:center;color:#64748b;'>
+            Upload a chart, set HCC + model version, and validate MEAT criteria.<br>
+            📄 <a href='{SAMPLE_PDF}' download>Download Sample PDF</a>
+        </p>
+    """)
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -106,7 +111,7 @@ with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
         with gr.Column(scale=2):
             output_md = gr.Markdown(
                 label="Validation Report",
-                value="📄 Upload a PDF and click **Run Validation** to start.",
+                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>",
             )
 
     run_btn.click(
@@ -118,3 +123,143 @@ with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
 
 if __name__ == "__main__":
     interface.queue().launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+#### old
+# import os
+# import gradio as gr
+# import json
+
+# from HCCDiagnosisListEngine import HCCDiagnosisListEngine
+# from chartdiagnosischecker import ChartDiagnosisChecker
+# from MeatValidatorAgent import MEATValidatorAgent
+
+# APP_TITLE = "Risk Adjustment (HCC Chart Validation)"
+# CSV_PATH = "hcc_mapping.csv"
+
+# # ---------- 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=CSV_PATH):
+#     try:
+#         if pdf_file is None:
+#             return "⚠️ Please upload a patient chart PDF."
+
+#         if not hcc_code or str(hcc_code).strip() == "":
+#             return "⚠️ Please enter a valid HCC Code before running validation."
+
+#         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)
+
+#         # ✅ New condition: only proceed if at least one "yes"
+#         has_positive = any(entry.get("answer", "").lower() == "yes" for entry in checked)
+#         if not has_positive:
+#             return f"**❌ Not found diagnosis in the chart related to HCC code {hcc_code}.**"
+
+#         # 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)))