Spaces:
Paused
Paused
File size: 4,088 Bytes
c8c487a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | from __future__ import annotations
import os
from dataclasses import dataclass
from pathlib import Path
from dotenv import load_dotenv
load_dotenv()
def _split_csv(value: str) -> tuple[str, ...]:
return tuple(item.strip() for item in value.split(",") if item.strip())
def _resolve_file_path(env_var: str, *fallbacks: str) -> Path | None:
backend_root = Path(__file__).resolve().parent.parent
workspace_root = backend_root.parent
candidates: list[str] = []
val = os.getenv(env_var)
if val:
candidates.append(val)
candidates.extend(fallbacks)
for c in candidates:
p = Path(c)
if p.is_file():
return p.resolve()
p_b = (backend_root / c).resolve()
if p_b.is_file():
return p_b
p_w = (workspace_root / c).resolve()
if p_w.is_file():
return p_w
p_top = (workspace_root.parent / c).resolve()
if p_top.is_file():
return p_top
return None
@dataclass(frozen=True)
class Settings:
environment: str = os.getenv("SATQUERY_ENV", "development")
artifact_dir: Path = Path(os.getenv("SATQUERY_ARTIFACT_DIR", "artifacts"))
max_upload_mb: int = int(os.getenv("SATQUERY_MAX_UPLOAD_MB", "256"))
cors_origins: tuple[str, ...] = _split_csv(
os.getenv("SATQUERY_CORS_ORIGINS", "http://localhost:5173,http://localhost:8080")
)
earthdial_endpoint: str | None = os.getenv("SATQUERY_EARTHDIAL_ENDPOINT") or None
earthdial_rgb_endpoint: str | None = (
os.getenv("SATQUERY_EARTHDIAL_RGB_ENDPOINT") or os.getenv("SATQUERY_EARTHDIAL_ENDPOINT") or None
)
earthdial_ms_endpoint: str | None = (
os.getenv("SATQUERY_EARTHDIAL_MS_ENDPOINT") or os.getenv("SATQUERY_EARTHDIAL_ENDPOINT") or None
)
croma_endpoint: str | None = os.getenv("SATQUERY_CROMA_ENDPOINT") or None
remoteclip_endpoint: str | None = os.getenv("SATQUERY_REMOTECLIP_ENDPOINT") or None
change_endpoint: str | None = os.getenv("SATQUERY_CHANGE_ENDPOINT") or None
vlm_endpoint: str | None = os.getenv("SATQUERY_VLM_ENDPOINT") or None
vlm_api_key: str | None = os.getenv("SATQUERY_VLM_API_KEY") or None
vlm_model_name: str = os.getenv("SATQUERY_VLM_MODEL_NAME", "managed-geospatial-vlm")
tile_size: int = int(os.getenv("SATQUERY_TILE_SIZE", "448"))
tile_overlap: int = int(os.getenv("SATQUERY_TILE_OVERLAP", "64"))
max_model_tiles: int = int(os.getenv("SATQUERY_MAX_MODEL_TILES", "32"))
database_path: Path = Path(os.getenv("SATQUERY_DATABASE_PATH", "artifacts/satquery.sqlite3"))
water_model_checkpoint: Path | None = _resolve_file_path(
"SATQUERY_WATER_CHECKPOINT",
"models/satquery_water_bundle/satquery_water_state_dict.pt",
"satquery_water_state_dict.pt",
)
water_model_config: Path | None = _resolve_file_path(
"SATQUERY_WATER_CONFIG",
"models/satquery_water_bundle/satquery_water_config.json",
"satquery_water_config.json",
)
water_model_metrics: Path | None = _resolve_file_path(
"SATQUERY_WATER_METRICS",
"models/satquery_water_bundle/satquery_water_metrics.json",
"satquery_water_metrics.json",
)
water_model_enabled: bool = os.getenv("SATQUERY_WATER_MODEL_ENABLED", "true").lower() in ("true", "1", "yes")
landcover_model_checkpoint: Path | None = _resolve_file_path(
"SATQUERY_LANDCOVER_CHECKPOINT",
"models/finetuned/landcover/satquery_landcover_v1/best_checkpoint.pt",
"models/landcover/best_checkpoint.pt",
)
buildings_model_checkpoint: Path | None = _resolve_file_path(
"SATQUERY_BUILDINGS_CHECKPOINT",
"models/finetuned/buildings/satquery_buildings_v1/satquery_buildings_state_dict.pt",
"models/buildings/satquery_buildings_state_dict.pt",
)
landcover_model_enabled: bool = os.getenv("SATQUERY_LANDCOVER_MODEL_ENABLED", "true").lower() in ("true", "1", "yes")
buildings_model_enabled: bool = os.getenv("SATQUERY_BUILDINGS_MODEL_ENABLED", "true").lower() in ("true", "1", "yes")
settings = Settings()
|