corrected last commit

app.py

@@ -25,10 +25,20 @@ async def predict_ndvi_api(file: UploadFile = File(...)):
         norm_img = normalize_rgb(np.array(img))
         pred_ndvi = predict_ndvi(model, norm_img)
 
-        img_io = create_visualization(norm_img, pred_ndvi)
-        img_io.seek(0)
-        img_base64 = base64.b64encode(img_io.read()).decode("utf-8")
+        # Prepare visualization image
+        vis_img_io = create_visualization(norm_img, pred_ndvi)
+        vis_img_io.seek(0)
+        vis_img_base64 = base64.b64encode(vis_img_io.read()).decode("utf-8")
 
-        return JSONResponse(content={"image_base64": img_base64})
+        # Convert NDVI array to nested list (e.g., for JSON)
+        ndvi_list = pred_ndvi.tolist()
+
+        return JSONResponse(
+            content={
+                "ndvi_array": ndvi_list,
+                "visualization": vis_img_base64
+            }
+        )
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
+