Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -28,7 +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")
+    logging.info("Inside get messages")
     def fetch_message_wrapper(message_data):
         message_id = message_data.get("id")
         if message_id:
@@ -216,7 +216,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,websocket,brand_name)
-    print("websocket_main")
+    logging.info("websocket_main")
     await get_messages(access_token,websocket,brand_name)
     
     # print(messages)