Update app.py

app.py

@@ -8,15 +8,14 @@ from HCCDiagnosisListEngine import HCCDiagnosisListEngine
 from chartdiagnosischecker import ChartDiagnosisChecker
 from MeatValidatorAgent import MEATValidatorAgent
 from dotenv import load_dotenv
-
+load_dotenv()
 
 APP_TITLE = "Risk Adjustment (HCC Chart Validation)"
 CSV_PATH = "hcc_mapping.csv"
-SAMPLE_PDF = "sample_patient_chart.pdf"
-
+SAMPLE_PDF = "sample_patient_chart.pdf"  # Place a sample PDF in the same folder
 
 
-# ---------- Helper: JSON → Markdown (Unchanged) ----------
+# ---------- Helper: JSON → Markdown ----------
 def json_to_markdown(meat_results) -> str:
     if not meat_results:
         return "<div style='color:orange; font-weight:bold;'>⚠️ No results found.</div>"
@@ -42,23 +41,35 @@ def json_to_markdown(meat_results) -> str:
     return md
 
 
-# ---------- Pipeline (Unchanged Core Logic) ----------
+# ---------- Pipeline ----------
 def process_pipeline(pdf_file, hcc_code, model_version, csv_path=CSV_PATH):
     try:
+        if pdf_file is None:
+            return "<div style='color:orange; font-weight:bold;'>⚠️ Please upload a patient chart PDF.</div>"
+
+        # Safely handle None hcc_code
+        hcc_code_str = str(hcc_code or "").strip()
+        if not hcc_code_str:
+            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}")
 
-        diag_engine = HCCDiagnosisListEngine(hcc_code, model_version, csv_path)
+        # Step 1: HCC Diagnoses
+        diag_engine = HCCDiagnosisListEngine(hcc_code_str, model_version, csv_path)
         diagnoses = diag_engine.run()
 
+        # Step 2: Chart Checking
         checker = ChartDiagnosisChecker(pdf_path)
         checked = checker.run(diagnoses)
 
+        # ✅ 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"<div style='border:2px solid #f59e0b; border-radius:12px; padding:15px; background-color:#fffbeb;'><h3 style='color:#d97706;'>⚠️ No Matching Diagnosis Found</h3><p>No diagnosis in the chart was found to be directly related to <b>HCC code {hcc_code}</b>.</p></div>"
+            return f"<b>❌ Not found diagnosis in the chart related to HCC code {hcc_code_str}.</b>"
 
+        # Step 3: MEAT Validation
         meat_validator = MEATValidatorAgent()
         meat_results = meat_validator.run(checked)
 
@@ -68,43 +79,8 @@ def process_pipeline(pdf_file, hcc_code, model_version, csv_path=CSV_PATH):
         print(f"[ERROR] {e}")
         return f"<div style='color:red; font-weight:bold;'>⚠️ Error during processing: {e}</div>"
 
-# ---------- NEW: Gradio UI Event Handler to run the pipeline and switch views ----------
-def run_and_display_pdf(pdf_file, hcc_code, model_version):
-    # 1. Validate inputs first
-    if pdf_file is None:
-        return {
-            output_md: "<div style='color:orange; font-weight:bold;'>⚠️ Please upload a patient chart PDF.</div>"
-        }
-    hcc_code_str = str(hcc_code or "").strip()
-    if not hcc_code_str:
-        return {
-            output_md: "<div style='color:orange; font-weight:bold;'>⚠️ Please enter a valid HCC Code before running validation.</div>"
-        }
-
-    # 2. Run the main processing pipeline
-    report_markdown = process_pipeline(pdf_file, hcc_code_str, model_version)
-
-    # 3. Return a dictionary of updates for each component on the screen
-    return {
-        output_md: report_markdown,
-        pdf_viewer: gr.update(value=pdf_file.name, visible=True),
-        input_controls: gr.update(visible=False),
-        reset_btn: gr.update(visible=True)
-    }
-
-# ---------- NEW: Function to reset the UI to its initial state ----------
-def reset_ui():
-    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>"
-    return {
-        pdf_upload: gr.update(value=None),
-        hcc_code: gr.update(value=""),
-        output_md: initial_md,
-        pdf_viewer: gr.update(visible=False, value=None),
-        input_controls: gr.update(visible=True),
-        reset_btn: gr.update(visible=False)
-    }
-
-# ---------- Gradio Theme (Unchanged) ----------
+
+# ---------- Gradio Theme ----------
 simple_theme = gr.themes.Soft(
     primary_hue=gr.themes.colors.blue,
     secondary_hue=gr.themes.colors.slate,
@@ -117,7 +93,19 @@ simple_theme = gr.themes.Soft(
     background_fill_secondary="#f8fafc",
 )
 
-# ---------- MODIFIED Gradio UI ----------
+
+# ---------- Helper for Sample PDF ----------
+def load_sample_pdf():
+    if not os.path.exists(SAMPLE_PDF):
+        raise FileNotFoundError(f"Sample PDF not found at {SAMPLE_PDF}")
+    # Create a simple wrapper so it has a `.name` attribute
+    class PDFWrapper:
+        def __init__(self, path):
+            self.name = path
+    return PDFWrapper(SAMPLE_PDF)
+
+
+# ---------- 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>
@@ -126,46 +114,306 @@ with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
         </p>
     """)
 
-    with gr.Row(equal_height=False):
-        # --- Left Column: Contains both the inputs and the PDF viewer ---
+    with gr.Row():
         with gr.Column(scale=1):
-            # Group all input controls to easily hide/show them together
-            with gr.Column(visible=True) as input_controls:
-                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")
-                gr.Examples(
-                    examples=[[SAMPLE_PDF, "12", "V24"]],
-                    inputs=[pdf_upload, hcc_code, model_version],
-                )
-
-            # PDF viewer, initially hidden
-            pdf_viewer = gr.File(label="Patient Chart Viewer", visible=False, height=800)
-
-            # Reset button, also initially hidden
-            reset_btn = gr.Button("🔄 Analyze New Chart", visible=False)
-
-        # --- Right Column: For the report output ---
+            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="<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>",
             )
 
-    # ---------- Connect the button clicks to the new handler functions ----------
+    # ---------- Run validation ----------
     run_btn.click(
-        fn=run_and_display_pdf,
+        fn=process_pipeline,
         inputs=[pdf_upload, hcc_code, model_version],
-        outputs=[output_md, pdf_viewer, input_controls, reset_btn],
+        outputs=[output_md],
     )
 
-    reset_btn.click(
-        fn=reset_ui,
-        inputs=None,
-        # The outputs list must include all components that the reset_ui function updates
-        outputs=[pdf_upload, hcc_code, output_md, pdf_viewer, input_controls, reset_btn]
+    # ---------- Sample PDF using Gradio Examples ----------
+    gr.Examples(
+        examples=[[SAMPLE_PDF]],
+        inputs=[pdf_upload],
+        outputs=[output_md],
+        fn=lambda x: process_pipeline(load_sample_pdf(), hcc_code="12", model_version="V24"),
+        cache_examples=False
     )
 
+
 if __name__ == "__main__":
     interface.queue().launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+
+
+
+
+
+# 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"
+# 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 "<div style='color:orange; font-weight:bold;'>⚠️ No results found.</div>"
+
+#     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"<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", {})
+#         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 "<div style='color:orange; font-weight:bold;'>⚠️ Please upload a patient chart PDF.</div>"
+
+#         if not hcc_code or str(hcc_code).strip() == "":
+#             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()
+
+#         # Step 2: Chart Checking
+#         checker = ChartDiagnosisChecker(pdf_path)
+#         checked = checker.run(diagnoses)
+
+#         # ✅ 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"<b>❌ Not found diagnosis in the chart related to HCC code {hcc_code}.</b>"
+
+#         # 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"<div style='color:red; font-weight:bold;'>⚠️ Error during processing: {e}</div>"
+
+
+# # ---------- 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>
+#         <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")
+#             sample_btn = gr.Button("📄 Use Sample PDF", variant="secondary")
+
+#         with gr.Column(scale=2):
+#             output_md = gr.Markdown(
+#                 label="Validation Report",
+#                 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>",
+#             )
+
+#     # ---------- Button Events ----------
+#     sample_btn.click(
+#         fn=lambda: gr.File.update(value=SAMPLE_PDF),
+#         inputs=[],
+#         outputs=[pdf_upload],
+#     )
+
+#     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)))
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+#### 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)))