Update main.py

main.py

@@ -1,14 +1,14 @@
-from fastapi import FastAPI, HTTPException, Request
+from fastapi import FastAPI, HTTPException
 from argopy import DataFetcher
 import matplotlib.pyplot as plt
-import uuid
-import os
+from fastapi.responses import StreamingResponse
+import io
+
 
 app = FastAPI()
 
 @app.get("/plot_trajectory")
-def plot_trajectory(request: Request,
-                    lon_min: float, lon_max: float, lat_min: float, lat_max: float, 
+def plot_trajectory(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,19 +18,22 @@ def plot_trajectory(request: Request,
         # Plotting
         plt.figure()
         f.plot('trajectory', add_legend=False)
-
-        # Instead of saving the plot, construct the URL
-        base_url = str(request.base_url)
+        buf = io.BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        plt.close()
+        StreamingResponse(buf, media_type="image/png")
+        
         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 {"image_url": image_url}
-
+        
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)