Update app.py

app.py

@@ -47,10 +47,13 @@ async def run_code(code: Code):
     exec(code.code)
     # img_buffer.seek(0)  # Reset buffer position
 
-    file_path = "graph.png"
-    plt.savefig(file_path)
+    file_path = "graph.pdf"
+    fig.save_fig(file_path)
     plt.close()
+    # plt.savefig(file_path)
+    # plt.close()
 
     # Return image as response
     # return Response(content=img_buffer.getvalue(), media_type="image/png")
-    return FileResponse(file_path, media_type="image/png")
+    # return FileResponse(file_path, media_type="image/png")
+     return FileResponse(file_path, media_type="application/pdf", filename="graph.pdf")