Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from PIL import Image
 import torch
 from torchvision import models, transforms
-import PyPDF2
+import fitz  # PyMuPDF for better PDF parsing
 import logging
 import os
 
@@ -101,7 +101,7 @@ def predict_xray(image):
         logger.error(f"Error in predict_xray: {str(e)}")
         return f"Error: {str(e)}", "", ""
 
-# Enhanced function to analyze patient reports (PDFs)
+# Enhanced function to analyze patient reports (PDFs) using PyMuPDF (fitz)
 def analyze_report(file):
     text = ""
     patient_condition = "Unclear"
@@ -110,10 +110,11 @@ def analyze_report(file):
 
     if file and file.name.endswith(".pdf"):
         try:
-            pdf_reader = PyPDF2.PdfReader(file)
-            for page in pdf_reader.pages:
-                text += page.extract_text() or ""
-            
+            # Open the PDF using PyMuPDF
+            pdf_reader = fitz.open(file.name)
+            for page in pdf_reader:
+                text += page.get_text("text")  # Extract text from each page
+
             # Example: Let's search for conditions in the text
             if "stroke" in text.lower():
                 patient_condition = "Stroke"