Update app.py

app.py

@@ -49,12 +49,12 @@ async def run_code(code: Code):
 
     file_path = "graph.pdf"
 
-    if code.code == "plt.subplots(":
+    if "plt.subplots(" in code.code:
         print("SUBPLOTS DETECTED")
-        exec(code.code + "\nfig.savefig(file_path)")
+        exec(code.code + "\nfig.savefig(file_path)\nplt.close()")
     else:
         print("NO SUBPLOTS")
-        exec(code.code + "\nplt.savefig(file_path)")
+        exec(code.code + "\nplt.savefig(file_path)\nplt.close()")
 
     plt.close()
     # plt.savefig(file_path)