Update controllers/ws_controller_1.py

controllers/ws_controller_1.py

@@ -213,7 +213,13 @@ def process_attachment(filename: str, data: str):
 
 async def process_message(session: aiohttp.ClientSession, access_token: str, message_data: dict):
     subject = extract_subject_from_mail(message_data)
-    company_from_mail = extract_domain_name(message_data['payload']['headers'], subject)
+    if 'payload' in message_data and 'headers' in message_data['payload']:
+        company_from_mail = extract_domain_name(message_data['payload']['headers'], subject)
+    else:
+        # If 'payload' or 'headers' is missing, use a default value or extract from another field
+        company_from_mail = "unknown"  # or some other default value
+        logging.warning(f"Payload or headers missing for message ID: {message_data.get('id', 'unknown')}")
+
     body = extract_body_from_mail(message_data)
     
     attachments = []