Update app.py

app.py

@@ -10,6 +10,8 @@ from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib import colors
 from reportlab.platypus import Table, TableStyle
+import smtplib
+from email.message import EmailMessage
 
 # Load the trained model
 model = tf.keras.models.load_model("my_keras_model.h5")
@@ -21,8 +23,39 @@ 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"))]
 
+# Create a folder for reports
+REPORTS_DIR = "reports"
+os.makedirs(REPORTS_DIR, exist_ok=True)
+
+# Email Configuration
+SENDER_EMAIL = "[email protected]"  # Change this
+SENDER_PASSWORD = "your-app-password"  # Use an App Password if using Gmail
+
+# Function to send email with PDF attachment
+def send_email_with_attachment(to_email, file_path, patient_name):
+    msg = EmailMessage()
+    msg["Subject"] = f"Bone Fracture Report for {patient_name}"
+    msg["From"] = SENDER_EMAIL
+    msg["To"] = to_email
+    msg.set_content(f"Dear {patient_name},\n\nAttached is your bone fracture detection report.\n\nThank you!")
+
+    # Attach PDF
+    with open(file_path, "rb") as f:
+        file_data = f.read()
+        file_name = os.path.basename(file_path)
+        msg.add_attachment(file_data, maintype="application", subtype="pdf", filename=file_name)
+
+    # Send Email
+    try:
+        with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+            server.login(SENDER_EMAIL, SENDER_PASSWORD)
+            server.send_message(msg)
+        print(f"✅ Email sent to {to_email}")
+    except Exception as e:
+        print(f"❌ Failed to send email: {e}")
+
 # 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):
@@ -34,7 +67,7 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     # Predict fracture
     prediction = predict_fracture(xray)
-    diagnosed_class = "normal" if prediction > 0.5 else "Fractured"
+    diagnosed_class = "Normal" if prediction > 0.5 else "Fractured"
 
     # Injury severity classification
     severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
@@ -56,11 +89,11 @@ 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 = os.path.join(REPORTS_DIR, f"{name}_xray.png")
     img.save(img_path)
 
     # Generate PDF report
-    report_path = f"{name}_fracture_report.pdf"
+    report_path = os.path.join(REPORTS_DIR, f"{name}_fracture_report.pdf")
     c = canvas.Canvas(report_path, pagesize=letter)
     
     # Report title
@@ -115,15 +148,14 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     c.save()
 
-    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
+    # Send Email with the Report
+    send_email_with_attachment(email, report_path, name)
+    
+    return report_path  # Return path for download
 
 # 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():
@@ -138,25 +170,19 @@ with gr.Blocks() as app:
     with gr.Row():
         allergies = gr.Textbox(label="Allergies (if any)")
         cause = gr.Textbox(label="Cause of Injury")
+        email = gr.Textbox(label="Patient Email")  # New Email Input Field
 
     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")
 
     submit_button = gr.Button("Generate Report")
     output_file = gr.File(label="Download Report")
 
-    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],
+        inputs=[name, age, gender, weight, height, allergies, cause, xray, email],
         outputs=[output_file],
     )
 
-# Launch the Gradio app
 if __name__ == "__main__":
     app.launch()