Modify google credentials code.

chat_application/main.py

@@ -5,7 +5,10 @@ from datetime import datetime, timedelta
 import random
 import time
 import math
+import os
+import json
 import google.auth
+from google.oauth2 import service_account
 from google.auth.transport.requests import AuthorizedSession
 from vertexai.tuning import sft
 from vertexai.generative_models import GenerativeModel
@@ -29,15 +32,21 @@ app.config["SECRET_KEY"] = "supersecretkey"
 socketio = SocketIO(app)
 
 # Setup for Vertex API calls
-credentials, _ = google.auth.default(
-    scopes=["https://www.googleapis.com/auth/cloud-platform"]
+serviceAccount_json = os.environ["GOOGLE_SERVICE_ACCOUNT_JSON"]
+serviceAccount_info = json.loads(serviceAccount_json)
+
+credentials = service_account.Credentials.from_service_account_info(
+    serviceAccount_info,
+    scopes=["https://www.googleapis.com/auth/cloud-platform"],
 )
 google_session = AuthorizedSession(credentials)
+
 vertex_client = genai.Client(
         vertexai=True,
-        project="frozone-475719",
-        location="us-central1"
-        )
+        project=os.environ["GOOGLE_CLOUD_PROJECT"],
+        location=os.environ.get("VERTEX_LOCATION", "us-central1"),
+        credentials=credentials,
+)
 """
 #original lines before separating system instructions and prompts
 # Initialize the bots