Update app.py

app.py

@@ -60,19 +60,17 @@ async def run_code(code: Code):
     # plt.savefig(file_path)
     # plt.close()
 
-    # Get all open figure numbers
+    # Get all open figures
     figures = [plt.figure(i) for i in plt.get_fignums()]
-
     
-    # Save all figures in a single PDF
-    if figures:
-        figures[0].savefig("graph.pdf", format="pdf")
-        for fig in figures[1:]:
-            fig.savefig("graph.pdf", format="pdf", bbox_inches="tight", append=True)
-        
-        plt.close("all")  # Close all figures
-    else:
-        print("No open figures found.")
+    # Save all figures in a single PDF using PdfPages
+    pdf_filename = "all_graphs.pdf"
+    with PdfPages(pdf_filename) as pdf:
+        for fig in figures:
+            pdf.savefig(fig)  # Save each figure as a page in the PDF
+            plt.close(fig)  # Close the figure to free memory
+    
+    print(f"Saved all figures to {pdf_filename}")
 
     
     # return Response(content=img_buffer.getvalue(), media_type="image/png")