Update main.py

main.py

@@ -13,6 +13,7 @@ from spatial_queries import get_terrain_metrics, distance_to_water
 from vulnerability import calculate_vulnerability_index
 from gee_auth import initialize_gee
 from height_predictor.inference import get_predictor
+from height_predictor.get_height_gba import GlobalBuildingAtlasHeight
 
 # SHAP Explainer Initialization
 try:
@@ -39,6 +40,8 @@ templates = Jinja2Templates(directory="templates")
 # Thread pool for batch processing
 executor = ThreadPoolExecutor(max_workers=10)
 
+gba_getter = GlobalBuildingAtlasHeight()
+
 
 # DATA MODEL
 class SingleAssessment(BaseModel):
@@ -435,6 +438,20 @@ async def assess_multihazard(data: SingleAssessment) -> Dict:
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Assessment failed: {e}")
 
+@app.post("/get_height_gba")
+async def get_height_gba(data: SingleAssessment):
+    try:
+        result = gba_getter.get_height_m(data.latitude, data.longitude, buffer_m=20.0)
+
+        if result.get("status") != "success":
+            raise HTTPException(status_code=404, detail="GBA height not found for this location")
+
+        return result
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 
 @app.get("/health")
 async def health_check() -> Dict: