image_base64 + ndvi band

app.py

@@ -7,6 +7,7 @@ from io import BytesIO
 import numpy as np
 import io
 import os
+import base64
 
 app = FastAPI()
 model = load_model("ndvi_best_model.keras")
@@ -24,10 +25,10 @@ async def predict_ndvi_api(file: UploadFile = File(...)):
         norm_img = normalize_rgb(np.array(img))
         pred_ndvi = predict_ndvi(model, norm_img)
 
-        img_byte_arr = create_visualization(norm_img, pred_ndvi)  # already a BytesIO object
-        img_byte_arr.seek(0)
+        img_io = create_visualization(norm_img, pred_ndvi)
+        img_io.seek(0)
+        img_base64 = base64.b64encode(img_io.read()).decode("utf-8")
 
-        return StreamingResponse(img_byte_arr, media_type="image/png")
+        return JSONResponse(content={"image_base64": img_base64})
     except Exception as e:
         return JSONResponse(status_code=500, content={"error": str(e)})
-