Update app.py

app.py

@@ -184,10 +184,11 @@ def segment_image(image: Image.Image, model: nn.Module) -> dict:
 def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Image.Image, 
                        diagnosis: str, grade: str, bright_area: float, red_area: float):
     try:
+        # Create PDF in memory
         pdf = FPDF()
         pdf.add_page()
         
-        # Use built-in helvetica font (works without file)
+        # Use built-in font that supports basic characters
         pdf.set_font("helvetica", "B", 16)
         pdf.cell(0, 10, "Diabetic Retinopathy Diagnosis Report", 0, 1, 'C')
         pdf.ln(10)
@@ -216,17 +217,15 @@ def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Im
         pdf.ln(15)
         
         # Helper function to add images
-        def add_pdf_image(pdf, image, title, max_width=180):
+        def add_pdf_image(pdf, image, title):
             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.image(io.BytesIO(img_byte_arr.getvalue()), x=10, w=180)
             pdf.ln(5)
         
-        # Add images (resize if needed)
+        # Add images
         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:")
@@ -236,9 +235,9 @@ def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Im
         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
+        # Get PDF bytes without encoding
+        pdf_output = pdf.output(dest='S')
+        return pdf_output.encode('latin1') if isinstance(pdf_output, str) else pdf_output
     
     except Exception as e:
         st.error(f"PDF generation failed: {str(e)}")