Update main.py

main.py

@@ -1,18 +1,15 @@
 from fastapi import FastAPI, HTTPException
 from argopy import DataFetcher
 import matplotlib.pyplot as plt
-import uuid
-import os
-import matplotlib
+from fastapi.responses import StreamingResponse
+import io
 
-# Set MPLCONFIGDIR to a writable directory
-os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib_config'
 
 app = FastAPI()
 
 @app.get("/plot_trajectory")
 def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: float, 
-                    depth_min: int, depth_max: int, 
+                    depth_min: float, depth_max: float, 
                     date_start: str, date_end: str):
     try:
         # Create data fetcher with user inputs
@@ -21,20 +18,17 @@ def plot_trajectory(lon_min: float, lon_max: float, lat_min: float, lat_max: flo
         # Plotting
         plt.figure()
         f.plot('trajectory', add_legend=False)
-        
-        # Save the plot to a file
-        filename = str(uuid.uuid4()) + ".png"  # Unique filename
-        file_path = "/path/to/public/directory/" + filename  # Adjust this path
-        plt.savefig(file_path)
+        buf = io.BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
         plt.close()
 
-        # Generate public URL
-        public_url = "http://yourserver.com/" + filename  # Replace with your server's URL
+        return StreamingResponse(buf, media_type="image/png")
 
-        return {"image_url": public_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)