AIEmailAutomationUtility 0.0.30__py3-none-any.whl → 0.0.32__py3-none-any.whl

AIEmailAutomationUtility/Email_Read.py

@@ -10,6 +10,11 @@ import json
 import os
 import csv
 
+import io
+import pandas as pd
+import docx
+import PyPDF2
+
 from .Save_Transaction import Save_Transaction
 from .Email_Classification import Email_Classification
 from .EmailReplyAssistant import EmailReplyAssistant
@@ -110,7 +115,9 @@ class Email_Read:
             file_JsonArray, categories = self.read_JSON_File(templateName)
 
         except Exception as e:
+            login_status = "Failed"
             logger.log(f"Login failed: {e}")
+            raise Exception(e)   
 
         # Log the result
         self.log_email_login(user_id, reciever_email_addr, Model_Name, login_status)
@@ -248,6 +255,7 @@ class Email_Read:
 
         except Exception as e:
             logger.log(f"Login failed: {e}")
+            return f"Login failed: {e}"
 
         # Log the result
         self.log_email_login(user_id, reciever_email_addr, Model_Name, login_status)
@@ -294,7 +302,23 @@ class Email_Read:
 
                                     if content_type == "text/plain":
                                         email_body += part.get_payload(decode=True).decode('utf-8', errors='replace')
-                                    
+
+        
+                            # For attachment
+                            if msg.is_multipart():
+                                for part in msg.walk():
+                                    content_disposition = str(part.get("Content-Disposition") or "")
+                                    content_type = part.get_content_type()
+                                    if "attachment" in content_disposition.lower():
+                                        filename = part.get_filename() or "attachment"
+                                        content_bytes = part.get_payload(decode=True)
+                                        if content_bytes:
+                                            extracted_content = self.Extract_attachment_content(filename, content_bytes)
+                                            extracted_content =f"\n\n--- The content of the attachment '{filename}' is below ---\n{extracted_content}\n"
+                                            # email_body += f"\n\n--- The content of the attachment '{filename}' is below ---\n{extracted_content}\n"
+                                    else:
+                                        extracted_content="NA"
+                             
                             else:
                                 email_body = msg.get_payload(decode=True).decode('utf-8', errors='replace')                              
                                 content_type = msg.get_content_type()
@@ -348,7 +372,8 @@ class Email_Read:
                                     "uid": uid,
                                     "to_email_addr": to_email_addr,
                                     "user_id": user_id,
-                                    "is_html": is_html
+                                    "is_html": is_html,
+                                    "extracted_content" : extracted_content
                                 }
                                 processcategory = Process_Category()
                                 processcategory.process_cat(emailCategory, dataValues)
@@ -554,7 +579,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")
@@ -573,5 +598,41 @@ class Email_Read:
     def read_status(self):
         global shared_status
         return shared_status
+    
+    def Extract_attachment_content(self,filename, content_bytes):
+        extension = os.path.splitext(filename)[1].lower()
+
+        try:
+            if extension == ".pdf":
+                # Extract text from PDF
+                reader = PyPDF2.PdfReader(io.BytesIO(content_bytes))
+                text = "\n".join(page.extract_text() or "" for page in reader.pages)
+                return text if text.strip() else "[No text content found in PDF]"
+
+            elif extension in [".docx", ".doc"]:
+                # Extract text from Word document
+                doc = docx.Document(io.BytesIO(content_bytes))
+                text = "\n".join([para.text for para in doc.paragraphs])
+                return text if text.strip() else "[No text content found in Word document]"
+
+            elif extension in [".xls", ".xlsx"]:
+                # Extract text from Excel sheet
+                df = pd.read_excel(io.BytesIO(content_bytes), engine='openpyxl')
+                return df.to_string(index=False)
+
+            elif extension == ".csv":
+                # Extract text from CSV file
+                df = pd.read_csv(io.BytesIO(content_bytes))
+                return df.to_string(index=False)
+
+            elif extension in [".txt"]:
+                # Decode plain text file
+                return content_bytes.decode('utf-8', errors='replace')
+
+            else:
+                return "[Unsupported attachment type]"
+
+        except Exception as e:
+            return f"[Error extracting content: {str(e)}]"
 
    

AIEmailAutomationUtility/Process_Category.py

@@ -106,12 +106,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:
@@ -161,8 +162,17 @@ class Process_Category:
             # Extract customer code once
             customer_code = customer_data.get("customer_code", "")
 
+
             # Step 5: Identify product from email using AI
-            products = self.identify_products(email_body, subject, Model_Name, openai_api_key, geminiAI_APIKey, localAIURL, parameters["Product_Assistant_Id"])
+            
+            #If there is attachment then append the content in the body.
+            extracted_content = dataValues.get('extracted_content')
+            if extracted_content != "NA":
+                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:
+                products = self.identify_products(email_body, subject, Model_Name, openai_api_key, geminiAI_APIKey, localAIURL, parameters["Product_Assistant_Id"])
+            
             logger.log(f'Identified Products are ::: {products}')
 
             product_determination=products
@@ -207,6 +217,8 @@ class Process_Category:
                 else:
                     logger.log(f"Process_Category - Quotation [0] No base price found for item '{item_no}' .")
                     product["rate"] = "NA"
+                    product["Price Pickup Source"]="NA" 
+                    product["Discount"]="NA" 
                     continue
 
                 # Condition 1: Exact match in special_rate_customer_wise
@@ -226,6 +238,7 @@ class Process_Category:
                     found_rate = True
                     logger.log(f"Process_Category - Quotation [1] Special Rate for item '{item_no}' and customer '{customer_code}' is {rate}")
                     price_pickup_source="SPECIAL_RATE_CUSTOMER_WISE"
+                    Discount="NA"
 
                 # Condition 2: Customer + Manufacturer discount
                 if not found_rate:
@@ -241,6 +254,7 @@ class Process_Category:
 
                     if discount_result:
                         discount_percent = discount_result[0]
+                        Discount= discount_percent
                         rate = raw_price * (1 - int(discount_percent) / 100)
                         rate = round(rate, 2)
                         found_rate = True
@@ -265,6 +279,7 @@ class Process_Category:
 
                     if past_sales_result:
                         most_common_make, past_discount = past_sales_result
+                        Discount=past_discount
                         if isinstance(raw_price, (int, float)):
                             rate = raw_price * (1 - int(past_discount) / 100)
                             rate = round(rate, 2)
@@ -288,6 +303,7 @@ class Process_Category:
 
                     if general_discount_result:
                         general_discount_percent = general_discount_result[0]
+                        Discount=general_discount_percent
                         rate = raw_price * (1 - int(general_discount_percent) / 100)
                         rate = round(rate, 2)
                         found_rate = True
@@ -299,9 +315,11 @@ class Process_Category:
                     rate = raw_price
                     logger.log(f"Process_Category - Quotation [5] No discounts applied. Using base price for item '{item_no}': {rate}")
                     price_pickup_source="PRICE_LIST"
+                    Discount = "NA"
 
                 product["rate"] = rate
                 product["Price Pickup Source"]=price_pickup_source 
+                product["Discount"]=Discount 
 
             db_connection.close()
 
@@ -349,7 +367,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(
@@ -409,7 +427,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"
@@ -421,12 +439,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:
@@ -462,7 +480,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:

{AIEmailAutomationUtility-0.0.30.dist-info → AIEmailAutomationUtility-0.0.32.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: AIEmailAutomationUtility
-Version: 0.0.30
-Summary: Get database access from local path for any system.
+Version: 0.0.32
+Summary: Make changes in code, should run for windows and ubuntu
 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
 
-Get database access from local path for any system.
+Make changes in code, should run for windows and ubuntu

{AIEmailAutomationUtility-0.0.30.dist-info → AIEmailAutomationUtility-0.0.32.dist-info}/RECORD

@@ -2,14 +2,14 @@ AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=R_wJna3-ITsVxQEccryhM93T_
 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=pRqc9295r_qqdfWIFJFG2XH9J-VlvO1E0yP3iv_cRV4,27802
+AIEmailAutomationUtility/Email_Read.py,sha256=zE2dK4h0XfKWuSuMJs3E2U-yMPHzRtBfNSdP74h_pDI,30743
 AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
-AIEmailAutomationUtility/Process_Category.py,sha256=zijeYAIy5fj3hXmRQ_1CEPuikw-ELNolxuaEdyJj5OY,43745
+AIEmailAutomationUtility/Process_Category.py,sha256=xXiXYf_dmscKO6scFOMRCWLgKhsV8RsCVK1AXPnE1DI,44695
 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.30.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
-AIEmailAutomationUtility-0.0.30.dist-info/METADATA,sha256=pzhpoUozUqOinZaKi0dn4qEi4md36AQa4lGD3wqJq94,603
-AIEmailAutomationUtility-0.0.30.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-AIEmailAutomationUtility-0.0.30.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
-AIEmailAutomationUtility-0.0.30.dist-info/RECORD,,
+AIEmailAutomationUtility-0.0.32.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
+AIEmailAutomationUtility-0.0.32.dist-info/METADATA,sha256=m0b6nXm2W1DSYwxU9Mbt8oPd7ujQ90BDc8I9ohdK3ck,611
+AIEmailAutomationUtility-0.0.32.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+AIEmailAutomationUtility-0.0.32.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
+AIEmailAutomationUtility-0.0.32.dist-info/RECORD,,