Update app.py

app.py

@@ -5,6 +5,7 @@ from torchvision import models, transforms
 import fitz  # PyMuPDF
 import logging
 import os
+import re  # For better pattern matching
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -109,28 +110,34 @@ def predict_xray(image):
         logger.error(f"Error in prediction: {e}")
         return f"Error: {str(e)}"
 
-# Analyze PDF report
+# Enhanced PDF analysis function
 def analyze_report(file):
     if not file or not file.name.endswith(".pdf"):
         return "Please upload a valid PDF file."
+    
     try:
         doc = fitz.open(file.name)
-        text = "".join(page.get_text() for page in doc)
+        text = "".join(page.get_text() for page in doc)  # Extract all text from the PDF
         doc.close()
 
-        condition, disease, status = "Unclear", "Unknown", "Pending"
-
-        if "stroke" in text.lower():
-            condition, disease, status = "Stroke", "Brain Disorder", "Urgent Care Needed"
-        elif "cancer" in text.lower():
-            condition, disease, status = "Cancer", "Malignant Growth", "Consult Oncologist"
-        elif "fracture" in text.lower():
-            condition, disease, status = "Fracture", "Bone Injury", "Orthopedic Attention Required"
-
+        # Improved keyword matching
+        condition = "Unclear"
+        if re.search(r'\bstroke\b', text, re.IGNORECASE):
+            condition = "Stroke"
+        elif re.search(r'\bcancer\b', text, re.IGNORECASE):
+            condition = "Cancer"
+        elif re.search(r'\bfracture\b', text, re.IGNORECASE):
+            condition = "Fracture"
+        elif re.search(r'\bpneumonia\b', text, re.IGNORECASE):
+            condition = "Pneumonia"
+        # Add more keyword matching conditions as needed
+
+        disease_status = f"Disease Identified: {condition}"
         preview = text[:300] + "..." if text else "No readable content."
-        return f"Condition: {condition}\nDisease: {disease}\nStatus: {status}\n\nPreview:\n{preview}"
+        return f"{disease_status}\n\nPreview:\n{preview}"
 
     except Exception as e:
+        logger.error(f"Failed to process PDF: {e}")
         return f"Failed to process PDF: {str(e)}"
 
 # Gradio interface