Update app.py

app.py

@@ -26,7 +26,7 @@ condition_details = {
     "Normal": {"description": "No abnormal signs detected.", "recommendation": "Routine check-ups recommended."},
     "Pneumonia": {"description": "Lung inflammation detected, possibly infectious.", "recommendation": "Seek medical attention for treatment."},
     "Cancer": {"description": "Suspicious masses suggest cancer; further imaging needed.", "recommendation": "Consult an oncologist."},
-    "TB": {"description": "Cavitary lesions indicate tuberculosis.", "recommendation": "Immediate medical evaluation required."},
+    "TB": {"description": "Cavitary lesions or nodular opacities indicate tuberculosis.", "recommendation": "Immediate medical evaluation and anti-TB therapy required."},
     "Other": {"description": "Unclear abnormality; further investigation needed.", "recommendation": "Consult a radiologist."},
     "Coronary Artery Disease": {"description": "Narrowing of coronary arteries detected.", "recommendation": "Cardiology consultation required."},
     "Aortic Aneurysm": {"description": "Abnormal enlargement of the aorta.", "recommendation": "Vascular surgery evaluation."},
@@ -41,7 +41,7 @@ condition_details = {
     "Pulmonary Embolism": {"description": "Blockage in pulmonary arteries.", "recommendation": "Urgent medical attention."},
     "Fractures": {"description": "Bone break detected.", "recommendation": "Orthopedic evaluation."},
     "Arthritis": {"description": "Joint inflammation detected.", "recommendation": "Rheumatology consultation."},
-    "Osteoporosis": {"description": "Reduced bone density observed.", "recommendation": "Bone health specialist consultation."},
+    "Osteoporosis": {"description": "Reduced bone density observed.", "recommendation: Bone health specialist consultation."},
     "Appendicitis": {"description": "Inflammation of the appendix.", "recommendation": "Surgical evaluation."},
     "Gallstones": {"description": "Stones in the gallbladder detected.", "recommendation": "Gastroenterology consultation."},
     "Kidney Stones": {"description": "Stones in the kidneys detected.", "recommendation": "Urology consultation."},
@@ -110,7 +110,7 @@ def predict_xray(image):
         logger.error(f"Error in prediction: {e}")
         return f"Error: {str(e)}"
 
-# Enhanced PDF analysis function (No preview)
+# Enhanced PDF analysis function to extract patient details and disease
 def analyze_report(file):
     if not file or not file.name.endswith(".pdf"):
         return "Please upload a valid PDF file."
@@ -120,21 +120,74 @@ def analyze_report(file):
         text = "".join(page.get_text() for page in doc)  # Extract all text from the PDF
         doc.close()
 
-        # Improved keyword matching
+        # Initialize variables for patient details and condition
+        patient_info = {
+            "Name": "Not found",
+            "Age": "Not found",
+            "Gender": "Not found",
+            "Medical Record Number": "Not found",
+            "Date of Exam": "Not found"
+        }
         condition = "Unclear"
-        if re.search(r'\bstroke\b', text, re.IGNORECASE):
+        confidence = "Not specified"
+
+        # Extract patient details using regex patterns
+        name_match = re.search(r'Name:\s*([^\n]+)', text, re.IGNORECASE)
+        if name_match:
+            patient_info["Name"] = name_match.group(1).strip()
+
+        age_match = re.search(r'Age:\s*(\d+)', text, re.IGNORECASE)
+        if age_match:
+            patient_info["Age"] = age_match.group(1).strip()
+
+        gender_match = re.search(r'Gender:\s*(Male|Female|Other)', text, re.IGNORECASE)
+        if gender_match:
+            patient_info["Gender"] = gender_match.group(1).strip()
+
+        mrn_match = re.search(r'Medical Record Number:\s*([^\n]+)', text, re.IGNORECASE)
+        if mrn_match:
+            patient_info["Medical Record Number"] = mrn_match.group(1).strip()
+
+        date_match = re.search(r'Date of Exam:\s*([^\n]+)', text, re.IGNORECASE)
+        if date_match:
+            patient_info["Date of Exam"] = date_match.group(1).strip()
+
+        # Improved condition matching with regex
+        if re.search(r'\btuberculosis\b|\bTB\b', text, re.IGNORECASE):
+            condition = "TB"
+            # Look for percentage of lung involvement
+            percentage_match = re.search(r'lung involvement:\s*approximately\s*(\d+)%', text, re.IGNORECASE)
+            if percentage_match:
+                confidence = f"{percentage_match.group(1)}% lung involvement"
+        elif 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"
+            condition = "Fractures"
         elif re.search(r'\bpneumonia\b', text, re.IGNORECASE):
             condition = "Pneumonia"
-        # Add more keyword matching conditions as needed
+        # Add more conditions as needed from the conditions list
 
-        # Return only the identified condition without preview
-        disease_status = f"Disease Identified: {condition}"
-        return f"{disease_status}"
+        # Fetch condition details
+        info = condition_details.get(condition, condition_details["Other"])
+
+        # Construct summary output
+        summary = f"""
+        <div style="font-family:Arial">
+            <h3>Summary</h3>
+            <p><b>Disease Identified:</b> {condition}</p>
+            <p><b>Cause/Status:</b> {info['description']} {f'({confidence})' if confidence != 'Not specified' else ''}</p>
+            <p><b>Treatment/Recommendation:</b> {info['recommendation']}</p>
+            <h4>Patient Details</h4>
+            <p><b>Name:</b> {patient_info['Name']}</p>
+            <p><b>Age:</b> {patient_info['Age']}</p>
+            <p><b>Gender:</b> {patient_info['Gender']}</p>
+            <p><b>Medical Record Number:</b> {patient_info['Medical Record Number']}</p>
+            <p><b>Date of Exam:</b> {patient_info['Date of Exam']}</p>
+        </div>
+        """
+        return summary
 
     except Exception as e:
         logger.error(f"Failed to process PDF: {e}")
@@ -155,7 +208,7 @@ def create_interface():
 
             with gr.TabItem("Report Analysis"):
                 pdf_input = gr.File(label="Upload PDF Report", file_types=[".pdf"])
-                summary_output_report = gr.Textbox(label="Summary Result", lines=5)
+                summary_output_report = gr.HTML(label="Summary Result")  # Changed to HTML for consistent formatting
                 with gr.Row():
                     gr.Button("Analyze Report", elem_id="analyze_button", scale=0.3).click(analyze_report, inputs=pdf_input, outputs=summary_output_report)
                     gr.Button("Clear", elem_id="clear_button", scale=0.3).click(lambda: [None, ""], inputs=None, outputs=[pdf_input, summary_output_report])
@@ -164,4 +217,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(server_port=7860, ssr_mode=False)
+    demo.launch(server_port=7860, ssr_mode=False)