Avinashnalla7 commited on
Commit
5958bb8
·
1 Parent(s): 20919fe

Allow PDF_PIPELINE_GMAIL_* env fallbacks

Browse files
Files changed (1) hide show
  1. backend/worker/worker.py +9 -0
backend/worker/worker.py CHANGED
@@ -4,6 +4,13 @@ import os
4
  import requests
5
  GIT_SHA=os.getenv('HF_REVISION','') or os.getenv('GIT_SHA','')
6
 
 
 
 
 
 
 
 
7
  def _upload_pdf_to_api(pdf_id, pdf_path, pdf_name):
8
  base = (os.environ.get("PDF_TRAINER_API_BASE") or "").strip()
9
  if not base:
@@ -83,6 +90,8 @@ class Settings:
83
 
84
  def load_settings() -> Settings:
85
  base = Path(__file__).resolve().parents[1] # backend/
 
 
86
  creds = resolve_json_or_path("GMAIL_CREDENTIALS_JSON", base / "credentials.json", Path("/tmp/credentials.json"))
87
  token = resolve_json_or_path("GMAIL_TOKEN_JSON", base / "token.json", Path("/tmp/token.json"))
88
 
 
4
  import requests
5
  GIT_SHA=os.getenv('HF_REVISION','') or os.getenv('GIT_SHA','')
6
 
7
+ def _alias_env(primary: str, fallback: str) -> None:
8
+ if (os.environ.get(primary) or "").strip():
9
+ return
10
+ fb = (os.environ.get(fallback) or "").strip()
11
+ if fb:
12
+ os.environ[primary] = fb
13
+
14
  def _upload_pdf_to_api(pdf_id, pdf_path, pdf_name):
15
  base = (os.environ.get("PDF_TRAINER_API_BASE") or "").strip()
16
  if not base:
 
90
 
91
  def load_settings() -> Settings:
92
  base = Path(__file__).resolve().parents[1] # backend/
93
+ _alias_env("GMAIL_CREDENTIALS_JSON", "PDF_PIPELINE_GMAIL_CREDENTIALS_JSON")
94
+ _alias_env("GMAIL_TOKEN_JSON", "PDF_PIPELINE_GMAIL_TOKEN_JSON")
95
  creds = resolve_json_or_path("GMAIL_CREDENTIALS_JSON", base / "credentials.json", Path("/tmp/credentials.json"))
96
  token = resolve_json_or_path("GMAIL_TOKEN_JSON", base / "token.json", Path("/tmp/token.json"))
97