Update app.py

app.py

@@ -4,17 +4,12 @@ import os.path, json
 from typing import List
 
 from google.auth.transport.requests import Request
-from google.oauth2.credentials import Credentials
-from google_auth_oauthlib.flow import InstalledAppFlow
+from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
 from langchain.schema import Document
 
 
-DOCS_TOKEN = os.getenv("GOOGLE_DOCS_TOKEN_JSON")
-DOCS_CREDENTIALS = os.getenv("GOOGLE_DOCS_CREDENTIALS_JSON")
-
-
 SCOPES = [
     "https://www.googleapis.com/auth/drive.readonly",
     "https://www.googleapis.com/auth/documents.readonly",
@@ -30,29 +25,38 @@ LOCAL_SEO_FILEID = "1uc4qH5roh6_xzv5x4osZG7nrPHpPUqRz9qPn31Y1azY"
 
 creds = None
 
-### Google Drive Authentication
+### Google Drive Authentication with Service Account(Google Cloud)
 def auth_google_docs():
-    global creds
-    if DOCS_TOKEN:
-        try:
-            token_json = json.loads(DOCS_TOKEN)
-            creds = Credentials.from_authorized_user_file(token_json, SCOPES)
-        except Exception:
-            creds = None
-
-    if not creds or not creds.valid:
-        if creds and creds.expired and creds.refresh_token: 
-            creds.refresh(Request())
-        else: 
-            if DOCS_CREDENTIALS: 
-                client_config_json = json.loads(DOCS_CREDENTIALS)
-                flow = InstalledAppFlow.run_local_server(
-                    client_config_json, SCOPES, port=0
-                )
+    service_account_info = json.loads(os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON"))
+    creds =service_account.Credentials.from_service_account_info(
+        service_account_info,
+        scopes=SCOPES
+    )
+    return creds
+
+### Previous Logic (Local Development)
+# def auth_google_docs():
+#     global creds
+#     if DOCS_TOKEN:
+#         try:
+#             token_json = json.loads(DOCS_TOKEN)
+#             creds = Credentials.from_authorized_user_file(token_json, SCOPES)
+#         except Exception:
+#             creds = None
+
+#     if not creds or not creds.valid:
+#         if creds and creds.expired and creds.refresh_token: 
+#             creds.refresh(Request())
+#         else: 
+#             if DOCS_CREDENTIALS: 
+#                 client_config_json = json.loads(DOCS_CREDENTIALS)
+#                 flow = InstalledAppFlow.run_local_server(
+#                     client_config_json, SCOPES, port=0
+#                 )
                 
-                creds = flow.run_console()
+#                 creds = flow.run_console()
 
-    return creds
+#     return creds
 
 
 def get_docs_text() -> List[Document]: