Update app.py

app.py

@@ -247,7 +247,7 @@ def document_data_tool(question):
     return query_response
 
 
-def send_email_with_attachment(recipient_email, subject, body, attachment_path):
+def send_email_with_attachment(recipient_email, subject, body, image_data):
     sender_email = os.getenv("EMAIL_SENDER")
     sender_password = os.getenv("EMAIL_PASSWORD")
 
@@ -259,7 +259,7 @@ def send_email_with_attachment(recipient_email, subject, body, attachment_path):
 
     # 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")
@@ -275,6 +275,20 @@ def send_email_with_attachment(recipient_email, subject, body, attachment_path):
     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
@@ -544,8 +558,9 @@ def answer_question(user_question, chatbot, audio=None):
                 recipient_email=user_email,
                 subject="Warehouse Inventory Report",
                 body=response.get("output", "").split(":")[0],
+                image_data = img_str
                 # attachment_path=chart_path
-                attachment_path=os.getenv("IMAGE_PATH")
+                #attachment_path=os.getenv("IMAGE_PATH")
             )
 
         if "send email to" in user_question: