Omkar008 commited on
Commit
03c7264
1 Parent(s): 07b8e82

Update controllers/ws_controller.py

Browse files
Files changed (1) hide show
  1. controllers/ws_controller.py +6 -4
controllers/ws_controller.py CHANGED
@@ -267,14 +267,16 @@ def extract_attachments_from_mail(access_token: str, message_data: dict) -> List
267
  data = attachment_data.get("data", "")
268
  try:
269
  raw_text = ut.extract_text_from_attachment(filename, data)
270
- struct_data = ut.strcuture_document_data(raw_text)
271
- if struct_data:
272
- structured_data.append(struct_data)
273
  except Exception as e:
274
  print(f"Error processing attachment {filename}: {str(e)}")
275
  continue
276
- if struct_data:
 
277
  structured_data.append(struct_data)
 
 
 
278
 
279
  attachments.append(Attachment(attachment_len = len(attachment_data.get("data", "")),filename=filename, data=attachment_data.get("data", "")))
280
  return attachments,structured_data
 
267
  data = attachment_data.get("data", "")
268
  try:
269
  raw_text = ut.extract_text_from_attachment(filename, data)
270
+
 
 
271
  except Exception as e:
272
  print(f"Error processing attachment {filename}: {str(e)}")
273
  continue
274
+ struct_data = ut.strcuture_document_data(raw_text)
275
+ if struct_data is not None or if struct_data :
276
  structured_data.append(struct_data)
277
+ else:
278
+ struct_data = None
279
+
280
 
281
  attachments.append(Attachment(attachment_len = len(attachment_data.get("data", "")),filename=filename, data=attachment_data.get("data", "")))
282
  return attachments,structured_data