Update get_gmail_data.py

get_gmail_data.py

@@ -53,7 +53,7 @@ class GmailDataExtractor:
         access_token = self.__jwt
         print("access token")
         print(access_token)
-        receipt_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases)&maxResults=3'
+        receipt_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases)&maxResults=15'
         if self.__user_input is not None:
             receipt_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases)  AND subject:{self.__user_input}'
         gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={receipt_query}&maxResults=3"
@@ -148,6 +148,8 @@ class GmailDataExtractor:
             for header in headers:
                 if header['name'] == 'Subject':
                     subject = header['value']
+                        
+                    
         body = ''
         text=''
         links = []
@@ -165,6 +167,10 @@ class GmailDataExtractor:
                 if fromdata['name'] == 'From':
                     company_from_gmail = self.extract_domain_from_email(fromdata['value'])
                     break
+            if 'chanel' in subject.lower():
+                company_from_gmail = 'chanel'
+            if 'louis vuitton' in subject.lower():
+                company_from_gmail = 'Louis Vuitton'
 
             for part in parts:
                 if 'mimeType' not in part: