Update app.py

app.py

@@ -1,7 +1,8 @@
-import gradio as gr
+=import gradio as gr
 from PIL import Image
 import torch
 from torchvision import models, transforms
+import fitz  # PyMuPDF
 import logging
 import os
 
@@ -108,10 +109,34 @@ def predict_xray(image):
         logger.error(f"Error in prediction: {e}")
         return f"Error: {str(e)}"
 
+# Analyze PDF report
+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)
+        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"
+
+        preview = text[:300] + "..." if text else "No readable content."
+        return f"Condition: {condition}\nDisease: {disease}\nStatus: {status}\n\nPreview:\n{preview}"
+
+    except Exception as e:
+        return f"Failed to process PDF: {str(e)}"
+
 # Gradio interface
 def create_interface():
     with gr.Blocks() as demo:
-        gr.Markdown("<h1 style='text-align:center;'>🩻 RadiologyScan AI</h1><p style='text-align:center;'>AI-powered X-ray Analysis</p>")
+        gr.Markdown("<h1 style='text-align:center;'>🩻 RadiologyScan AI</h1><p style='text-align:center;'>AI-powered X-ray and Report Analysis</p>")
 
         with gr.Tabs():
             with gr.TabItem("X-ray Analysis"):
@@ -121,6 +146,13 @@ def create_interface():
                     gr.Button("Analyze X-ray", elem_id="analyze_button", scale=0.3).click(predict_xray, inputs=img_input, outputs=summary_output)
                     gr.Button("Clear", elem_id="clear_button", scale=0.3).click(lambda: [None, ""], inputs=None, outputs=[img_input, summary_output])
 
+            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)
+                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])
+
         return demo
 
 if __name__ == "__main__":