Update app.py

app.py

@@ -179,72 +179,70 @@ def segment_image(image: Image.Image, model: nn.Module) -> dict:
         'bright_area': (np.sum(final_mask == 1) / total_pixels * 100),
         'red_area': (np.sum(final_mask == 2) / total_pixels * 100)
     }
-
+    
 # ====================== PDF REPORT GENERATION ======================
 def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Image.Image, 
                        diagnosis: str, grade: str, bright_area: float, red_area: float):
-    pdf = FPDF()
-    pdf.add_page()
-
-    
-    # Title
-    pdf.set_font("Arial", 'B', 16)
-    pdf.cell(0, 10, "Diabetic Retinopathy Diagnosis Report", 0, 1, 'C')
-    pdf.ln(10)
-    
-    # Patient Info (placeholder)
-    pdf.set_font("Arial", '', 12)
-    pdf.cell(0, 10, "Patient: ___________________________", 0, 1)
-    pdf.cell(0, 10, "Date: _____________________________", 0, 1)
-    pdf.ln(10)
-    
-    # Diagnosis
-    pdf.set_font("Arial", 'B', 14)
-    pdf.cell(0, 10, "Diagnosis:", 0, 1)
-    pdf.set_font("Arial", '', 12)
-    pdf.cell(0, 10, f"Stage: {diagnosis}", 0, 1)
-    pdf.cell(0, 10, f"UK Grading: {grade}", 0, 1)
-    pdf.ln(10)
-    
-    # Lesion Analysis
-    pdf.set_font("Arial", 'B', 14)
-    pdf.cell(0, 10, "Lesion Analysis:", 0, 1)
-    pdf.set_font("Arial", '', 12)
-    pdf.cell(0, 10, f"Bright Lesions: {bright_area:.2f}%", 0, 1)
-    pdf.cell(0, 10, f"Red Lesions: {red_area:.2f}%", 0, 1)
-    pdf.cell(0, 10, f"Total Affected Area: {bright_area + red_area:.2f}%", 0, 1)
-    pdf.ln(15)
-    
-    # Convert images to bytes
-    def pil_to_bytes(img, format='PNG'):
-        img_byte_arr = io.BytesIO()
-        img.save(img_byte_arr, format=format)
-        return img_byte_arr.getvalue()
-    
-    # Original Image
-    pdf.set_font("Arial", 'B', 12)
-    pdf.cell(0, 10, "Original Retinal Image:", 0, 1)
-    pdf.image(io.BytesIO(pil_to_bytes(original_img)), x=50, w=100)
-    pdf.ln(5)
-    
-    # Segmentation Mask
-    mask_img = Image.fromarray((mask * 85).astype(np.uint8))
-    pdf.set_font("Arial", 'B', 12)
-    pdf.cell(0, 10, "Lesion Segmentation Mask:", 0, 1)
-    pdf.image(io.BytesIO(pil_to_bytes(mask_img)), x=50, w=100)
-    pdf.ln(5)
-    
-    # Overlay Image
-    pdf.set_font("Arial", 'B', 12)
-    pdf.cell(0, 10, "Lesion Overlay:", 0, 1)
-    pdf.image(io.BytesIO(pil_to_bytes(overlay)), x=50, w=100)
-    
-    # Footer
-    pdf.ln(15)
-    pdf.set_font("Arial", 'I', 10)
-    pdf.cell(0, 10, "This report was generated by DR Analysis System", 0, 1, 'C')
+    try:
+        pdf = FPDF()
+        pdf.add_page()
+        
+        # Use built-in helvetica font (works without file)
+        pdf.set_font("helvetica", "B", 16)
+        pdf.cell(0, 10, "Diabetic Retinopathy Diagnosis Report", 0, 1, 'C')
+        pdf.ln(10)
+        
+        # Patient Info
+        pdf.set_font("helvetica", "", 12)
+        pdf.cell(0, 10, "Patient: ___________________________", 0, 1)
+        pdf.cell(0, 10, "Date: _____________________________", 0, 1)
+        pdf.ln(10)
+        
+        # Diagnosis
+        pdf.set_font("helvetica", "B", 14)
+        pdf.cell(0, 10, "Diagnosis:", 0, 1)
+        pdf.set_font("helvetica", "", 12)
+        pdf.cell(0, 10, f"Stage: {diagnosis}", 0, 1)
+        pdf.cell(0, 10, f"UK Grading: {grade}", 0, 1)
+        pdf.ln(10)
+        
+        # Lesion Analysis
+        pdf.set_font("helvetica", "B", 14)
+        pdf.cell(0, 10, "Lesion Analysis:", 0, 1)
+        pdf.set_font("helvetica", "", 12)
+        pdf.cell(0, 10, f"Bright Lesions: {bright_area:.2f}%", 0, 1)
+        pdf.cell(0, 10, f"Red Lesions: {red_area:.2f}%", 0, 1)
+        pdf.cell(0, 10, f"Total Affected Area: {bright_area + red_area:.2f}%", 0, 1)
+        pdf.ln(15)
+        
+        # Helper function to add images
+        def add_pdf_image(pdf, image, title, max_width=180):
+            img_byte_arr = io.BytesIO()
+            image.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
+            
+            pdf.set_font("helvetica", "B", 12)
+            pdf.cell(0, 10, title, 0, 1)
+            pdf.image(io.BytesIO(img_byte_arr), x=10, w=max_width)
+            pdf.ln(5)
+        
+        # Add images (resize if needed)
+        add_pdf_image(pdf, original_img, "Original Retinal Image:")
+        add_pdf_image(pdf, Image.fromarray((mask * 85).astype(np.uint8)), "Lesion Segmentation Mask:")
+        add_pdf_image(pdf, overlay, "Lesion Overlay:")
+        
+        # Footer
+        pdf.ln(10)
+        pdf.set_font("helvetica", "I", 10)
+        pdf.cell(0, 10, "This report was generated by DR Analysis System", 0, 1, 'C')
+        
+        # Return PDF bytes
+        pdf_output = pdf.output(dest='S').encode('latin1')
+        return pdf_output
     
-    return pdf.output(dest='S')
+    except Exception as e:
+        st.error(f"PDF generation failed: {str(e)}")
+        return None
     
 
 # ====================== MAIN APP ======================