Update test.py

test.py

@@ -37,14 +37,12 @@ async def login_google():
     oauth_url_hr = f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI_hr}&scope=openid%20profile%20email%20https://www.googleapis.com/auth/gmail.readonly&access_type=offline&state=receipts"
     webbrowser.open(oauth_url)
     return {
-        "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&access_type=offline",
         "url_hr": oauth_url_hr
     }
 
 @app.post("/auth/google")
 async def auth_google(request: Request):
-    #This code is basically the authorization code and this authorization code helps us to get the access token with the required scopes that we have set .
-    #We require the gmail.readonly scopes that requires verification of our application and all.
+
     data = await request.json()
     code = data.get("access_token")
     print("Printing the access token")
@@ -64,10 +62,8 @@ async def auth_google(request: Request):
     # response = requests.post(token_url, data=data)
     # access_token = response.json().get("access_token")
     access_token_new = code
-    # print(response.json())
-    # print(access_token_new)
+
     user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token_new}"})
-    # user_query = data.get("user_query")
     page_token = None
     messages = []
     jobs_query = "subject:receipt OR subject:receipts has:attachment"
@@ -91,18 +87,6 @@ async def auth_google(request: Request):
         else:
             break  # No more pages, exit the loop
 
-
-    # print(len(messages))
-
-    
-    # query = "subject:receipt OR subject:receipts has:attachment"
-    # gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={query}"
-    # gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token_new}"})
-    # messages = gmail_response.json().get("messages", [])
-    # print(messages)
-    # print("Printing gmail response")
-    # print(gmail_response.json())
-    # Fetch attachments from the first message
     attachments = []
     attachment_no = 0
     data_new = {}
@@ -131,13 +115,7 @@ async def auth_google(request: Request):
 
                             if data:
                                 data_new[filename]=data
-                                # data_new.append({filename:data.encode("UTF-8")})
-                                # Decode base64-encoded attachment data
-                                # print(filename)
                                 attachment_content = base64.urlsafe_b64decode(data.encode("UTF-8"))
-                                # print(attachment_content)
-                                print("yo")
-                                # print(data_new)
                                 attachment_no+=1
 
                             # if data:
@@ -151,8 +129,6 @@ async def auth_google(request: Request):
 
                             #     attachments.append(save_path)
 
-    # print(attachments)
-    # print(len(data_new))
     return {"attachment_count":attachment_no,"attachment_content":data_new}
 
 @app.websocket("/ws")
@@ -323,9 +299,5 @@ async def test_websocket(websocket: WebSocket):
 async def get_token(token: str = Depends(oauth2_scheme)):
     return jwt.decode(token, GOOGLE_CLIENT_SECRET, algorithms=["HS256"])
 
-# if __name__ == "__main__":
-#     import uvicorn
-
-#     uvicorn.run(app, host="0.0.0.0", port=8000)