Confirm Geometry and Add Index API

app.py

@@ -817,6 +817,57 @@ with gr.Blocks(theme=theme, title="Kamlan: KML Analyzer") as demo:
         </div>
     """)
 
+def compute_index_histogram(file_url: str, index_name: str, start_date: str, end_date: str):
+    """Computes histogram percentage for a vegetation index within a polygon and date range."""
+    one_time_setup()
+    decoded_url = unquote(file_url)
+    gdf = get_gdf_from_url(decoded_url)
+    gdf = preprocess_gdf(gdf)
+
+    geometry_gdf = next((gdf.iloc[[i]] for i in range(len(gdf)) if is_valid_polygon(gdf.iloc[[i]])), None)
+    if geometry_gdf is None:
+        raise HTTPException(status_code=400, detail="No valid polygon found.")
+
+    ee_geometry = ee.Geometry(json.loads(geometry_gdf.to_crs(4326).to_json())['features'][0]['geometry'])
+
+    evi_vars = {'G': 2.5, 'C1': 6.0, 'C2': 7.5, 'L': 1.0, 'C': 2.4}
+    collection = (
+        ee.ImageCollection("COPERNICUS/S2_SR_HARMONIZED")
+        .filterDate(start_date, end_date)
+        .select(["B2", "B3", "B4", "B8"], ["Blue", "Green", "Red", "NIR"])
+        .map(lambda img: add_indices(img, 'NIR', 'Red', 'Blue', 'Green', evi_vars))
+        .filterBounds(ee_geometry)
+    )
+
+    if collection.size().getInfo() == 0:
+        raise HTTPException(status_code=404, detail="No imagery found for the polygon in given date range.")
+
+    # Use median composite to avoid cloud spikes
+    mosaic = collection.median()
+
+    bins = [0, 0.2, 0.4, 0.6, 0.8, 1.0]
+    histogram, _ = get_histogram(index_name, mosaic.select(index_name), ee_geometry, bins)
+
+    total = histogram.sum()
+    result = {}
+    for i in range(len(bins) - 1):
+        label = f"{bins[i]}-{bins[i+1]}"
+        pct = (histogram[i] / total * 100) if total > 0 else 0
+        result[label] = round(pct, 2)
+
+    return result
+
+
+# Register endpoints for multiple indices with date range
+for idx in ["NDVI", "EVI", "EVI2", "RandomForest", "CI", "GujVDI"]:
+    endpoint_path = f"/api/{idx}"
+
+    async def index_hist_endpoint(file_url: str, start_date: str, end_date: str, index_name=idx):
+        return compute_index_histogram(file_url, index_name, start_date, end_date)
+
+    app.get(endpoint_path)(index_hist_endpoint)
+
+
 
 # Mount the Gradio app onto the FastAPI app
 app = gr.mount_gradio_app(app, demo, path="/")