Update app.py

app.py

@@ -1,4 +1,8 @@
 import os
+import smtplib
+import ssl
+from email.message import EmailMessage
+
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # Force TensorFlow to use CPU
 
 import gradio as gr
@@ -21,8 +25,41 @@ 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 send email
+def send_email(patient_email, report_path, patient_name):
+    sender_email = "[email protected]"
+    sender_password = "your_email_password"
+
+    subject = f"Bone Fracture Report for {patient_name}"
+    body = f"""
+    Dear {patient_name},
+
+    Your bone fracture diagnosis report is attached.
+
+    If you have any concerns, consult your doctor.
+
+    Regards,
+    Hospital Team
+    """
+
+    msg = EmailMessage()
+    msg["From"] = sender_email
+    msg["To"] = patient_email
+    msg["Subject"] = subject
+    msg.set_content(body)
+
+    # Attach PDF file
+    with open(report_path, "rb") as file:
+        msg.add_attachment(file.read(), maintype="application", subtype="pdf", filename=os.path.basename(report_path))
+
+    # Secure email sending
+    context = ssl.create_default_context()
+    with smtplib.SMTP_SSL("smtp.gmail.com", 465, context=context) as server:
+        server.login(sender_email, sender_password)
+        server.send_message(msg)
+
 # Function to process X-ray and generate a PDF report
-def generate_report(name, age, gender, weight, height, allergies, cause, xray):
+def generate_report(name, age, gender, weight, height, allergies, cause, xray, email):
     image_size = (224, 224)
 
     def predict_fracture(xray_path):
@@ -115,6 +152,9 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     c.save()
 
+    # Send the report via email
+    send_email(email, report_path, name)
+
     return report_path  # Return path for auto-download
 
 # Function to select a sample image
@@ -139,6 +179,9 @@ with gr.Blocks() as app:
         allergies = gr.Textbox(label="Allergies (if any)")
         cause = gr.Textbox(label="Cause of Injury")
 
+    with gr.Row():
+        email = gr.Textbox(label="Patient Email", type="email")
+
     with gr.Row():
         xray = gr.Image(type="filepath", label="Upload X-ray Image")
     
@@ -153,7 +196,7 @@ with gr.Blocks() as app:
 
     submit_button.click(
         generate_report,
-        inputs=[name, age, gender, weight, height, allergies, cause, xray],
+        inputs=[name, age, gender, weight, height, allergies, cause, xray, email],
         outputs=[output_file],
     )