Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -28,6 +28,7 @@ async def get_messages(code: str,websocket:WebSocket,brand_name: Optional[str] =
     # gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token}"})
     # gmail_data = gmail_response.json()
     # messages.append(gmail_data['messages'])
+    print("Inside get messages")
     def fetch_message_wrapper(message_data):
         message_id = message_data.get("id")
         if message_id:
@@ -215,8 +216,9 @@ 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,websocket,brand_name)
+    print("websocket_main")
     await get_messages(access_token,websocket,brand_name)
-    # print("websocket_main")
+    
     # print(messages)
     # # logging.info(f"brand_name:{brand_name}")
     # await websocket.send_json({"total_messages": len(messages)})