Spaces:
Runtime error
Runtime error
Commit ·
5958bb8
1
Parent(s): 20919fe
Allow PDF_PIPELINE_GMAIL_* env fallbacks
Browse files- 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 |
|