Update test.py

test.py

@@ -45,16 +45,22 @@ async def login_google():
 
 
 app.get("/test")
-async def auth_google(request: Request):
+async def auth_google(code:str):
 
-    data = await request.json()
-    code = data.get("access_token")
-    brand_name = data.get("brand_name")
-    print("Printing the access token")
+    token_url = "https://accounts.google.com/o/oauth2/token"
     print(code)
-    if not code:
+    data = {
+        "code": code,
+        "client_id": GOOGLE_CLIENT_ID,
+        "client_secret": GOOGLE_CLIENT_SECRET,
+        "redirect_uri": GOOGLE_REDIRECT_URI,
+        "grant_type": "authorization_code",
+    }
+    response = requests.post(token_url, data=data)
+    access_token = response.json().get("access_token")
+    if not access_token:
         raise HTTPException(status_code=400, detail="Authorization code not provided")
-    access_token_new = code
+    access_token_new = access_token
 
     user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token_new}"})
     page_token = None
@@ -117,8 +123,8 @@ async def auth_google(request: Request):
 
                             if data:
                                 data_new[filename]=data
-                                attachment_content = base64.urlsafe_b64decode(data)
-                                extracted_text = await extract_text_from_attachment(filename, attachment_content)
+                                # attachment_content = base64.urlsafe_b64decode(data)
+                                # extracted_text = await extract_text_from_attachment(filename, attachment_content)
 
                                 attachment_no+=1