Update app.py

app.py

@@ -11,8 +11,8 @@ from googleapiclient.errors import HttpError
 from langchain.schema import Document
 
 
-TOKEN_PATH = os.getenv("GOOGLE_DOCS_TOKEN_JSON", "secrets/token.json")
-CREDENTIALS_PATH = os.getenv("GOOGLE_DOCS_CREDENTIALS_JSON", "secrets/credentials.json")
+DOCS_TOKEN = os.getenv("GOOGLE_DOCS_TOKEN_JSON")
+DOCS_CREDENTIALS = os.getenv("GOOGLE_DOCS_CREDENTIALS_JSON")
 
 
 SCOPES = [
@@ -33,21 +33,24 @@ creds = None
 ### Google Drive Authentication
 def auth_google_docs():
     global creds
-    if os.path.exists(TOKEN_PATH):
-        creds = Credentials.from_authorized_user_file(TOKEN_PATH, SCOPES)
+    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: 
-            flow = InstalledAppFlow.from_client_config(
-                CREDENTIALS_PATH, SCOPES
-            )
-            creds = flow.run_local_server(port=8080, open_browser=False)
-
-        #Save the credentials
-        with open(TOKEN_PATH, "w") as token:
-            token.write(creds.to_json())
+            if DOCS_CREDENTIALS: 
+                client_config_json = json.loads(DOCS_CREDENTIALS)
+                flow = InstalledAppFlow.from_client_config(
+                    client_config_json, SCOPES
+                )
+                
+                creds = flow.run_local_server(port=8080, open_browser=False)
 
     return creds