AIEmailAutomationUtility 0.0.29__py3-none-any.whl → 0.0.31__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)
@@ -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

@@ -161,13 +161,22 @@ 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
 
-            db_connection = sqlite3.connect('/home/base/git/ai-email-automation/AI_Email/database/fetchprice.db')
+            db_connection = sqlite3.connect('database/fetchprice.db')
             for product in products:
                 item_no = product.get("item_no", "").strip()
                 make = product.get("make", "").strip()
@@ -207,6 +216,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 +237,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 +253,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 +278,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 +302,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 +314,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()
 

{AIEmailAutomationUtility-0.0.29.dist-info → AIEmailAutomationUtility-0.0.31.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: AIEmailAutomationUtility
-Version: 0.0.29
-Summary: Set Bydefault document type as blank to avoid garbage data while uploading
+Version: 0.0.31
+Summary: If attachement in email download it and pass attachement file data to product identification service
 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
 
-Set Bydefault document type as blank to avoid garbage data while uploading
+If attachement in email download it and pass attachement file data to product identification service

{AIEmailAutomationUtility-0.0.29.dist-info → AIEmailAutomationUtility-0.0.31.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=6obif-ZLhxuatNcGhO8en_R7-Q56EOhzeyILb8vZBjM,30743
 AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
-AIEmailAutomationUtility/Process_Category.py,sha256=PsOsZJ3ZVI6-h4YrvZZNl1MEXbV8u2OOnaOxWJafKOM,43789
+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.29.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
-AIEmailAutomationUtility-0.0.29.dist-info/METADATA,sha256=mnluYbYahxvbznRwtqHnW4lDPhCl56lvNqgylhFluDU,649
-AIEmailAutomationUtility-0.0.29.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-AIEmailAutomationUtility-0.0.29.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
-AIEmailAutomationUtility-0.0.29.dist-info/RECORD,,
+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,,