Update get_gmail_data.py

get_gmail_data.py

@@ -159,9 +159,9 @@ class GmailDataExtractor:
 
                     if data:
                         # Save only the first 10 characters of the attachment data
-                        return subject,body , links , {filename:data}
+                        return subject,body ,{"filename":filename , "base64URLData":data}
 
-        return subject, body, links , None
+        return subject, body,None
     
     def extract_text_and_links(html_content: str) -> tuple:
         """
@@ -200,7 +200,7 @@ class GmailDataExtractor:
 
         Returns:
             dict: A dictionary containing the extracted messages with their subjects, bodies, links, and attachment statuses.
-            format:{"results":[{"subjec":"test subject" , "body":"it would be text" , "attachment_data":{"filename":base64URL format}},{second message with same content of subject , body , attachment_data}]}
+            format:{"results":[{"subjec":"test subject" , "body":"it would be text" , "attachment_data":{"filename":filename,"base64URLData":data}},{second message with same content of subject , body , attachment_data}]}
 
         """
         messages = self.__fetch_messages()
@@ -213,6 +213,9 @@ class GmailDataExtractor:
             body = body if body is not None else ""
             attachment_data = attachment_data if attachment_data is not None else {}
 
-            results.append({"subject": subject, "body": body, "attachment_data": attachment_data})
+            if attachment_data:  # Check if attachment_data is present
+                results.append({"attachment_data": attachment_data})
+            elif body:  # If attachment_data is absent, append body only if it's not empty
+                results.append({"body": body})
 
         return {"results": results}