Update azure_ocr.py

azure_ocr.py

@@ -5,14 +5,12 @@ from PyPDF2 import PdfReader, PdfWriter
 import tempfile
 import re
 
-AZURE_ENDPOINT = os.env.get("AZURE_ENDPOINT")
-AZURE_KEY = os.env.get("AZURE_KEY")
-
-if not AZURE_ENDPOINT or not AZURE_KEY:
-    raise RuntimeError("Set AZURE_ENDPOINT and AZURE_KEY in .env")
-
-AZURE_ENDPOINT = AZURE_ENDPOINT.rstrip("/")
-
+def get_azure_config():
+    endpoint = os.environ.get("AZURE_ENDPOINT")
+    key = os.environ.get("AZURE_KEY")
+    if not endpoint or not key:
+        raise RuntimeError("Set AZURE_ENDPOINT and AZURE_KEY in .env or environment")
+    return endpoint.rstrip("/"), key
 
 def read_file_bytes(path):
     with open(path, "rb") as f: