Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # Force TensorFlow to use CPU
-
+import smtplib
+import mimetypes
+from email.message import EmailMessage
 import gradio as gr
 import tensorflow as tf
 import numpy as np
@@ -18,7 +19,13 @@ model = tf.keras.models.load_model("my_keras_model.h5")
 sample_images = [f"samples/{img}" for img in os.listdir("samples") if img.endswith((".png", ".jpg", ".jpeg"))]
 
 # Function to process X-ray and generate a PDF report
-def generate_report(name, age, gender, weight, height, address, parent_name, allergies, cause, xray):
+def generate_report(name, age, gender, weight, height, address, parents, allergies, cause, email, xray):
+    # Input validation
+    name = name[:50]  
+    address = address[:100]  
+    parents = parents[:50]  
+    cause = ' '.join(cause.split()[:100])  
+
     image_size = (224, 224)
 
     def predict_fracture(xray_path):
@@ -28,53 +35,49 @@ def generate_report(name, age, gender, weight, height, address, parent_name, all
         prediction = model.predict(img_array)[0][0]
         return prediction
 
-    # Predict fracture
     prediction = predict_fracture(xray)
     diagnosed_class = "Normal" if prediction > 0.5 else "Fractured"
     severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
 
-    # Hospital details
-    hospital_name = "City Care Hospital"
-    hospital_address = "123 Health Street, MedCity, India"
-    doctor_name = "Dr. Anil Sharma (Orthopedic Specialist)"
+    treatment_data = [
+        ["Severity Level", "Recommended Treatment", "Recovery Duration"],
+        ["Mild", "Rest, pain relievers, follow-up X-ray", "4-6 weeks"],
+        ["Moderate", "Plaster cast, minor surgery if needed", "6-10 weeks"],
+        ["Severe", "Major surgery, metal implants, physiotherapy", "Several months"]
+    ]
+
+    cost_duration_data = [
+        ["Hospital Type", "Estimated Cost", "Recovery Time"],
+        ["Government", f"₹{2000 if severity == 'Mild' else 8000 if severity == 'Moderate' else 20000} - ₹{5000 if severity == 'Mild' else 15000 if severity == 'Moderate' else 50000}", "4-12 weeks"],
+        ["Private", f"₹{10000 if severity == 'Mild' else 30000 if severity == 'Moderate' else 100000}+", "6 weeks - Several months"]
+    ]
 
-    # Save X-ray image for report
     img = Image.open(xray).resize((300, 300))
     img_path = f"{name}_xray.png"
     img.save(img_path)
 
-    # Generate PDF report
     report_path = f"{name}_fracture_report.pdf"
     c = canvas.Canvas(report_path, pagesize=letter)
-
-    # Set page margins
-    c.translate(20, 20)
-
-    # Report title
+    
     c.setFont("Helvetica-Bold", 16)
-    c.drawString(180, 750, hospital_name)
-    c.setFont("Helvetica", 12)
-    c.drawString(140, 735, hospital_address)
-    c.drawString(180, 720, f"Attending Doctor: {doctor_name}")
+    c.drawString(200, 770, "Bone Fracture Detection Report")
 
-    # Patient details
     patient_data = [
-        ["Patient Name", name[:50]],
+        ["Patient Name", name],
         ["Age", age],
         ["Gender", gender],
-        ["Parent's Name", parent_name[:50]],
-        ["Address", address[:70]],
         ["Weight", f"{weight} kg"],
         ["Height", f"{height} cm"],
-        ["Allergies", allergies[:50] if allergies else "None"],
-        ["Cause of Injury", cause[:50] if cause else "Not Provided"],
+        ["Address", address],
+        ["Parent's Name", parents],
+        ["Allergies", allergies if allergies else "None"],
+        ["Cause of Injury", cause],
         ["Diagnosis", diagnosed_class],
         ["Injury Severity", severity]
     ]
 
-    # Format and align tables
     def format_table(data):
-        table = Table(data, colWidths=[200, 350])
+        table = Table(data, colWidths=[270, 270])
         table.setStyle(TableStyle([
             ('BACKGROUND', (0, 0), (-1, 0), colors.darkblue),
             ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
@@ -86,83 +89,80 @@ def generate_report(name, age, gender, weight, height, address, parent_name, all
         ]))
         return table
 
-    # Draw patient details table
     patient_table = format_table(patient_data)
-    patient_table.wrapOn(c, 450, 500)
+    patient_table.wrapOn(c, 480, 500)
     patient_table.drawOn(c, 50, 620)
 
-    # Load and insert X-ray image
-    c.drawInlineImage(img_path, 50, 350, width=250, height=250)
+    c.drawInlineImage(img_path, 170, 320, width=250, height=250)
     c.setFont("Helvetica-Bold", 12)
-    c.drawString(120, 320, f"Fractured: {'Yes' if diagnosed_class == 'Fractured' else 'No'}")
+    c.drawString(250, 290, f"Fractured: {'Yes' if diagnosed_class == 'Fractured' else 'No'}")
+
+    treatment_table = format_table(treatment_data)
+    treatment_table.wrapOn(c, 480, 200)
+    treatment_table.drawOn(c, 50, 200)
 
-    # Injury details
-    c.setFont("Helvetica-Bold", 14)
-    c.drawString(50, 270, "Injury Details and Treatment Recommendations")
-    c.setFont("Helvetica", 12)
-    c.drawString(50, 250, "• Immobilization and pain management")
-    c.drawString(50, 235, "• Follow-up X-rays required")
-    c.drawString(50, 220, "• Surgical intervention if needed")
-    c.drawString(50, 205, "• Physiotherapy for recovery")
+    cost_table = format_table(cost_duration_data)
+    cost_table.wrapOn(c, 480, 150)
+    cost_table.drawOn(c, 50, 80)
 
     c.save()
 
-    return report_path  # Return path for auto-download
+    send_email(email, report_path)  
+
+    return report_path  
+
+# Function to send email
+def send_email(receiver_email, attachment_path):
+    sender_email = "[email protected]"  
+    sender_password = "yourpassword"  
+
+    msg = EmailMessage()
+    msg["Subject"] = "Your Bone Fracture Report"
+    msg["From"] = sender_email
+    msg["To"] = receiver_email
+    msg.set_content("Please find attached your bone fracture diagnosis report.")
+
+    mime_type, _ = mimetypes.guess_type(attachment_path)
+    mime_type = mime_type or "application/octet-stream"
+
+    with open(attachment_path, "rb") as attachment:
+        msg.add_attachment(attachment.read(), maintype=mime_type.split("/")[0], subtype=mime_type.split("/")[1], filename=os.path.basename(attachment_path))
+
+    with smtplib.SMTP_SSL("smtp.example.com", 465) as server:
+        server.login(sender_email, sender_password)
+        server.send_message(msg)
 
 # Function to select a sample image
 def use_sample_image(sample_image_path):
-    return sample_image_path  # Returns selected sample image filepath
+    return sample_image_path  
 
 # Define Gradio Interface
 with gr.Blocks() as app:
-    gr.Markdown("## **Bone Fracture Detection System**")
+    gr.Markdown("## Bone Fracture Detection System")
     
-    # Informative Blog Section
-    with gr.Accordion("Bone Fractures - Symptoms, Causes, & Treatment", open=True):
-        gr.Markdown("""
-        **A fracture** is a break or crack in a bone caused by excessive force.  
-        **Common Causes:**  
-        - Traumatic injuries (sports, accidents, falls)  
-        - Osteoporosis or cancer (weakened bones)  
-
-        **Symptoms:**  
-        - Severe pain, swelling, bruising  
-        - Deformity or inability to use the limb  
-
-        **Diagnosis:**  
-        - X-rays, CT scans, MRI scans  
-
-        **Treatment:**  
-        - Plaster casts, splints, surgery if needed  
-        - Pain management and physiotherapy  
-
-        **First Aid:**  
-        - Immobilize the area  
-        - Apply a cold pack  
-        - Seek medical help immediately  
-        """)
-
-    # Patient Details Form
     with gr.Row():
-        name = gr.Textbox(label="Patient Name", max_length=50)
+        name = gr.Textbox(label="Patient Name (Max 50 chars)", max_chars=50)
         age = gr.Number(label="Age")
         gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
-
-    with gr.Row():
-        parent_name = gr.Textbox(label="Parent's Name", max_length=50)
-        address = gr.Textbox(label="Address", max_length=70)
-
+    
     with gr.Row():
         weight = gr.Number(label="Weight (kg)")
         height = gr.Number(label="Height (cm)")
-
+    
     with gr.Row():
-        allergies = gr.Textbox(label="Allergies (if any, max 50 chars)", max_length=50)
-        cause = gr.Textbox(label="Cause of Injury (max 50 chars)", max_length=50)
+        address = gr.Textbox(label="Address (Max 100 chars)", max_chars=100)
+        parents = gr.Textbox(label="Parent's Name (Max 50 chars)", max_chars=50)
+    
+    with gr.Row():
+        allergies = gr.Textbox(label="Allergies (if any)")
+        cause = gr.Textbox(label="Cause of Injury (Max 100 words)")
 
+    with gr.Row():
+        email = gr.Textbox(label="Patient's Email (To receive report)", type="email")
+    
     with gr.Row():
         xray = gr.Image(type="filepath", label="Upload X-ray Image")
-
+    
     with gr.Row():
         sample_selector = gr.Dropdown(choices=sample_images, label="Use Sample Image")
         select_button = gr.Button("Load Sample Image")
@@ -174,10 +174,9 @@ with gr.Blocks() as app:
 
     submit_button.click(
         generate_report,
-        inputs=[name, age, gender, weight, height, address, parent_name, allergies, cause, xray],
+        inputs=[name, age, gender, weight, height, address, parents, allergies, cause, email, xray],
         outputs=[output_file],
     )
 
-# Launch the Gradio app
 if __name__ == "__main__":
     app.launch()