Update app.py

app.py

@@ -23,20 +23,41 @@ def _normalize_to_png(arr: np.ndarray, clip_high_if_needed=True) -> Image.Image:
     a = (a * 255).astype(np.uint8)
     return Image.fromarray(a)
 
-
 def _mask_from_geojson(ds: rasterio.io.DatasetReader, gj_text: str | None) -> np.ndarray:
-    if not gj_text:
+    """Return a boolean mask for the AOI. If text is empty/invalid, return full-extent mask."""
+    # Full-extent mask helper
+    def full_mask():
         return np.ones((ds.height, ds.width), dtype=bool)
+
+    if not gj_text or not str(gj_text).strip():
+        return full_mask()
+
+    # Try parse JSON; on failure -> full mask (do not crash job)
     try:
         gj = json.loads(gj_text)
     except Exception:
-        raise ValueError("Invalid GeoJSON text provided.")
-    if gj.get("type") == "FeatureCollection":
-        geoms = [shape(f["geometry"]) for f in gj["features"]]
-    elif gj.get("type") == "Feature":
-        geoms = [shape(gj["geometry"])]
-    else:
-        geoms = [shape(gj)]
+        return full_mask()
+
+    # Accept FeatureCollection / Feature / raw geometry
+    try:
+        if isinstance(gj, dict) and gj.get("type") == "FeatureCollection":
+            geoms = [shape(f["geometry"]) for f in gj.get("features", []) if "geometry" in f]
+        elif isinstance(gj, dict) and gj.get("type") == "Feature":
+            geoms = [shape(gj["geometry"])]
+        elif isinstance(gj, dict) and "type" in gj:
+            geoms = [shape(gj)]
+        # Also accept a bare coordinate ring (list) like [[x,y], ...]
+        elif isinstance(gj, (list, tuple)) and gj and isinstance(gj[0], (list, tuple)):
+            from shapely.geometry import Polygon
+            geoms = [Polygon(gj)]
+        else:
+            return full_mask()
+    except Exception:
+        return full_mask()
+
+    if not geoms:
+        return full_mask()
+
     mask = rasterize(
         [(g, 1) for g in geoms],
         out_shape=(ds.height, ds.width),
@@ -48,6 +69,7 @@ def _mask_from_geojson(ds: rasterio.io.DatasetReader, gj_text: str | None) -> np
     return mask
 
 
+
 def _open_geotiff_strict(path: str) -> rasterio.io.DatasetReader:
     """
     Open the raster and ensure it's a GeoTIFF/COG.