AIEmailAutomationUtility 0.0.10__py3-none-any.whl → 0.0.12__py3-none-any.whl

AIEmailAutomationUtility/Email_Classification.py

@@ -7,7 +7,7 @@ from flask import Flask,request
 
 class Email_Classification:
     def detect_category_openai(self, openai_api_key, categories, email_body):
-        logger.log("Inside detect_category_openai::")
+        print("Inside detect_category_openai::")
         try:
             categories_str = ', '.join(categories)
             message = [{
@@ -27,14 +27,14 @@ class Email_Classification:
                 presence_penalty=0,
             )
             category = json.loads(result.choices[0].message.content.replace("\n```", "").replace("```", "").replace("json","").replace("JSON","").replace("csv","").replace("CSV",""))['category']
-            logger.log(f"category:: {category}")
+            print(f"category:: {category}")
             return {"status": "Success", "message": category}
         except Exception as e:
             logger.log(f"Error detecting category with OpenAI: {str(e)}")
             return {"success": "Failed", "message": f"Error detecting category with OpenAI: {str(e)}"}
 
     def detect_category_gemini(self, gemini_api_key, categories, email_body, detect_email_category=True, signature=None):
-        logger.log("Inside detect_category_gemini::")
+        print("Inside detect_category_gemini::")
         try:
             categories_str = ', '.join(categories) 
             if detect_email_category:
@@ -74,23 +74,23 @@ class Email_Classification:
                         if final_result.startswith("{{") and final_result.endswith("}}"):
                             final_result = final_result[1:-1]
                         final_result = json.loads(final_result)
-                        logger.log(f"finalResult:::  {final_result}")
+                        print(f"finalResult:::  {final_result}")
                     except json.JSONDecodeError:
-                        logger.log(f"Exception : Invalid JSON Response GEMINI 1.5: {final_result} {type(final_result)}")
+                        print(f"Exception : Invalid JSON Response GEMINI 1.5: {final_result} {type(final_result)}")
 
             if detect_email_category:
                 category = final_result.get('category', 'Others')
                 return {"status": "Success", "message": category}
             else:
-                logger.log(f"finalResult:::  {final_result}")
+                print(f"finalResult:::  {final_result}")
                 return {"status": "Success", "message": final_result}
 
         except Exception as e:
-            logger.log(f"Error with Gemini AI detection/generation: {str(e)}")
+            print(f"Error with Gemini AI detection/generation: {str(e)}")
             return {"success": "Failed", "message": f"Error with Gemini AI detection/generation: {str(e)}"}
 
     def detect_category_local(self, openai_api_key, categories, email_body, detect_email_category=True, signature=None, local_ai_url=None):
-        logger.log("Inside detect_category_local::")
+        print("Inside detect_category_local::")
         try:
             categories_str = ', '.join(categories)
             if detect_email_category:
@@ -125,17 +125,17 @@ class Email_Classification:
                     if json_start != -1 and json_end != -1:
                         json_str = final_result[json_start:json_end]
                         final_result = json.loads(json_str)
-                        logger.log(f"finalResult:::  {final_result}")
+                        print(f"finalResult:::  {final_result}")
                         category = final_result.get('category', 'Others')
-                        logger.log(f"category::1037 {category}")
+                        print(f"category::1037 {category}")
                         return {"status": "Success", "message": category}
                     else:
                         raise ValueError("No valid JSON object found in the response")
                 except json.JSONDecodeError as e:
-                    logger.log(f"JSON decode error: {e}")
+                    print(f"JSON decode error: {e}")
                     raise
             else:
-                logger.log(f"finalResult:1040  {final_result}")
+                print(f"finalResult:1040  {final_result}")
                 return {"status": "Success", "message": final_result}
 
         except Exception as e:

AIEmailAutomationUtility/Email_DocumentUploader.py

@@ -16,7 +16,7 @@ class Email_DocumentUploader:
             
         # Save file temporarily
         file_path = os.path.join(temp_dir, file_data['filename'])
-        logger.log(f"file_path:: {file_path}")
+        print(f"file_path:: {file_path}")
         with open(file_path, 'wb') as f:
             f.write(file_data['content'])
         
@@ -46,7 +46,7 @@ class Email_DocumentUploader:
                 files=files,
                 data=params
             )
-        logger.log("file read")
+        print("file read")
         
         if response.status_code == 200:
             result = json.loads(response.text)
@@ -56,7 +56,7 @@ class Email_DocumentUploader:
             return str(response.status_code), f"Upload failed: {response.text}"
 
     def email_document_upload(self, file, parameters_details):
-        logger.log(f"file ::: {file}")
+        print(f"file ::: {file}")
         if not file:
             return "file not found"
 
@@ -81,5 +81,5 @@ class Email_DocumentUploader:
 
         response_status, restAPI_Result = self.upload_document(upload_config, file_data)
         
-        logger.log(f"Upload_Document response result: {restAPI_Result}")
+        print(f"Upload_Document response result: {restAPI_Result}")
         return response_status, restAPI_Result

AIEmailAutomationUtility/Email_Draft.py

@@ -110,7 +110,7 @@ class Email_Draft:
             success, draft_message = self.draft_email_response(email_details)
             
             if success == "Success":
-                logger.log(f"draft_message  {draft_message}")
+                print(f"draft_message  {draft_message}")
                 return draft_message
 
         except Exception as e:

AIEmailAutomationUtility/Email_Read.py

@@ -99,7 +99,7 @@ class Email_Read:
 
         mail = imaplib.IMAP4_SSL(host, port)
         mail.login(reciever_email_addr, receiver_email_pwd)
-        logger.log("login successfully")
+        print("login successfully")
         mail.select('inbox')
                     
         file_JsonArray, categories = self.read_JSON_File(templateName)
@@ -112,8 +112,8 @@ class Email_Read:
                 email_ids = email_ids[0].split()
 
                 if not email_ids: 
-                    logger.log("Email not found, going to check new mail")
-                    logger.log("Email not found,\ngoing to check new mail \n")
+                    print("Email not found, going to check new mail")
+                    print("Email not found,\ngoing to check new mail \n")
                 else:
                 
                     for email_id in email_ids:
@@ -140,8 +140,8 @@ class Email_Read:
                                 email_body = msg.get_payload(decode=True).decode('utf-8', errors='replace')                              
                             
                             openai_Process_Input  = email_body 
-                            logger.log("\nEmail Subject:::", subject)
-                            logger.log("\nEmail body:::", openai_Process_Input )
+                            logger.log(f"\nEmail Subject::: {subject}")
+                            logger.log(f"\nEmail body::: {openai_Process_Input}")
 
                             openai_api_key = email_config.get('openai_api_key', '') 
                             geminiAI_APIKey = email_config.get('gemini_api_key', '') 
@@ -162,7 +162,7 @@ class Email_Read:
                                 email_classification = Email_Classification()
                                 emailCategory = email_classification.detect_category(email_cat_data)
                                 emailCategory = emailCategory['message']
-                                logger.log(f"\nDetected Email category ::: {emailCategory}")
+                                print(f"\nDetected Email category ::: {emailCategory}")
                                 
                                 if emailCategory == 'Others':
                                     logger.log(f"Marking email as UNREAD. ")
@@ -172,7 +172,7 @@ class Email_Read:
                                     mail.copy(email_id, LABEL)
                                     mail.store(email_id, '+FLAGS', '\\Deleted')  # Mark for deletion
                                     mail.expunge()                         
-                                    logger.log(f"Mail removed from inbox and added to '{LABEL}' label.")
+                                    print(f"Mail removed from inbox and added to '{LABEL}' label.")
                                 
                                 elif emailCategory == "Product Enquiry":
                                     
@@ -188,7 +188,7 @@ class Email_Read:
 
                                             email_draft = Email_Draft()
                                             status, response = email_draft.draft_email(email_config, email_details, openai_Response['message'])
-                                            logger.log(f"status ::: {status}")
+                                            print(f"status ::: {status}")
                                         else :
                                             message = f"Invalid response method received '{responseMethod}' for category : '{emailCategory}'"
                                             raise ValueError(message)
@@ -201,7 +201,7 @@ class Email_Read:
 
                                         email_draft = Email_Draft()
                                         status, response = email_draft.draft_email(email_config, email_details, LocalAI_Response)
-                                        logger.log(f"status ::: {status}")
+                                        print(f"status ::: {status}")
                                     elif Model_Name == "GeminiAI":
                                         logger.log("GeminiAI")
                                         Detect_Email_category = False
@@ -211,7 +211,7 @@ class Email_Read:
 
                                         email_draft = Email_Draft()
                                         status, response = email_draft.draft_email(email_config, email_details, GeminiAI_Response)
-                                        logger.log(f"status ::: {status}")
+                                        print(f"status ::: {status}")
                                     else:
                                         raise ValueError(f"Invalid Model Name provided : '{Model_Name}'")
 
@@ -226,18 +226,18 @@ class Email_Read:
                                             email_upload_document = Email_DocumentUploader()               
                                             with open(f"temp/{fileName}", "rb") as file:
                                                 response_status, restAPI_Result = email_upload_document.email_document_upload(file, parameters)
-                                                logger.log(f"email_upload_document_response ::: {restAPI_Result}")
+                                                print(f"email_upload_document_response ::: {restAPI_Result}")
                                         else:
                                             
                                             new_fileName = self.create_file_from_emailBody(email_body, sender_email_addr, parameters)
                                             with open(f"temp/{new_fileName}", "rb") as file:
                                                 response_status, restAPI_Result = email_upload_document.email_document_upload(file, parameters)
-                                                logger.log(f"email_upload_document_response ::: {restAPI_Result}")
+                                                print(f"email_upload_document_response ::: {restAPI_Result}")
 
                                         if response_status == "200" :
-                                            logger.log(f"Attachment uploaded sucessfully against Document id: '{restAPI_Result}'.")
+                                            print(f"Attachment uploaded sucessfully against Document id: '{restAPI_Result}'.")
                                         else:
-                                            logger.log(restAPI_Result)
+                                            print(restAPI_Result)
                                     
                                     else :
                                         message = f"Invalid response method received '{responseMethod}' for category : '{emailCategory}'"
@@ -281,6 +281,9 @@ class Email_Read:
             receiver_email_pwd = data.get("receiver_email_pwd")
             host = data.get("host")
             port = data.get("port")
+            openai_api_key = data.get("openai_api_key") 
+            geminiAI_APIKey = data.get("GeminiAI_APIKey")
+            localAIURL = data.get("LOCAL_AI_URL")
 
             if not all([reciever_email_addr, receiver_email_pwd, host, port]):
                 raise ValueError("Missing required email configuration fields.")
@@ -294,7 +297,10 @@ class Email_Read:
                 'email': reciever_email_addr,
                 'password': receiver_email_pwd,
                 'host': host,
-                'port': int(port)
+                'port': int(port),
+                'openai_api_key': openai_api_key,
+                'gemini_api_key': geminiAI_APIKey,
+                'local_ai_url': localAIURL
             }
 
             emails = self.read_email(email_config)            
@@ -370,7 +376,6 @@ class Email_Read:
         if not os.path.exists(dir):
             os.mkdir(dir)
         fileName = sender_email_addr[sender_email_addr.find("<")+1:sender_email_addr.find("@")].strip().replace(".","_")
-        logger.log(f"{fileName = }")
         
         if parameters["FILE_TYPE"] == "pdf":
             fileName = fileName + ".pdf"

{AIEmailAutomationUtility-0.0.10.dist-info → AIEmailAutomationUtility-0.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: AIEmailAutomationUtility
-Version: 0.0.10
+Version: 0.0.12
 Summary: Create 2 different methods in read_email for different client
 Author: Proteus Technology PVT. LTD.
 Author-email: <apps@baseinformation.com>

AIEmailAutomationUtility-0.0.12.dist-info/RECORD

@@ -0,0 +1,14 @@
+AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=NQlBPugGdEvNwGFOoaiJNb4JDQB_DYvCy_hWLgAVLns,3938
+AIEmailAutomationUtility/Email_Classification.py,sha256=nq958GJ5GwOF7FgMSoItp1AOD0CUUg0Dz25ERLxy1GI,9283
+AIEmailAutomationUtility/Email_DocumentUploader.py,sha256=bkDWekmhOOd7G6dAoVYYTEcerT0rGP49xlMIORU6mEw,3035
+AIEmailAutomationUtility/Email_Draft.py,sha256=uCFiu9OUWDGmUBnWv9RqElv2P43SFnRw1HB6YuQ5p_8,5239
+AIEmailAutomationUtility/Email_Read.py,sha256=qRtHY7qxQPe90JjMv27yW3bZoNtnTyuZqAl1cBB3PcQ,20732
+AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
+AIEmailAutomationUtility/Save_Draft.py,sha256=yzLgFN14I_lXE6qL0I3tKNduvcnWdbsY9i2mKdTtio4,5348
+AIEmailAutomationUtility/Save_Transaction.py,sha256=Gg1w6hhzHmEFjsuzYvkq-3-EsWReetjLHsYSv5YIGgM,3816
+AIEmailAutomationUtility/__init__.py,sha256=bB7N-qQD85RdMaXHg_ZU5oPtHtx9zgeG6-ue4rhcfYI,379
+AIEmailAutomationUtility-0.0.12.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
+AIEmailAutomationUtility-0.0.12.dist-info/METADATA,sha256=p2FYQU14fI60ODJSPRJSS2mGQ8z5jF_U80F2tTHCgTw,623
+AIEmailAutomationUtility-0.0.12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+AIEmailAutomationUtility-0.0.12.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
+AIEmailAutomationUtility-0.0.12.dist-info/RECORD,,

AIEmailAutomationUtility-0.0.10.dist-info/RECORD

@@ -1,14 +0,0 @@
-AIEmailAutomationUtility/EmailReplyAssistant.py,sha256=NQlBPugGdEvNwGFOoaiJNb4JDQB_DYvCy_hWLgAVLns,3938
-AIEmailAutomationUtility/Email_Classification.py,sha256=eL52Td50zo7V0QTIqjN4Khhg-5HKvf2RUITIcKz5yZ0,9343
-AIEmailAutomationUtility/Email_DocumentUploader.py,sha256=ImTmMz_JeU6Xynt9kyu7lFv7vqrxzqAtBF-A7014fYc,3055
-AIEmailAutomationUtility/Email_Draft.py,sha256=BfseewnnlwNl1moodq3kZiUPXDUE9a_nQjuFQsUp3fY,5244
-AIEmailAutomationUtility/Email_Read.py,sha256=bmRS_jt_v0BxsOWQZcH12VKhcWUbc9dNuw3qkvbGCN4,20516
-AIEmailAutomationUtility/Email_Upload_Document.py,sha256=3bdkxfDlwoeRp-46KPw2Gs1dqBhEIoA1yE5GCudpdV8,1320
-AIEmailAutomationUtility/Save_Draft.py,sha256=yzLgFN14I_lXE6qL0I3tKNduvcnWdbsY9i2mKdTtio4,5348
-AIEmailAutomationUtility/Save_Transaction.py,sha256=Gg1w6hhzHmEFjsuzYvkq-3-EsWReetjLHsYSv5YIGgM,3816
-AIEmailAutomationUtility/__init__.py,sha256=bB7N-qQD85RdMaXHg_ZU5oPtHtx9zgeG6-ue4rhcfYI,379
-AIEmailAutomationUtility-0.0.10.dist-info/LICENCE.txt,sha256=2qX9IkEUBx0VJp1Vh9O2dsRwE-IpYId0lXDyn7OVsJ8,1073
-AIEmailAutomationUtility-0.0.10.dist-info/METADATA,sha256=iG4GkrRFAa5-1rXSU-nXQmnMbSqkIso7QKXPF7j4S78,623
-AIEmailAutomationUtility-0.0.10.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-AIEmailAutomationUtility-0.0.10.dist-info/top_level.txt,sha256=3jTWrTUblVkaP7mpwY2UBSnrlfot5Ykpfsehyke-Uzw,25
-AIEmailAutomationUtility-0.0.10.dist-info/RECORD,,