AIEmailAutomationUtility 0.0.24__py3-none-any.whl → 0.0.25__py3-none-any.whl

AIEmailAutomationUtility/Email_Classification.py

@@ -18,7 +18,7 @@ class Email_Classification:
             logger.log(f"Final GPT message for detecting category::: {message}")
             client = OpenAI(api_key=openai_api_key)
             result = client.chat.completions.create(
-                model="gpt-4.1",
+                model="gpt-4.1-mini",
                 messages=message,
                 temperature=0,
                 max_tokens=1800,

AIEmailAutomationUtility/Email_DocumentUploader.py

@@ -9,11 +9,12 @@ from datetime import datetime
 
 class Email_DocumentUploader:
     def upload_document(self, upload_config, file_data):
+        document_type = upload_config['document_type']
         # try:
         logger.log("inside function" )
         # Create temp directory if needed
         today_date = datetime.today().strftime('%Y-%m-%d')
-        temp_dir = os.path.join("ORDERS", today_date)
+        temp_dir = os.path.join(document_type, today_date)
             
         # Save file temporarily
         file_path = os.path.join(temp_dir, file_data['filename'])

AIEmailAutomationUtility/Process_Category.py

@@ -8,6 +8,7 @@ import loggerutility as logger
 from datetime import datetime
 from openai import OpenAI
 from fpdf import FPDF
+import mimetypes
 import sqlite3
 import openai
 import json
@@ -102,24 +103,36 @@ class Process_Category:
             logger.log(f"parameters ::: {parameters}")
 
             # Download the attachment
-            fileName = self.download_attachment(msg)
-
-            # Get today's date folder path
-            today_date = datetime.today().strftime('%Y-%m-%d')
-            order_folder = os.path.join("ORDERS", today_date)
+            fileName, document_type = self.download_attachment(msg)
+            logger.log(f"fileName 107::: {fileName}")
+            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:
                     file_path = os.path.join(order_folder, fileName)  # Correct file path
 
                     with open(file_path, "rb") as file:
+                        parameters["DOCUMENT_TYPE"] = document_type
+                        
+                        logger.log(f"Updated Parameters ::: {parameters}")
                         response_status, restAPI_Result = email_upload_document.email_document_upload(file, parameters)
                         logger.log(f"email_upload_document_response ::: {restAPI_Result}")
                 else:
+                    document_type = "Order Email"
+                    # Get today's date folder path
+                    today_date = datetime.today().strftime('%Y-%m-%d')
+                    order_folder = os.path.join(document_type, today_date)
+
+                    parameters["DOCUMENT_TYPE"] = document_type
+                    logger.log(f"Updated Parameters ::: {parameters}")
                     new_fileName = self.create_file_from_emailBody(email_body, sender_email_addr, parameters)
                     new_file_path = os.path.join(order_folder, new_fileName)
-
+                    
                     with open(new_file_path, "rb") as file:
                         response_status, restAPI_Result = email_upload_document.email_document_upload(file, parameters)
                         logger.log(f"email_upload_document_response ::: {restAPI_Result}")
@@ -144,13 +157,13 @@ class Process_Category:
             # Step 4: Identify customer from email using AI
             customer_data = self.identify_customer(email_body, subject, Model_Name, openai_api_key, geminiAI_APIKey, localAIURL, parameters["Customer_Assistant_Id"])
             logger.log(f"Identified customer: {customer_data}")
-            logger.log(f'this is the customer data we are getting',customer_data)
+            logger.log(f'This is the customer data we are getting ::: {customer_data}')
             # 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"])
-            logger.log(f'this is the products data we are getting',products)
+            logger.log(f'this is the products data we are getting ::: {products}')
 
             db_connection = sqlite3.connect('/home/base/git/ai-email-automation/AI_Email/database/fetchprice.db')
             for product in products:
@@ -219,7 +232,7 @@ class Process_Category:
 
                     if discount_result:
                         discount_percent = discount_result[0]
-                        rate = raw_price * (1 - discount_percent / 100)
+                        rate = raw_price * (1 - int(discount_percent) / 100)
                         rate = round(rate, 2)
                         found_rate = True
                         logger.log(f"[2] Discounted rate for '{make}' ({discount_percent}%) on price {raw_price}: {rate}")
@@ -241,7 +254,7 @@ class Process_Category:
                     if past_sales_result:
                         most_common_make, past_discount = past_sales_result
                         if isinstance(raw_price, (int, float)):
-                            rate = raw_price * (1 - past_discount / 100)
+                            rate = raw_price * (1 - int(past_discount) / 100)
                             rate = round(rate, 2)
                             found_rate = True
                             logger.log(f"[3] Fallback: Most used make '{most_common_make}' for customer '{customer_code}' got {past_discount}% discount. Rate: {rate}")
@@ -262,7 +275,7 @@ class Process_Category:
 
                     if general_discount_result:
                         general_discount_percent = general_discount_result[0]
-                        rate = raw_price * (1 - general_discount_percent / 100)
+                        rate = raw_price * (1 - int(general_discount_percent) / 100)
                         rate = round(rate, 2)
                         found_rate = True
                         logger.log(f"[4] General Discount for '{make}' ({general_discount_percent}%) on price {raw_price}: {rate}")
@@ -320,10 +333,11 @@ class Process_Category:
             mail.expunge()   
             logger.log(f"Mail removed from inbox and added to '{LABEL}' label.")
 
-        filename = f"{uid}"
+        current_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        filename = f'{uid}{current_timestamp}'
         logger.log(f"the file name is  ---------------- {filename}")
         self.store_email_details_to_csv(
-            email_id=uid,
+            email_id=f'{uid}{current_timestamp}',
             to_email=parseaddr(to_email_addr)[1],
             from_email=parseaddr(sender_email_addr)[1],
             cc_email=parseaddr(cc_email_addr)[1] if cc_email_addr else '',
@@ -334,11 +348,11 @@ class Process_Category:
             filename=filename,
             user_id=user_id,
             status=csv_data_status,
-            response=csv_data_response
-
+            response=csv_data_response,
+            current_timestamp=current_timestamp
         )
         self.store_email_as_eml(
-            uid=uid,
+            uid=f'{uid}{current_timestamp}',
             from_email=sender_email_addr,
             to_email=to_email_addr,
             cc_email=cc_email_addr,
@@ -363,21 +377,37 @@ class Process_Category:
         return responseMethod, parameters
     
     def download_attachment(self, msg):
-        base_folder = "ORDERS"  # Main folder for storing orders
-        today_date = datetime.today().strftime('%Y-%m-%d')  # Format: YYYY-MM-DD
-        date_folder = os.path.join(base_folder, today_date)  # Path: ORDERS/YYYY-MM-DD
-
-        # Ensure folders exist
-        os.makedirs(date_folder, exist_ok=True)
 
         filename = ""
+        mime_type = ""
+        document_type = "Orders"
 
         for part in msg.walk():
             if part.get_content_maintype() == 'multipart':
                 continue
             if part.get('Content-Disposition') is None:
                 continue
+
             filename = part.get_filename()
+            mime_type = part.get_content_type().lower()
+
+            if "pdf" in mime_type:
+                document_type = "Orders"
+            elif "msword" in mime_type or "wordprocessingml" in mime_type or filename.endswith(".docx"):
+                document_type = "Order Excel"
+            elif "excel" in mime_type or "spreadsheetml" in mime_type or filename.endswith((".xls", ".xlsx")):
+                document_type = "Order Excel"
+            elif "csv" in mime_type or filename.endswith(".csv"):
+                document_type = "Order Excel"
+            elif "plain" in mime_type or filename.endswith(".txt"):
+                document_type = "Order Email"
+            elif "rtf" in mime_type or filename.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:
                 filepath = os.path.join(date_folder, filename)  # Save inside date-wise folder
 
@@ -386,10 +416,10 @@ class Process_Category:
                 logger.log(f"\nAttachment saved: '{filepath}'")
             else:
                 logger.log("\nNo Attachment found.")
-        return filename
+        return filename, document_type
     
     def create_file_from_emailBody(self, text, sender_email_addr, parameters):
-        base_folder = "ORDERS"
+        base_folder = parameters.get('DOCUMENT_TYPE', '')
         today_date = datetime.today().strftime('%Y-%m-%d')  # Format: YYYY-MM-DD
         order_folder = os.path.join(base_folder, today_date)
 
@@ -720,7 +750,7 @@ class Process_Category:
             logger.log(f"Error with Gemini AI detection/generation: {str(e)}")
             return {"success": "Failed", "message": f"Error with Gemini AI detection/generation: {str(e)}"}
         
-    def store_email_details_to_csv(self, email_id, to_email, from_email, cc_email, subject, body, email_type, action_performed, filename, user_id,status,response):
+    def store_email_details_to_csv(self, email_id, to_email, from_email, cc_email, subject, body, email_type, action_performed, filename, user_id,status,response,current_timestamp):
         """
         Stores the extracted email details to a CSV file inside 'Mail_log/mail_log_user_id' folder
         with the name user_id.csv.
@@ -734,9 +764,10 @@ class Process_Category:
         user_folder = os.path.join(log_folder, f"{user_id}")
         os.makedirs(user_folder, exist_ok=True)
 
+        filename = filename.lstrip()
+        logger.log(f"filename ::: [{filename}]")
         full_csv_path = os.path.join(user_folder, f"{filename}.csv")
-
-        current_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        logger.log(f"full_csv_path ::: {full_csv_path}")
 
         # Prepare the data to write
         csv_data = [{
@@ -786,6 +817,7 @@ class Process_Category:
         user_folder = os.path.join(archive_folder, f"{user_id}")
         os.makedirs(user_folder, exist_ok=True)
 
+        filename = filename.lstrip()
         # Define the full path for the CSV file (csv_filename.csv inside the mail_log_user_id folder)
         eml_file_path = os.path.join(user_folder, f"{filename}.eml")
 
@@ -820,5 +852,4 @@ class Process_Category:
                 if user_id:
                     return user_id.value
             return None
-
-   
+   

{AIEmailAutomationUtility-0.0.24.dist-info → AIEmailAutomationUtility-0.0.25.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: AIEmailAutomationUtility
-Version: 0.0.24
-Summary: Attractive UI integration
+Version: 0.0.25
+Summary: Save attachment according to doctype
 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
 
-Attractive UI integration
+Save attachment according to doctype

{AIEmailAutomationUtility-0.0.24.dist-info → AIEmailAutomationUtility-0.0.25.dist-info}/RECORD

@@ -1,15 +1,15 @@
 AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=oRobM6C1AwiRA3lQ2Sh51m-FaH2PpbUEidbkOG7O9Ww,6339
-AIEmailAutomationUtility/Email_Classification.py,sha256=QvNNj8PSsfWoByBeHcaXH9mRET2TQuYGKJZGg385dzs,9412
-AIEmailAutomationUtility/Email_DocumentUploader.py,sha256=P5P36rgDzALZEpO09aJWM0HEA-6vl48HNwz7GO6SbXc,3159
+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=UJDHdsr5RN_GG5OsOvUKbKziaw0a-kR3TT0iUv0oRmI,27795
 AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
-AIEmailAutomationUtility/Process_Category.py,sha256=z_OwAAz28G2oIFEW9Kye6tAWa4OFYOaijPbCD8azVYk,39308
+AIEmailAutomationUtility/Process_Category.py,sha256=fmcSdb-yfv1SgamNomTVFZ1m7RICRoTW7dGS4aa4wW0,41194
 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.24.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
-AIEmailAutomationUtility-0.0.24.dist-info/METADATA,sha256=TFVVAleDPTxYmpwEOIhT-pEAhoUYhFRxVDNKYC3bMWs,551
-AIEmailAutomationUtility-0.0.24.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-AIEmailAutomationUtility-0.0.24.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
-AIEmailAutomationUtility-0.0.24.dist-info/RECORD,,
+AIEmailAutomationUtility-0.0.25.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
+AIEmailAutomationUtility-0.0.25.dist-info/METADATA,sha256=wJMVGNyydAOlOVXUGd7AsB0ndMtZr_o9o5h-zrxP9fo,573
+AIEmailAutomationUtility-0.0.25.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+AIEmailAutomationUtility-0.0.25.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
+AIEmailAutomationUtility-0.0.25.dist-info/RECORD,,