PyPI - AIEmailAutomationUtility - Versions diffs - 0.0.31__py3-none-any.whl → 0.0.33__py3-none-any.whl - Mend

AIEmailAutomationUtility 0.0.31py3-none-any.whl → 0.0.33py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (10) hide show

AIEmailAutomationUtility/Email_DocumentUploader.py CHANGED Viewed

@@ -6,7 +6,6 @@ import loggerutility as logger
 from flask import Flask,request
 from datetime import datetime
 class Email_DocumentUploader:
     def upload_document(self, upload_config, file_data):
         document_type = upload_config['document_type']

AIEmailAutomationUtility/Email_Draft.py CHANGED Viewed

@@ -73,6 +73,7 @@ class Email_Draft:
                     "<html><body>"
                     f"<p>{body_html}</p>"
                     f"{table_html}"
+                    f"<b>GST Extra as per applicable.</b>"
                     f"<p>{signature_html}</p>"
                     "<hr>"
                     f"<p>{mail_details}</p>"

AIEmailAutomationUtility/Email_Read.py CHANGED Viewed

@@ -7,7 +7,7 @@ from datetime import datetime
 import threading
 import traceback
 import json
-import os
+import os, shutil
 import csv
 import io
@@ -112,7 +112,7 @@ class Email_Read:
             login_status = "Success"
             mail.select('inbox')
-            file_JsonArray, categories = self.read_JSON_File(templateName)
+            file_JsonArray, categories = self.read_JSON_File(templateName, user_id)
         except Exception as e:
             login_status = "Failed"
@@ -251,7 +251,7 @@ class Email_Read:
             login_status = "Success"
             mail.select('inbox')
-            file_JsonArray, categories = self.read_JSON_File(templateName)
+            file_JsonArray, categories = self.read_JSON_File(templateName, user_id)
         except Exception as e:
             logger.log(f"Login failed: {e}")
@@ -484,7 +484,7 @@ class Email_Read:
         templateName = "ai_email_automation.json"
         fileName = ""
-        file_JsonArray, categories = self.read_JSON_File(templateName)
+        file_JsonArray, categories = self.read_JSON_File(templateName, user_id)
         # Call the `extract_all_email_info` method to extract details from the eml content
         extracted_info,msg = self.extract_all_email_info(eml_content)
@@ -552,24 +552,37 @@ class Email_Read:
         return "success"
-    def read_JSON_File(self, json_fileName):
+    def read_JSON_File(self, json_fileName, user_id):
         category_list               = []
         categories                  = ""
         try:
-            if os.path.exists(json_fileName):
-                with open(json_fileName, "r") as fileObj:
-                    file_JsonArray = json.load(fileObj)
+            logger.log(f"\nEmail_Read() read_JSON_File user_id ::: {user_id}")
+            user_file = json_fileName
+            if user_id:
+                user_dir = os.path.join('user_data', user_id)
+                logger.log(f"\nEmail_Read() read_JSON_File user_dir ::: {user_dir}")
+                if not os.path.exists(user_dir):
+                    os.makedirs(user_dir, exist_ok=True)
+                user_file = os.path.join(user_dir, json_fileName)
+                if not os.path.exists(user_file) and os.path.exists(json_fileName):
+                    shutil.copy(json_fileName, user_file)
+            logger.log(f"\nEmail_Read() read_JSON_File user_file ::: {user_file}")
+            if os.path.exists(user_file):
+                with open(user_file, "r") as fileObj:
+                    file_JsonArray = json.load(fileObj)
                     for eachJson in file_JsonArray :
                         for key, value in eachJson.items():
-                            if key == "Category" and value:
+                            if key == "Category":
                                 category_list.append(value)
                         # categories = ", ".join(category_list)
                 return file_JsonArray, category_list
             else:
-                message = f"{json_fileName} file not found."
+                message = f"{user_file} file not found."
                 raise Exception(message)
         except Exception as e:
             msg = f"'{json_fileName}' file is empty. Please provide JSON parameters in the filename."
@@ -579,7 +592,7 @@ class Email_Read:
     def log_email_login(self, user_id, email, model_name, login_status):
         base_dir="EMail_log"
-        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
         log_dir = os.path.join(base_dir, user_id)
         os.makedirs(log_dir, exist_ok=True)
         log_file_path = os.path.join(log_dir, f"{user_id}.csv")

AIEmailAutomationUtility/Process_Category.py CHANGED Viewed

@@ -14,6 +14,8 @@ import openai
 import json
 import csv
 import os
+from email import message_from_string
+import re
 class Process_Category:
@@ -106,12 +108,13 @@ class Process_Category:
             logger.log(f"document_type 108::: {document_type}")
             if responseMethod == "Upload_Document":
-                # Get today's date folder path
-                today_date = datetime.today().strftime('%Y-%m-%d')
-                order_folder = os.path.join(document_type, today_date)
                 email_upload_document = Email_DocumentUploader()
                 if len(fileName) != 0 and document_type != '':
+                    # Get today's date folder path
+                    today_date = datetime.today().strftime('%Y-%m-%d')
+                    order_folder = os.path.join(document_type, today_date)
                     file_path = os.path.join(order_folder, fileName)  # Correct file path
                     with open(file_path, "rb") as file:
@@ -128,7 +131,22 @@ class Process_Category:
                     parameters["DOCUMENT_TYPE"] = document_type
                     logger.log(f"Updated Parameters ::: {parameters}")
-                    new_fileName = self.create_file_from_emailBody(email_body, sender_email_addr, parameters)
+                    email_parts = []
+                    if sender_email_addr:
+                        email_parts.append(f"From: {sender_email_addr}")
+                    if to_email_addr:
+                        email_parts.append(f"To: {to_email_addr}")
+                    if cc_email_addr:
+                        email_parts.append(f"CC: {cc_email_addr}")
+                    if subject:
+                        email_parts.append(f"Subject: {subject}")
+                    email_parts.append(email_body)
+                    email_body_with_details = "\n".join(email_parts)
+                    logger.log(f"email_body ::: {email_body}")
+                    new_fileName = self.create_file_from_emailBody(email_body_with_details, sender_email_addr, parameters)
                     new_file_path = os.path.join(order_folder, new_fileName)
                     with open(new_file_path, "rb") as file:
@@ -161,12 +179,11 @@ class Process_Category:
             # Extract customer code once
             customer_code = customer_data.get("customer_code", "")
-            # Step 5: Identify product from email using AI
+            # Step 5: Identify product from email using AI
             #If there is attachment then append the content in the body.
             extracted_content = dataValues.get('extracted_content')
-            if extracted_content != "NA":
+            logger.log(f"extracted_content is ::: {extracted_content}")
+            if extracted_content != "NA" and extracted_content != None:
                 attachment_email_body =email_body + " \n\n" + extracted_content
                 products = self.identify_products(attachment_email_body, subject, Model_Name, openai_api_key, geminiAI_APIKey, localAIURL, parameters["Product_Assistant_Id"])
             else:
@@ -366,7 +383,7 @@ class Process_Category:
             mail.expunge()
             logger.log(f"Mail removed from inbox and added to '{LABEL}' label.")
-        current_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        current_timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
         filename = f'{uid}{current_timestamp}'
         logger.log(f"The file name for csv and eml is ::: {filename}")
         self.store_email_details_to_csv(
@@ -426,7 +443,7 @@ class Process_Category:
             filename = part.get_filename()
             mime_type = part.get_content_type().lower()
-            if "pdf" in mime_type:
+            if "pdf" in mime_type or filename.lower().endswith(".pdf"):
                 document_type = "Orders"
             elif "msword" in mime_type or "wordprocessingml" in mime_type or filename.lower().endswith(".docx"):
                 document_type = "Order Excel"
@@ -438,12 +455,12 @@ class Process_Category:
                 document_type = "Order Email"
             elif "rtf" in mime_type or filename.lower().endswith(".rtf"):
                 document_type = "Order Excel"
-            today_date = datetime.today().strftime('%Y-%m-%d')  # Format: YYYY-MM-DD
-            date_folder = os.path.join(document_type, today_date)  # Path: ORDERS/YYYY-MM-DD
-            os.makedirs(date_folder, exist_ok=True)
             if filename:
+                today_date = datetime.today().strftime('%Y-%m-%d')  # Format: YYYY-MM-DD
+                date_folder = os.path.join(document_type, today_date)  # Path: ORDERS/YYYY-MM-DD
+                os.makedirs(date_folder, exist_ok=True)
                 filepath = os.path.join(date_folder, filename)  # Save inside date-wise folder
                 with open(filepath, 'wb') as f:
@@ -479,7 +496,7 @@ class Process_Category:
             fileName = fileName + ".txt"
             filePath = os.path.join(order_folder, fileName)
-            with open(filePath, "w") as file:
+            with open(filePath, "w", encoding="utf-8") as file:
                 file.write(text)
                 logger.log(f"New TXT file created from email body and stored in '{filePath}'")
         else:
@@ -525,7 +542,7 @@ class Process_Category:
         logger.log("Inside identify_products")
         if model_type == "OpenAI":
-            prompt = f"""/* Extract complete item pricing information from the following mixed-format email content. The email may contain a combination of:- descriptive product listings- tabular structured price info- semi-structured lines with HSN, quantity, material, etc. Give me price in INR and information of all items in following format requested_description, item_no, make, description, price, price unit, inventory unit strictly in JSON format. If the item is not available in the assistant files return requested description from email data and rest of the columns as NA. Do not include any instruction as the output will be directly in a program. */ /n {email_body}. """
+            prompt = f"""/* Extract complete item pricing information from the following mixed-format email content. The email may contain a combination of:- descriptive product listings- tabular structured price info- semi-structured lines with HSN, quantity, material, etc. Give me price in INR and information of all items in following format requested_description, item_no, make, description, price, Quantity, price unit, inventory unit strictly in JSON format. If the item is not available in the assistant files return requested description from email data and rest of the columns as NA. Do not include any instruction as the output will be directly in a program. */ /n {email_body}. """
             emailreplyassistant = EmailReplyAssistant()
             ai_result = emailreplyassistant.identify_customer_product_reply_assitant(openai_api_key, assistant_id, email_body, subject, prompt)
@@ -554,13 +571,31 @@ class Process_Category:
         return product_data
     def generate_quotation_draft(self, customer_data, products, model_type, openai_api_key, gemini_api_key, local_ai_url, assistant_id, email_body, subject, signature):
-        logger.log("Inside generate_quotation_draft")
         customer = customer_data
         product_table = "Products:\n"
         for product in products:
-            product_table += f'- {product.get("requested_description")} (Code: {product.get("item_no")}) = ${product.get("rate")}\n'
+            rate = product.get("rate", "NA")
+            quantity = product.get("Quantity", "NA")
+            try:
+                total = float(rate) * float(quantity)
+            except:
+                total = rate if rate != "NA" and rate not in ("", None) else "-"
+            product_table += (
+                f'- Requested Description: {product.get("requested_description", "-")}, '
+                f'Item Code: {product.get("item_no", "-")}, '
+                f'Item Description: {product.get("description", "-")}, '
+                f'Manufacturer: {product.get("make", "-")}, '
+                f'Price: {product.get("price", "-")}, '
+                f'Inventory Unit: {product.get("inventory unit", "-")}, '
+                f'Discount: {product.get("Discount", "-")}, '
+                f'Rate: {rate}, '
+                f'Quantity: {quantity}, '
+                f'Total: {total}, '
+                f'Price Pickup Source: {product.get("Price Pickup Source", "-")}, '
+                f'Availability: ""\n'
+            )
         if model_type == "OpenAI":
             prompt = f"""
@@ -570,7 +605,12 @@ class Process_Category:
                 Customer Code: {customer.get('customer_code', '')}
                 {product_table}
-                product_table must contain price column, if it is none(set it as -).
+                Ensure the table has the following columns in this exact order:
+                Sr. No., Requested Description, Item Code, Item Description, Manufacturer,  Price, Inventory Unit, Discount, Rate, Quantity, Total, Price Pickup Source, Availability
+                - If any value is missing, use a dash ("-") instead.
+                - "Total" is calculated as Rate × Quantity.
+                - "Availability" should be a blank column.
                 Original Email Subject: {subject}
                 Return only the following JSON String format:
@@ -787,6 +827,17 @@ class Process_Category:
         with the name user_id.csv.
         """
+        logger.log(f"The response am getting in csv is : {response}")
+        match = re.search(
+            r'Content-Transfer-Encoding:\s*base64\s+([\s\S]+?)\n--',
+            response,
+            re.IGNORECASE
+        )
+        if match:
+            response = self.extract_html_from_mime(response)
+        logger.log(f"The response after am getting in csv is : {response}")
         # Ensure the Mail_log folder exists
         log_folder = "Mail_log"
         os.makedirs(log_folder, exist_ok=True)
@@ -910,4 +961,18 @@ class Process_Category:
                 if user_id:
                     return user_id.value
             return None
+    def extract_html_from_mime(self, raw_data):
+        msg = message_from_string(raw_data)
+        if msg.is_multipart():
+            for part in msg.walk():
+                content_type = part.get_content_type()
+                content_encoding = part.get("Content-Transfer-Encoding", "").lower()
+                # Look for text/html part with base64 encoding
+                if content_type == "text/html" and content_encoding == "base64":
+                    payload = part.get_payload(decode=True)
+                    return payload.decode(part.get_content_charset() or 'utf-8')
+        return "No HTML content found."

{AIEmailAutomationUtility-0.0.31.dist-info → AIEmailAutomationUtility-0.0.33.dist-info}/METADATA RENAMED Viewed

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: AIEmailAutomationUtility
-Version: 0.0.31
-Summary: If attachement in email download it and pass attachement file data to product identification service
+Version: 0.0.33
+Summary: Added new columns and  Modified the draft prompt to return draft in consistent format for all mails
 Author: Proteus Technology PVT. LTD.
 Author-email: <apps@baseinformation.com>
 Keywords: python,first package
@@ -13,4 +13,4 @@ Classifier: Operating System :: MacOS :: MacOS X
 Classifier: Operating System :: Microsoft :: Windows
 License-File: LICENCE.txt
-If attachement in email download it and pass attachement file data to product identification service
+Added new columns and  Modified the draft prompt to return draft in consistent format for all mails

AIEmailAutomationUtility-0.0.33.dist-info/RECORD ADDED Viewed

@@ -0,0 +1,15 @@
+AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=R_wJna3-ITsVxQEccryhM93T_Nf_Oxo8DXnS-sDN8VE,6679
+AIEmailAutomationUtility/Email_Classification.py,sha256=Ar0g4Ff8HOT7xICktd3nP_C_vCyeY-xCpUjVCVRWAyc,9417
+AIEmailAutomationUtility/Email_DocumentUploader.py,sha256=BWNRt2X-E2HCogBaKDfl7cZZNSkZUeIsVs8iXjFjH88,3218
+AIEmailAutomationUtility/Email_Draft.py,sha256=JYZijUh_zan2asyMYQwIBwIpGNJ5SSQGma5AL1meaXk,7808
+AIEmailAutomationUtility/Email_Read.py,sha256=Ehzx9SKjzCgWY9K4rWCnwokDzo6M0Nbhi2XLKYpeEis,31451
+AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
+AIEmailAutomationUtility/Process_Category.py,sha256=9X3b2BEw31T6RmJ66g0zMP8WYVlVqi8CDGMIsR6ATbU,47741
+AIEmailAutomationUtility/Save_Draft.py,sha256=yzLgFN14I_lXE6qL0I3tKNduvcnWdbsY9i2mKdTtio4,5348
+AIEmailAutomationUtility/Save_Transaction.py,sha256=Gg1w6hhzHmEFjsuzYvkq-3-EsWReetjLHsYSv5YIGgM,3816
+AIEmailAutomationUtility/__init__.py,sha256=Jad3IdPRsVMeLqEEh-FbCrc1lE2tzJO2DTG5Hgmxh5g,391
+AIEmailAutomationUtility-0.0.33.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
+AIEmailAutomationUtility-0.0.33.dist-info/METADATA,sha256=9kdhPKSz1FqZiWMkeQq6GFK3TefmevbwWikMIem7BHw,699
+AIEmailAutomationUtility-0.0.33.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+AIEmailAutomationUtility-0.0.33.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
+AIEmailAutomationUtility-0.0.33.dist-info/RECORD,,

AIEmailAutomationUtility-0.0.31.dist-info/RECORD DELETED Viewed

@@ -1,15 +0,0 @@
-AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=R_wJna3-ITsVxQEccryhM93T_Nf_Oxo8DXnS-sDN8VE,6679
-AIEmailAutomationUtility/Email_Classification.py,sha256=Ar0g4Ff8HOT7xICktd3nP_C_vCyeY-xCpUjVCVRWAyc,9417
-AIEmailAutomationUtility/Email_DocumentUploader.py,sha256=YJu4tuTHr0K-5vuds9gZfj-Hwsgm4MuAOP39Lmu_t98,3219
-AIEmailAutomationUtility/Email_Draft.py,sha256=DcyBeDaE8CReKHnHxLiz-o2tDxuUgwy91c4k0qhQbVw,7749
-AIEmailAutomationUtility/Email_Read.py,sha256=6obif-ZLhxuatNcGhO8en_R7-Q56EOhzeyILb8vZBjM,30743
-AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
-AIEmailAutomationUtility/Process_Category.py,sha256=Rhs_flyv-_LvDzu6aO_i-ZDjnMKWDIP_mAcqf8mkl_Y,44607
-AIEmailAutomationUtility/Save_Draft.py,sha256=yzLgFN14I_lXE6qL0I3tKNduvcnWdbsY9i2mKdTtio4,5348
-AIEmailAutomationUtility/Save_Transaction.py,sha256=Gg1w6hhzHmEFjsuzYvkq-3-EsWReetjLHsYSv5YIGgM,3816
-AIEmailAutomationUtility/__init__.py,sha256=Jad3IdPRsVMeLqEEh-FbCrc1lE2tzJO2DTG5Hgmxh5g,391
-AIEmailAutomationUtility-0.0.31.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
-AIEmailAutomationUtility-0.0.31.dist-info/METADATA,sha256=E_KUzOjVXHaejQTsqUsfCZqLQYTWJQIFY4xUrZufw_o,701
-AIEmailAutomationUtility-0.0.31.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-AIEmailAutomationUtility-0.0.31.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
-AIEmailAutomationUtility-0.0.31.dist-info/RECORD,,

{AIEmailAutomationUtility-0.0.31.dist-info → AIEmailAutomationUtility-0.0.33.dist-info}/LICENCE.txt RENAMED Viewed

File without changes

{AIEmailAutomationUtility-0.0.31.dist-info → AIEmailAutomationUtility-0.0.33.dist-info}/WHEEL RENAMED Viewed

File without changes

{AIEmailAutomationUtility-0.0.31.dist-info → AIEmailAutomationUtility-0.0.33.dist-info}/top_level.txt RENAMED Viewed

File without changes

AIEmailAutomationUtility 0.0.31__py3-none-any.whl → 0.0.33__py3-none-any.whl

AIEmailAutomationUtility 0.0.31py3-none-any.whl → 0.0.33py3-none-any.whl