Update app.py

app.py

@@ -17,10 +17,8 @@ CSV_PATH = "hcc_mapping.csv"
 SAMPLE_PDF = "sample_patient_chart.pdf"  # Place a sample PDF in the same folder
 
 
+# ---------- JSON to Markdown ----------
 def json_to_markdown(data) -> str:
-    """
-    Convert JSON (dict or list) into a clean, styled Markdown/HTML string.
-    """
     try:
         if isinstance(data, dict) and "final_analysis" in data:
             patient_id = data.get("patient_id", "Unknown Patient")
@@ -217,6 +215,15 @@ def load_sample_pdf():
     return PDFWrapper(SAMPLE_PDF)
 
 
+# ---------- Helper for PDF Preview ----------
+def pdf_to_iframe(file):
+    if file is None:
+        return "<p style='color:orange;'>No PDF uploaded.</p>"
+    return f"""
+    <iframe src="{file.name}" width="100%" height="600px" style="border:1px solid #ccc;"></iframe>
+    """
+
+
 # ---------- Gradio UI ----------
 with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
     gr.HTML(f"""
@@ -234,17 +241,17 @@ with gr.Blocks(theme=simple_theme, title=APP_TITLE) as interface:
 
     with gr.Row():
         with gr.Column(scale=1):
-            # --- CHANGE 1: Replaced gr.HTML with gr.PDF for better scrolling ---
-            pdf_preview = gr.PDF(label="📑 PDF Preview", height=600)
+            pdf_preview = gr.HTML(label="📑 PDF Preview", value="<p>Upload a PDF to preview</p>")
         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>",
             )
 
-    # --- CHANGE 2: Updated the event handler to pass the file path directly ---
-    pdf_upload.change(fn=lambda file: file.name if file else None, inputs=pdf_upload, outputs=pdf_preview)
+    # Connect PDF upload to preview
+    pdf_upload.change(fn=pdf_to_iframe, inputs=pdf_upload, outputs=pdf_preview)
 
+    # Connect run button
     run_btn.click(
         fn=process_pipeline,
         inputs=[pdf_upload, hcc_code, model_version],
@@ -264,4 +271,4 @@ if __name__ == "__main__":
     interface.queue().launch(
         server_name="0.0.0.0",
         server_port=int(os.environ.get("PORT", 7860))
-    )
+    )