Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from fastapi.responses import Response
+from fastapi.responses import FileResponse
 from pydantic import BaseModel
 
 import random
@@ -42,9 +43,14 @@ async def create_ticket(ticket: Ticket):
 
 @app.post("/run_code")
 async def run_code(code: Code):
-    img_buffer = io.BytesIO()
+    # img_buffer = io.BytesIO()
     exec(code.code)
-    img_buffer.seek(0)  # Reset buffer position
+    # img_buffer.seek(0)  # Reset buffer position
+
+    file_path = "graph.png"
+    plt.savefig(file_path)
+    plt.close()
 
     # Return image as response
-    return Response(content=img_buffer.getvalue(), media_type="image/png")
+    # return Response(content=img_buffer.getvalue(), media_type="image/png")
+    return FileResponse(file_path, media_type="image/png")