changes in gmail auth.py

app/gmail_auth.py

@@ -1,20 +1,50 @@
+import os
+from pathlib import Path
 from app.core.config import settings
+
+# CREATE TEMP DIRECTORY
+
+AUTH_DIR = Path("/tmp/google_auth")
+AUTH_DIR.mkdir(parents=True, exist_ok=True)
+
+# FULL FILE PATHS
+
+CREDENTIALS_PATH = AUTH_DIR / "credentials.json"
+TOKEN_PATH = AUTH_DIR / "token.json"
+
+# CREATE FILES FROM HF SECRETS
+
+if not CREDENTIALS_PATH.exists():
+    CREDENTIALS_PATH.write_text(os.environ["GOOGLE_CREDENTIALS"])
+
+if not TOKEN_PATH.exists():
+    TOKEN_PATH.write_text(os.environ["GOOGLE_TOKEN"])
+
+# GMAIL IMPORTS
+
 from langchain_google_community.gmail.utils import (
     build_resource_service,
     get_gmail_credentials,
 )
 
 from langchain_google_community import GmailToolkit
+
+# LOAD CREDENTIALS
+
 credentials = get_gmail_credentials(
-    token_file=settings.GMAIL_TOKEN_PATH,
+    token_file=str(TOKEN_PATH),
     scopes=["https://mail.google.com/"],
-    client_sercret_file=settings.GMAIL_CREDENTIALS_PATH,
+    client_secret_file=str(CREDENTIALS_PATH),
 )
 
+# BUILD API RESOURCE
+
 api_resource = build_resource_service(
     credentials=credentials
 )
 
+# TOOLKIT
+
 gmail_toolkit = GmailToolkit(
     api_resource=api_resource
 )

app/main.py

@@ -1,3 +1,5 @@
+import os
+from app.core.config import settings
 from fastapi import FastAPI, HTTPException, Depends
 from pydantic import BaseModel, EmailStr
 from typing import Optional, Dict, Any, TypedDict, Annotated, Sequence
@@ -11,25 +13,13 @@ from app.database.connection import get_session
 from app.database.utils import get_or_create_user
 from sqlalchemy.orm import Session
 from app.database.connection import SessionLocal
-from app.core.config import settings
 from fastapi import Request
-import os
 from app.database.models import User
 from app.core.auth import create_access_token,get_current_user
 
 # CREATE GMAIL AUTH FILES FROM HF SECRETS
 
 
-if not os.path.exists(settings.GMAIL_CREDENTIALS_PATH):
-
-    with open(settings.GMAIL_CREDENTIALS_PATH, "w") as f:
-        f.write(os.environ["GOOGLE_CREDENTIALS"])
-
-if not os.path.exists(settings.GMAIL_TOKEN_PATH):
-
-    with open(settings.GMAIL_TOKEN_PATH, "w") as f:
-        f.write(os.environ["GOOGLE_TOKEN"])
-
 
 logger = logging.getLogger(__name__)