Update app.py

app.py

@@ -17,17 +17,30 @@ 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(output_data: dict) -> str:
+def json_to_markdown(data) -> str:
     """
-    Convert final_analysis JSON into a clean, styled Markdown/HTML string.
+    Convert JSON (dict or list) into a clean, styled Markdown/HTML string.
+    Handles both:
+      - Full dict with patient_id, hcc_code, model_version, final_analysis
+      - Just a list of diagnoses
     """
     try:
-        patient_id = output_data.get("patient_id", "Unknown Patient")
-        hcc_code = output_data.get("hcc_code", "N/A")
-        model_version = output_data.get("model_version", "N/A")
-        analyses = output_data.get("final_analysis", [])
+        # Case 1: full dict
+        if isinstance(data, dict) and "final_analysis" in data:
+            patient_id = data.get("patient_id", "Unknown Patient")
+            hcc_code = data.get("hcc_code", "N/A")
+            model_version = data.get("model_version", "N/A")
+            analyses = data.get("final_analysis", [])
+        # Case 2: list of diagnoses
+        elif isinstance(data, list):
+            patient_id = "N/A"
+            hcc_code = "N/A"
+            model_version = "N/A"
+            analyses = data
+        else:
+            return "<div style='color:red; font-weight:bold;'>⚠️ Invalid data format for report.</div>"
 
+        # Header
         md = f"""
 <div style="border:2px solid #4CAF50; padding:15px; border-radius:10px; background:#f9fdf9;">
   <h2 style="color:#2e7d32;">📋 Patient Report</h2>
@@ -38,6 +51,7 @@ def json_to_markdown(output_data: dict) -> str:
 <br/>
 """
 
+        # Each diagnosis
         for idx, diag in enumerate(analyses, 1):
             md += f"""
 <div style="border:1px solid #ccc; padding:12px; border-radius:8px; margin-bottom:12px;">