Update app.py

app.py

@@ -91,7 +91,7 @@ class UNet(nn.Module):
 
 # ====================== CLASSIFIER ======================
 def create_classifier_model():
-    model = models.resnet152(pretrained=False)
+    model = models.resnet152(weights=None)  # Modern syntax
     num_ftrs = model.fc.in_features
     model.fc = nn.Sequential(
         nn.Linear(num_ftrs, 512),
@@ -188,32 +188,32 @@ def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Im
         pdf = FPDF()
         pdf.add_page()
         
-        # Modern FPDF2 syntax
+        # Use built-in font that supports basic characters
         pdf.set_font("helvetica", "B", 16)
-        pdf.cell(txt="Diabetic Retinopathy Diagnosis Report", new_x="LMARGIN", new_y="NEXT", align='C')
+        pdf.cell(text="Diabetic Retinopathy Diagnosis Report", new_x="LMARGIN", new_y="NEXT", align='C')
         pdf.ln(10)
         
         # Patient Info
         pdf.set_font("helvetica", "", 12)
-        pdf.cell(txt="Patient: ___________________________", new_x="LMARGIN", new_y="NEXT")
-        pdf.cell(txt="Date: _____________________________", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text="Patient: ___________________________", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text="Date: _____________________________", new_x="LMARGIN", new_y="NEXT")
         pdf.ln(10)
         
         # Diagnosis
         pdf.set_font("helvetica", "B", 14)
-        pdf.cell(txt="Diagnosis:", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text="Diagnosis:", new_x="LMARGIN", new_y="NEXT")
         pdf.set_font("helvetica", "", 12)
-        pdf.cell(txt=f"Stage: {diagnosis}", new_x="LMARGIN", new_y="NEXT")
-        pdf.cell(txt=f"Grading: {grade}", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text=f"Stage: {diagnosis}", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text=f"UK Grading: {grade}", new_x="LMARGIN", new_y="NEXT")
         pdf.ln(10)
         
         # Lesion Analysis
         pdf.set_font("helvetica", "B", 14)
-        pdf.cell(txt="Lesion Analysis:", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text="Lesion Analysis:", new_x="LMARGIN", new_y="NEXT")
         pdf.set_font("helvetica", "", 12)
-        pdf.cell(txt=f"Bright Lesions: {bright_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
-        pdf.cell(txt=f"Red Lesions: {red_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
-        pdf.cell(txt=f"Total Affected Area: {bright_area + red_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text=f"Bright Lesions: {bright_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text=f"Red Lesions: {red_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
+        pdf.cell(text=f"Total Affected Area: {bright_area + red_area:.2f}%", new_x="LMARGIN", new_y="NEXT")
         pdf.ln(15)
         
         # Helper function to add images
@@ -223,7 +223,7 @@ def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Im
             img_bytes = img_byte_arr.getvalue()
             
             pdf.set_font("helvetica", "B", 12)
-            pdf.cell(txt=title, new_x="LMARGIN", new_y="NEXT")
+            pdf.cell(text=title, new_x="LMARGIN", new_y="NEXT")
             pdf.image(io.BytesIO(img_bytes), x=10, w=180)
             pdf.ln(5)
         
@@ -235,22 +235,26 @@ def generate_pdf_report(original_img: Image.Image, mask: np.ndarray, overlay: Im
         # Footer
         pdf.ln(10)
         pdf.set_font("helvetica", "I", 10)
-        pdf.cell(txt="This report was generated by DR Analysis System", new_x="LMARGIN", new_y="NEXT", align='C')
+        pdf.cell(text="This report was generated by DR Analysis System", new_x="LMARGIN", new_y="NEXT", align='C')
         
-         # Get PDF as bytes
-        return pdf.output()
+        # Get PDF as bytes
+        pdf_output = pdf.output()
+        if isinstance(pdf_output, str):
+            return pdf_output.encode('latin1')
+        return pdf_output
     
     except Exception as e:
         st.error(f"PDF generation failed: {str(e)}")
         return None
-    
 
 # ====================== MAIN APP ======================
 def main():
     st.set_page_config(layout="wide")
     st.title("Diabetic Retinopathy Analysis")
     
-    uploaded_file = st.file_uploader("", type=["jpg", "jpeg", "png"])
+    uploaded_file = st.file_uploader("Upload retinal scan image", 
+                               type=["jpg", "jpeg", "png"], 
+                               label_visibility="visible")
     if not uploaded_file:
         st.info("Please upload an image")
         return