Update test.py

test.py

@@ -1,7 +1,6 @@
 from fastapi import FastAPI, Depends , Request,WebSocket
 from fastapi.security import OAuth2PasswordBearer
 import requests
-from jose import jwt
 import webbrowser
 import base64
 import logging
@@ -40,7 +39,6 @@ async def login_google():
     #Below is the URL to prompt the user to login to his specified gmail account and also give a readonly access to his gmail
     oauth_url = f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI}&scope=openid%20profile%20email%20https://www.googleapis.com/auth/gmail.readonly&access_type=offline"
     oauth_url_hr = f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI}&scope=openid%20profile%20email%20https://www.googleapis.com/auth/gmail.readonly&access_type=offline&state=receipts"
-    webbrowser.open(oauth_url)
     return {
         "url_hr": oauth_url_hr
     }
@@ -50,6 +48,7 @@ async def auth_google(request: Request):
 
     data = await request.json()
     code = data.get("access_token")
+    brand_name = data.get("brand_name")
     print("Printing the access token")
     print(code)
     if not code:
@@ -59,11 +58,11 @@ async def auth_google(request: Request):
     user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token_new}"})
     page_token = None
     messages = []
-    jobs_query = "subject:receipt OR subject:receipts has:attachment "
+    user_query = f"subject:((receipt {brand_name}) OR (receipts {brand_name})) has:attachment "
 
     while True:
         # Construct Gmail API request with pageToken
-        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={jobs_query}"
+        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={user_query}"
         if page_token:
             gmail_url += f"&pageToken={page_token}"
 
@@ -274,9 +273,7 @@ async def test_websocket(websocket: WebSocket):
     logging.info("Closing connection")
     await websocket.close()
     
-@app.get("/token")
-async def get_token(token: str = Depends(oauth2_scheme)):
-    return jwt.decode(token, GOOGLE_CLIENT_SECRET, algorithms=["HS256"])
+