gcp_document_ai / authenticate.py
Omkar008's picture
Update authenticate.py
9dfc538 verified
raw
history blame
591 Bytes
from google.oauth2 import service_account
from google.auth.transport.requests import Request
from dotenv import load_dotenv
import json
import os
load_dotenv()
def get_access_token():
# Load service account credentials from JSON file or environment variable
credentials = service_account.Credentials.from_service_account_info(
json.loads(os.getenv('ACCOUNT_CREDS')),
scopes=['https://www.googleapis.com/auth/cloud-platform']
)
# Refresh token if needed
if credentials.expired:
credentials.refresh(Request())
return credentials.token