From d73c342a26785ff5342784f720d3d35515f7ffda Mon Sep 17 00:00:00 2001
From: moonscape09 <munim.adil@outlook.com>
Date: Tue, 1 Apr 2025 03:26:58 -0400
Subject: [PATCH] Add code that refreshes google auth tokens to use trained
 model and retrieves new token to use in request auth headers.

---
 app/services/summarizer.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/services/summarizer.py b/app/services/summarizer.py
index 7c3fb6f..a87a555 100644
--- a/app/services/summarizer.py
+++ b/app/services/summarizer.py
@@ -1,6 +1,8 @@
 import os
 import google.generativeai as genai
 import requests
+import google.auth
+from google.auth.transport.requests import Request
 
 # Replace with your actual access token
 access_token = os.getenv("GCLOUD_ACCESS_TOKEN")
@@ -8,11 +10,13 @@
 project_id = os.getenv("PROJECT_ID")
 endpoint_id = os.getenv("ENDPOINT_ID")
 
-
+credentials = google.auth.default()[0]
+credentials.refresh(Request())
+# print(credentials.token)
 url = f"https://{model_region}-aiplatform.googleapis.com/v1/projects/{project_id}/locations/{model_region}/endpoints/{endpoint_id}:generateContent"
 
 headers = {
-    "Authorization": f"Bearer {access_token}",
+    "Authorization": f"Bearer {credentials.token}",
     "Content-Type": "application/json; charset=utf-8"
 }