Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -71,9 +71,9 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
 
     body = extract_body_from_mail(message_data)
 
-    attachments = extract_attachments_from_mail(access_token, message_data)
+    attachments,structed_attachment_data = extract_attachments_from_mail(access_token, message_data)
     high_level_company_type = get_company_type(company_from_mail)
-    structed_attachment_data = extract_json_from_attachments(access_token , message_data)
+    # structed_attachment_data = extract_json_from_attachments(access_token , message_data)
 
 
     body_len = 0
@@ -175,14 +175,21 @@ def fetch_attachment_data(access_token: str, message_id: str, attachment_id: str
 
 def extract_attachments_from_mail(access_token: str, message_data: dict) -> List[Attachment]:
     attachments = []
+    structured_data = []
     if "payload" in message_data and "parts" in message_data["payload"]:
         for part in message_data["payload"]["parts"]:
             if "body" in part and "attachmentId" in part["body"]:
                 attachment_id = part["body"]["attachmentId"]
                 attachment_data = fetch_attachment_data(access_token, message_data["id"], attachment_id)
                 filename = part.get("filename", "untitled.txt")
+                data = attachment_data.get("data", "")
+                raw_text=ut.extract_text_from_attachment(filename , data)
+                struct_data = ut.strcuture_document_data(raw_text)
+                if struct_data:
+                    structured_data.append(struct_data)
+                
                 attachments.append(Attachment(attachment_len = len(attachment_data.get("data", "")),filename=filename, data=attachment_data.get("data", "")))
-    return attachments
+    return attachments,structured_data
 
 
 def extract_text(html_content: str) -> str: