Update app.py

app.py

@@ -248,56 +248,61 @@ def document_data_tool(question):
 
 
 def send_email_with_attachment(recipient_email, subject, body, image_data):
-    sender_email = os.getenv("EMAIL_SENDER")
-    sender_password = os.getenv("EMAIL_PASSWORD")
-
-    # Create a multipart message
-    msg = MIMEMultipart()
-    msg['From'] = sender_email
-    msg['To'] = recipient_email
-    msg['Subject'] = subject
-
-    # Attach the body with the msg instance
-    msg.attach(MIMEText(body, 'plain'))
-    """
-    # Open the file to be sent
-    attachment = open(attachment_path, "rb")
-    print("Attached the image")
-    # Instance of MIMEBase and named as p
-    part = MIMEBase('application', 'octet-stream')
-
-    # To change the payload into encoded form
-    part.set_payload((attachment).read())
-
-    # Encode into base64
-    encoders.encode_base64(part)
-
-    part.add_header('Content-Disposition', f"attachment; filename= {attachment_path}")
-
-    # Attach the instance 'part' to instance 'msg'
-    msg.attach(part)"""
-    # Create a MIMEBase instance for the attachment
-    part = MIMEBase('application', 'octet-stream')
-
-    # Set the payload with the image data
-    part.set_payload(image_data)
-
-    # Encode the payload into base64
-    encoders.encode_base64(part)
-
-    # Add a header with the filename (you can set any filename you prefer)
-    part.add_header('Content-Disposition', 'attachment; filename="image.png"')
-
-    # Attach the instance 'part' to the instance 'msg'
-    msg.attach(part)
-
-    # Create SMTP session for sending the mail
-    server = smtplib.SMTP('smtp.gmail.com', 587)
-    server.starttls()
-    server.login(sender_email, sender_password)
-    text = msg.as_string()
-    server.sendmail(sender_email, recipient_email, text)
-    server.quit()
+    try:
+         sender_email = os.getenv("EMAIL_SENDER")
+        sender_password = os.getenv("EMAIL_PASSWORD")
+    
+        # Create a multipart message
+        msg = MIMEMultipart()
+        msg['From'] = sender_email
+        msg['To'] = recipient_email
+        msg['Subject'] = subject
+    
+        # Attach the body with the msg instance
+        msg.attach(MIMEText(body, 'plain'))
+        """
+        # Open the file to be sent
+        attachment = open(attachment_path, "rb")
+        print("Attached the image")
+        # Instance of MIMEBase and named as p
+        part = MIMEBase('application', 'octet-stream')
+    
+        # To change the payload into encoded form
+        part.set_payload((attachment).read())
+    
+        # Encode into base64
+        encoders.encode_base64(part)
+    
+        part.add_header('Content-Disposition', f"attachment; filename= {attachment_path}")
+    
+        # Attach the instance 'part' to instance 'msg'
+        msg.attach(part)"""
+        # Create a MIMEBase instance for the attachment
+        part = MIMEBase('application', 'octet-stream')
+    
+        # Set the payload with the image data
+        part.set_payload(image_data)
+    
+        # Encode the payload into base64
+        encoders.encode_base64(part)
+    
+        # Add a header with the filename (you can set any filename you prefer)
+        part.add_header('Content-Disposition', 'attachment; filename="image.png"')
+    
+        # Attach the instance 'part' to the instance 'msg'
+        msg.attach(part)
+    
+        # Create SMTP session for sending the mail
+        server = smtplib.SMTP('smtp.gmail.com', 587)
+        server.starttls()
+        server.login(sender_email, sender_password)
+        text = msg.as_string()
+        server.sendmail(sender_email, recipient_email, text)
+        server.quit()
+            
+    except error:
+        print error
+   
     # return 1