Update test.py

test.py

@@ -4,7 +4,7 @@ import requests
 from jose import jwt
 import webbrowser
 import base64
-
+import logging
 
 app = FastAPI()
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
@@ -22,6 +22,10 @@ GOOGLE_CLIENT_ID = "485753721652-5uta3e18va2g6cnkldib2d68q39t4vod.apps.googleuse
 GOOGLE_CLIENT_SECRET = "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"
 GOOGLE_REDIRECT_URI = "https://omkar008-receipt-radar-test.hf.space/auth/google"
 GOOGLE_REDIRECT_URI_hr = "https://receiptradar-0bb387d81174.herokuapp.com/auth/google"
+# Configure the logger
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
 
 @app.get("/")
 async def login_google():
@@ -159,12 +163,16 @@ async def test_websocket(websocket: WebSocket):
     # data = await request.json()
     # code = data.get("access_token")
     sucess =await websocket.accept()
-    await print(sucess)
-    await print("Hi hi working")
+    # await print(sucess)
+    # await print("Hi hi working")
+    logger.info("Hi hi succefull in connecting !!")
     # await websocket.send_json({"message":"Yes Websockets successfull"})
-    await print("working after line 163")
+    # await print("working after line 163")
+    logger.info("Now receiving json!!")
+
     data = await websocket.receive_json()
-    print(data)
+    logger.info("Received JSON data: %s", data)
+    # print(data)
     # code = data.get("message")
     # print(code)
     # code = raw_body.decode()