Update app.py

app.py

@@ -28,7 +28,7 @@ print(f"Loading model repo={REPO_ID} epoch={'latest' if EPOCH is None else EPOCH
 pipeline = Phase3Pipeline(
     repo_id=REPO_ID,
     epoch=EPOCH,
-    mc_passes=1,
+    mc_passes=2,
     uncertainty_threshold=0.05,
     token=TOKEN,
 )
@@ -96,70 +96,6 @@ def _build_text(r: dict, mc_passes: int) -> str:
     return "\n".join(lines)
 
 
-def _rejection_text(confidence: float, reason: str) -> str:
-    lines = [
-        "=" * 52,
-        "  GLAUCOMA CDSS — INPUT REJECTED",
-        "=" * 52,
-        "",
-        "  This image does not appear to be a retinal",
-        "  fundus photograph.",
-        "",
-        f"  Validator confidence : {confidence:.4f}",
-        f"  Validator detail     : {reason}",
-        "",
-        "  Please upload a proper retinal fundus image.",
-        "  Accepted sources:",
-        "    - Fundus camera photograph",
-        "    - JPEG or PNG format",
-        "    - Clear view of the optic disc",
-        "",
-        "=" * 52,
-    ]
-    return "\n".join(lines)
-
-
-
-# ── PDF BUILDER ───────────────────────────────────────────────────────────
-def _build_pdf(text: str):
-
-    try:
-        from fpdf import FPDF
-
-        # Remove unsupported unicode characters
-        safe_text = (
-            text.replace("—", "-")
-                .replace("•", "-")
-                .replace("→", "->")
-        )
-
-        pdf = FPDF()
-
-        pdf.add_page()
-
-        pdf.set_auto_page_break(auto=True, margin=15)
-
-        pdf.set_font("Courier", size=10)
-
-        for line in safe_text.splitlines():
-            pdf.multi_cell(0, 6, txt=line)
-
-        out = tempfile.NamedTemporaryFile(
-            delete=False,
-            suffix=".pdf"
-        )
-
-        pdf.output(out.name)
-
-        return out.name
-
-    except Exception as e:
-
-        print("PDF generation failed:", repr(e))
-
-        return None
-
-
 # ── INFERENCE ─────────────────────────────────────────────────────────────
 def analyse(file_path):
     if file_path is None:
@@ -175,13 +111,10 @@ def analyse(file_path):
         print("Pipeline inference completed.")
 
         text = _build_text(result["report"], pipeline.mc_passes)
-        pdf = _build_pdf(text)
-
         status = "Analysis completed successfully."
-        if pdf is None:
-            status = "Analysis completed, but PDF generation failed."
+        
 
-        return text, pdf, status
+        return text, status
 
     except Exception as e:
         print(f"Analysis error: {e}")
@@ -224,7 +157,6 @@ with gr.Blocks(title="Glaucoma CDSS") as demo:
         interactive=False,
     )
 
-    pdf_out = gr.File(label="Download PDF Report")
 
     gr.Markdown(
         "_Research prototype — NOT a medical device. "
@@ -327,7 +259,7 @@ All clinical decisions must be made by a qualified ophthalmologist.
     ).then(
         fn=analyse,
         inputs=[file_in],
-        outputs=[report_box, pdf_out, status_box],
+        outputs=[report_box, status_box],
     ).then(
         fn=set_ready,
         inputs=None,