Spaces:
Paused
Paused
| 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 | |
| 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() | |