Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -11,7 +11,7 @@ from fastapi import WebSocket
 
 def get_messages(code: str,brand_name: Optional[str] = None) -> List[Message]:
     access_token = code
-    g_query = 'subject:"your order" OR subject:receipts OR subject:receipt OR  subject:aankoopbon  OR subject:reçu OR subject:invoice OR subject:invoices OR category:purchases'
+    g_query = 'subject:"your order" OR subject:receipts OR subject:receipt OR  subject:aankoopbon OR subject:reçu OR subject:invoice OR subject:invoices OR category:purchases'
     if brand_name is not None:
         g_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR  subject: aankoopbon  OR subject:reçu OR subject:invoice OR subject:invoices OR category:purchases OR from:{brand_name}) AND subject:{brand_name}'    
     page_token = None
@@ -184,6 +184,7 @@ def extract_text(html_content: str) -> str:
 async def websocket_main(code: str,  websocket: WebSocket,brand_name: Optional[str] = None):
     access_token = code
     messages = get_messages(access_token,brand_name)
+    logging.info(f"brand_name:{brand_name}")
     await websocket.send_json({"total_messages": len(messages)})
     print("Total Length of messages")
     print(len(messages))