Update app.py

app.py

@@ -1,9 +1,12 @@
 import os
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # Force TensorFlow to use CPU
-
+import smtplib
 import gradio as gr
 import tensorflow as tf
 import numpy as np
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+from email.mime.base import MIMEBase
+from email import encoders
 from tensorflow.keras.preprocessing import image
 from PIL import Image
 from reportlab.lib.pagesizes import letter
@@ -11,6 +14,10 @@ from reportlab.pdfgen import canvas
 from reportlab.lib import colors
 from reportlab.platypus import Table, TableStyle
 
+# Ensure the "reports" directory exists
+if not os.path.exists("reports"):
+    os.makedirs("reports")
+
 # Load the trained model
 model = tf.keras.models.load_model("my_keras_model.h5")
 
@@ -21,7 +28,7 @@ with open("templates/re.html", "r", encoding="utf-8") as file:
 # List of sample images
 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
+# Function to generate and save the report
 def generate_report(name, age, gender, weight, height, allergies, cause, xray):
     image_size = (224, 224)
 
@@ -56,13 +63,13 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     # Save X-ray image for report
     img = Image.open(xray).resize((300, 300))
-    img_path = f"{name}_xray.png"
+    img_path = f"reports/{name}_xray.png"
     img.save(img_path)
 
     # Generate PDF report
-    report_path = f"{name}_fracture_report.pdf"
+    report_path = f"reports/{name}_fracture_report.pdf"
     c = canvas.Canvas(report_path, pagesize=letter)
-    
+
     # Report title
     c.setFont("Helvetica-Bold", 16)
     c.drawString(200, 770, "Bone Fracture Detection Report")
@@ -82,7 +89,7 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     # Format and align tables
     def format_table(data):
-        table = Table(data, colWidths=[270, 270])  # Set 90% width
+        table = Table(data, colWidths=[270, 270])
         table.setStyle(TableStyle([
             ('BACKGROUND', (0, 0), (-1, 0), colors.darkblue),
             ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
@@ -94,17 +101,15 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
         ]))
         return table
 
-    # Draw patient details table
+    # Draw tables and images
     patient_table = format_table(patient_data)
     patient_table.wrapOn(c, 480, 500)
     patient_table.drawOn(c, 50, 620)
 
-    # Load and insert X-ray image
     c.drawInlineImage(img_path, 50, 320, width=250, height=250)
     c.setFont("Helvetica-Bold", 12)
     c.drawString(120, 290, f"Fractured: {'Yes' if diagnosed_class == 'Fractured' else 'No'}")
 
-    # Draw treatment and cost tables
     treatment_table = format_table(treatment_data)
     treatment_table.wrapOn(c, 480, 200)
     treatment_table.drawOn(c, 50, 200)
@@ -117,46 +122,63 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     return report_path  # Return path for auto-download
 
-# Function to select a sample image
-def use_sample_image(sample_image_path):
-    return sample_image_path  # Returns selected sample image filepath
+# Function to send email with attachment
+def send_email(patient_email, report_path):
+    sender_email = "[email protected]"
+    sender_password = "your_email_password"
+    subject = "Bone Fracture Detection Report"
+    body = "Attached is your bone fracture detection report."
+
+    msg = MIMEMultipart()
+    msg["From"] = sender_email
+    msg["To"] = patient_email
+    msg["Subject"] = subject
+    msg.attach(MIMEText(body, "plain"))
+
+    with open(report_path, "rb") as attachment:
+        part = MIMEBase("application", "octet-stream")
+        part.set_payload(attachment.read())
+        encoders.encode_base64(part)
+        part.add_header("Content-Disposition", f"attachment; filename={os.path.basename(report_path)}")
+        msg.attach(part)
+
+    try:
+        server = smtplib.SMTP("smtp.gmail.com", 587)
+        server.starttls()
+        server.login(sender_email, sender_password)
+        server.sendmail(sender_email, patient_email, msg.as_string())
+        server.quit()
+        return "Email Sent Successfully!"
+    except Exception as e:
+        return f"Error sending email: {str(e)}"
 
 # Define Gradio Interface
 with gr.Blocks() as app:
-    gr.HTML(html_content)  # Display `re.html` content in Gradio
+    gr.HTML(html_content)
     gr.Markdown("## Bone Fracture Detection System")
     
     with gr.Row():
         name = gr.Textbox(label="Patient Name")
         age = gr.Number(label="Age")
         gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
-    
+        patient_email = gr.Textbox(label="Patient Email")
+
     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)")
-        cause = gr.Textbox(label="Cause of Injury")
 
     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")
+        allergies = gr.Textbox(label="Allergies")
+        cause = gr.Textbox(label="Cause of Injury")
 
-    submit_button = gr.Button("Generate Report")
+    xray = gr.Image(type="filepath", label="Upload X-ray Image")
+    generate_button = gr.Button("Generate & Download Report")
+    send_email_button = gr.Button("Send Report via Email")
     output_file = gr.File(label="Download Report")
+    status = gr.Textbox(label="Email Status", interactive=False)
 
-    select_button.click(use_sample_image, inputs=[sample_selector], outputs=[xray])
-
-    submit_button.click(
-        generate_report,
-        inputs=[name, age, gender, weight, height, allergies, cause, xray],
-        outputs=[output_file],
-    )
+    generate_button.click(generate_report, inputs=[name, age, gender, weight, height, allergies, cause, xray], outputs=[output_file])
+    send_email_button.click(send_email, inputs=[patient_email, output_file], outputs=[status])
 
-# Launch the Gradio app
 if __name__ == "__main__":
     app.launch()