resume-tailor / src /google_docs_api_processing.py
NvkAnirudh's picture
Update src/google_docs_api_processing.py
0e907e6 verified
from google.oauth2.credentials import Credentials
from google.auth.transport.requests import Request
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build
import os
# SCOPES for Google Docs API
SCOPES = ['https://www.googleapis.com/auth/documents']
# Authenticate and create Google Docs API service
def google_docs_auth():
creds = None
# Check if token.json exists (for existing OAuth tokens)
if os.path.exists('token.json'):
creds = Credentials.from_authorized_user_file('token.json', SCOPES)
# If no valid credentials, go through the flow to get new ones
if not creds or not creds.valid:
if creds and creds.expired and creds.refresh_token:
creds.refresh(Request())
else:
flow = InstalledAppFlow.from_client_secrets_file(
'credentials.json', SCOPES
)
# creds = flow.run_local_server(port=0)
creds = flow.run_console()
# Save the credentials for the next run
with open('token.json', 'w') as token:
token.write(creds.to_json())
return build('docs', 'v1', credentials=creds)
# Functions to create a Google doc with formatted text
def create_google_doc(service, text):
# Request body for creating a new document
doc = {
'title': 'Updated Resume'
}
doc = service.documents().create(body=doc).execute()
# Document ID
doc_id = doc.get('documentId')
# Insert text into the document with:
# - Times New Roman
# - Font size of 10
requests = [
{
'insertText': {
'location': {
'index': 1,
},
'text': text
}
},
{
'updateTextStyle': {
'range': {
'startIndex': 1,
'endIndex': len(text) + 1
},
'textStyle': {
'fontSize': {
'magnitude': 10,
'unit': 'PT'
},
'weightedFontFamily': {
'fontFamily': 'Times New Roman'
}
},
'fields': 'fontSize,weightedFontFamily'
}
}
]
# Send the requests to the Google Docs API
service.documents().batchUpdate(documentId=doc_id, body={'requests': requests}).execute()
# Return the document URL
return f'https://docs.google.com/document/d/{doc_id}/edit'