Ensure local Modal runtime autoload

app.py

@@ -5,9 +5,11 @@ from fastapi import Request
 from fastapi.responses import JSONResponse, RedirectResponse
 
 from dream_customs import zerogpu  # noqa: F401
+from dream_customs.runtime_env import auto_load_runtime_env_json
 from dream_customs.ui.app import build_demo
 
 
+auto_load_runtime_env_json()
 demo = build_demo()
 
 

dream_customs/runtime_env.py

@@ -0,0 +1,61 @@
+import json
+import os
+from pathlib import Path
+
+
+SPACE_ID = "build-small-hackathon/dream-customs"
+DEFAULT_RUNTIME_ENV_JSON = Path("/tmp/dream-customs-runtime.json")
+RUNTIME_ENV_KEYS = {
+    "DREAM_CUSTOMS_TEXT_ENDPOINT",
+    "DREAM_CUSTOMS_VISION_ENDPOINT",
+    "DREAM_CUSTOMS_ASR_ENDPOINT",
+    "DREAM_CUSTOMS_HOSTED_TOKEN",
+}
+
+
+def runtime_env_json_path() -> Path:
+    return Path(os.getenv("DREAM_CUSTOMS_RUNTIME_ENV_JSON", str(DEFAULT_RUNTIME_ENV_JSON)))
+
+
+def _runtime_env_autoload_enabled() -> bool:
+    value = os.getenv("DREAM_CUSTOMS_DISABLE_RUNTIME_ENV_JSON", "").strip().lower()
+    return value not in {"1", "true", "yes", "on"}
+
+
+def load_runtime_env_json(path: Path) -> dict:
+    if not path.exists():
+        return {"loaded": False, "path": str(path), "reason": "missing"}
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+    except (OSError, json.JSONDecodeError) as exc:
+        return {"loaded": False, "path": str(path), "reason": exc.__class__.__name__}
+
+    loaded_keys = []
+    for key in sorted(RUNTIME_ENV_KEYS):
+        value = str(data.get(key, "")).strip()
+        if value:
+            os.environ[key] = value
+            loaded_keys.append(key)
+    return {
+        "loaded": bool(loaded_keys),
+        "path": str(path),
+        "configured_keys": loaded_keys,
+    }
+
+
+def auto_load_runtime_env_json() -> dict:
+    if not _runtime_env_autoload_enabled():
+        return {"loaded": False, "reason": "disabled"}
+    if os.getenv("PYTEST_CURRENT_TEST") and not os.getenv("DREAM_CUSTOMS_RUNTIME_ENV_JSON"):
+        return {"loaded": False, "reason": "pytest"}
+    return load_runtime_env_json(runtime_env_json_path())
+
+
+def configured_env(space_id: str = SPACE_ID) -> dict:
+    return {
+        "space_id": os.getenv("SPACE_ID", space_id),
+        "text_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_TEXT_ENDPOINT", "").strip()),
+        "vision_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_VISION_ENDPOINT", "").strip()),
+        "asr_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_ASR_ENDPOINT", "").strip()),
+        "hosted_token_configured": bool(os.getenv("DREAM_CUSTOMS_HOSTED_TOKEN", "").strip()),
+    }

scripts/local_space_mirror.py

@@ -9,16 +9,17 @@ import os
 import sys
 from pathlib import Path
 
-
 ROOT = Path(__file__).resolve().parents[1]
-SPACE_ID = "build-small-hackathon/dream-customs"
-DEFAULT_RUNTIME_ENV_JSON = Path("/tmp/dream-customs-runtime.json")
-RUNTIME_ENV_KEYS = {
-    "DREAM_CUSTOMS_TEXT_ENDPOINT",
-    "DREAM_CUSTOMS_VISION_ENDPOINT",
-    "DREAM_CUSTOMS_ASR_ENDPOINT",
-    "DREAM_CUSTOMS_HOSTED_TOKEN",
-}
+root = str(ROOT)
+if root not in sys.path:
+    sys.path.insert(0, root)
+
+from dream_customs.runtime_env import (
+    DEFAULT_RUNTIME_ENV_JSON,
+    SPACE_ID,
+    configured_env,
+    load_runtime_env_json,
+)
 
 
 def _repo_root_on_path() -> None:
@@ -27,37 +28,6 @@ def _repo_root_on_path() -> None:
         sys.path.insert(0, root)
 
 
-def _configured_env() -> dict:
-    return {
-        "space_id": os.getenv("SPACE_ID", SPACE_ID),
-        "text_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_TEXT_ENDPOINT", "").strip()),
-        "vision_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_VISION_ENDPOINT", "").strip()),
-        "asr_endpoint_configured": bool(os.getenv("DREAM_CUSTOMS_ASR_ENDPOINT", "").strip()),
-        "hosted_token_configured": bool(os.getenv("DREAM_CUSTOMS_HOSTED_TOKEN", "").strip()),
-    }
-
-
-def load_runtime_env_json(path: Path) -> dict:
-    if not path.exists():
-        return {"loaded": False, "path": str(path), "reason": "missing"}
-    try:
-        data = json.loads(path.read_text(encoding="utf-8"))
-    except (OSError, json.JSONDecodeError) as exc:
-        return {"loaded": False, "path": str(path), "reason": exc.__class__.__name__}
-
-    loaded_keys = []
-    for key in sorted(RUNTIME_ENV_KEYS):
-        value = str(data.get(key, "")).strip()
-        if value:
-            os.environ[key] = value
-            loaded_keys.append(key)
-    return {
-        "loaded": bool(loaded_keys),
-        "path": str(path),
-        "configured_keys": loaded_keys,
-    }
-
-
 def mirror_manifest(host: str, port: int) -> dict:
     _repo_root_on_path()
     from dream_customs.defaults import DEFAULT_ASR_BACKEND, DEFAULT_TEXT_BACKEND, DEFAULT_VISION_BACKEND
@@ -72,7 +42,7 @@ def mirror_manifest(host: str, port: int) -> dict:
             "vision": DEFAULT_VISION_BACKEND,
             "asr": DEFAULT_ASR_BACKEND,
         },
-        "env": _configured_env(),
+        "env": configured_env(SPACE_ID),
         "notes": [
             "Uses the same app.py/build_demo path as the Hugging Face Space.",
             "Missing hosted endpoint secrets fall back to deterministic demo behavior.",

tests/test_local_space_mirror.py

@@ -1,10 +1,20 @@
 import json
+import os
+import subprocess
+import sys
 
 from scripts.local_space_mirror import load_runtime_env_json, mirror_manifest
 from scripts.smoke_local_space_mirror import inspect_config
 
 
-def test_local_space_mirror_manifest_matches_space_entrypoint():
+def test_local_space_mirror_manifest_matches_space_entrypoint(monkeypatch):
+    for key in [
+        "DREAM_CUSTOMS_TEXT_ENDPOINT",
+        "DREAM_CUSTOMS_VISION_ENDPOINT",
+        "DREAM_CUSTOMS_ASR_ENDPOINT",
+        "DREAM_CUSTOMS_HOSTED_TOKEN",
+    ]:
+        monkeypatch.delenv(key, raising=False)
     manifest = mirror_manifest("127.0.0.1", 7862)
 
     assert manifest["mode"] == "local-space-mirror"
@@ -61,6 +71,58 @@ def test_local_space_mirror_loads_runtime_env_json_without_printing_values(tmp_p
     assert "super-secret-token" not in serialized
 
 
+def test_app_import_auto_loads_local_runtime_env_json(tmp_path):
+    runtime_json = tmp_path / "runtime.json"
+    runtime_json.write_text(
+        json.dumps(
+            {
+                "DREAM_CUSTOMS_TEXT_ENDPOINT": "https://example.test/text",
+                "DREAM_CUSTOMS_VISION_ENDPOINT": "https://example.test/vision",
+                "DREAM_CUSTOMS_ASR_ENDPOINT": "https://example.test/asr",
+                "DREAM_CUSTOMS_HOSTED_TOKEN": "super-secret-token",
+            }
+        ),
+        encoding="utf-8",
+    )
+    env = os.environ.copy()
+    env.pop("PYTEST_CURRENT_TEST", None)
+    for key in [
+        "DREAM_CUSTOMS_TEXT_ENDPOINT",
+        "DREAM_CUSTOMS_VISION_ENDPOINT",
+        "DREAM_CUSTOMS_ASR_ENDPOINT",
+        "DREAM_CUSTOMS_HOSTED_TOKEN",
+    ]:
+        env.pop(key, None)
+    env["DREAM_CUSTOMS_RUNTIME_ENV_JSON"] = str(runtime_json)
+    result = subprocess.run(
+        [
+            sys.executable,
+            "-c",
+            (
+                "import json, os; import app; "
+                "print(json.dumps({key: bool(os.getenv(key)) for key in "
+                "['DREAM_CUSTOMS_TEXT_ENDPOINT','DREAM_CUSTOMS_VISION_ENDPOINT',"
+                "'DREAM_CUSTOMS_ASR_ENDPOINT','DREAM_CUSTOMS_HOSTED_TOKEN']}))"
+            ),
+        ],
+        check=True,
+        capture_output=True,
+        text=True,
+        env=env,
+    )
+
+    loaded = json.loads(result.stdout.strip())
+
+    assert loaded == {
+        "DREAM_CUSTOMS_TEXT_ENDPOINT": True,
+        "DREAM_CUSTOMS_VISION_ENDPOINT": True,
+        "DREAM_CUSTOMS_ASR_ENDPOINT": True,
+        "DREAM_CUSTOMS_HOSTED_TOKEN": True,
+    }
+    assert "https://example.test" not in result.stdout
+    assert "super-secret-token" not in result.stdout
+
+
 def test_local_space_mirror_config_smoke_requires_composer_debug_markers():
     config = {
         "title": "Dream QA",