Update main.py

main.py

@@ -1,14 +1,14 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from argopy import DataFetcher
 import matplotlib.pyplot as plt
-from fastapi.responses import StreamingResponse
-import io
-
+import uuid
+import os
 
 app = FastAPI()
 
 @app.get("/plot_trajectory")
-def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: float, 
+def plot_trajectory(request: Request,
+                    lon_min: float, lon_max: float, lat_min: float, lat_max: float, 
                     depth_min: float, depth_max: float, 
                     date_start: str, date_end: str):
     try:
@@ -18,12 +18,14 @@ def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: flo
         # Plotting
         plt.figure()
         f.plot('trajectory', add_legend=False)
-        buf = io.BytesIO()
-        plt.savefig(buf, format='png')
-        buf.seek(0)
+
+        # Instead of saving the plot, construct the URL
+        base_url = str(request.base_url)
+        image_url = f"{base_url}plot_trajectory?lon_min={lon_min}&lon_max={lon_max}&lat_min={lat_min}&lat_max={lat_max}&depth_min={depth_min}&depth_max={depth_max}&date_start={date_start}&date_end={date_end}"
+
         plt.close()
 
-        return StreamingResponse(buf, media_type="image/png")
+        return {"image_url": image_url}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))