Update app.py

app.py

@@ -17,6 +17,9 @@ from reportlab.lib.units import inch
 from datetime import datetime
 import logging
 from flask_mail import Mail, Message
+import base64
+from sendgrid import SendGridAPIClient
+from sendgrid.helpers.mail import (Mail, Attachment, FileContent, FileName, FileType, Disposition)
 
 app = Flask(__name__)
 app.secret_key = "e3f6f40bb8b2471b9f07c4025d845be9"
@@ -152,24 +155,45 @@ def preprocess_image(image_bytes):
     image_array = tf.keras.utils.img_to_array(image)
     return np.expand_dims(image_array, axis=0) / 255.0
 
-# NEW: Function to send email in a background thread
+# REPLACE your old function with this new one
 def send_email_async(app_context, report_data):
     with app_context:
         try:
+            # 1. Create the PDF report
             pdf_path = f"/tmp/report_{report_data['scan_id']}.pdf"
             generate_pdf(report_data, pdf_path)
+
+            # 2. Create the email message
+            message = Mail(
+                from_email='your-verified-email@gmail.com',  # IMPORTANT: Use the email you verified on SendGrid
+                to_emails=report_data['email'],
+                subject='Your SnapSkin Diagnostic Report',
+                html_content=f"<strong>Dear {report_data['name']},</strong><p>Please find your diagnostic report from SnapSkin attached.</p>"
+            )
+
+            # 3. Read the PDF and attach it to the email
+            with open(pdf_path, 'rb') as f:
+                data = f.read()
+            encoded_file = base64.b64encode(data).decode()
+            attachedFile = Attachment(
+                FileContent(encoded_file),
+                FileName(f"SnapSkin_Report_{report_data['scan_id']}.pdf"),
+                FileType('application/pdf'),
+                Disposition('attachment')
+            )
+            message.attachment = attachedFile
+
+            # 4. Use the API key from secrets to send the email
+            sg = SendGridAPIClient(os.environ.get('SENDGRID_API_KEY'))
+            response = sg.send(message)
             
-            msg = Message('Your SnapSkin Diagnostic Report', sender=app.config['MAIL_USERNAME'], recipients=[report_data['email']])
-            msg.body = f"Dear {report_data['name']},\n\nPlease find your diagnostic report attached."
-            
-            with app.open_resource(pdf_path) as fp:
-                msg.attach(f"SnapSkin_Report_{report_data['scan_id']}.pdf", "application/pdf", fp.read())
-            
-            mail.send(msg)
+            # 5. Clean up the created PDF file
             os.remove(pdf_path)
-            logger.info(f"Report sent successfully to {report_data['email']}")
+            
+            logger.info(f"Report sent via SendGrid, status code: {response.status_code}")
+
         except Exception as e:
-            logger.error(f"Failed to send email in background: {e}")
+            logger.error(f"Failed to send email via SendGrid: {e}")
 
 def generate_pdf(report, filepath):
     c = canvas.Canvas(filepath, pagesize=A4)