Update app.py

app.py

@@ -315,19 +315,32 @@ def main():
     display_centers()
     display_other_properties()
 
+    # Plot triangle
     fig = plot_triangle(vertices, properties)
 
-    if st.button("Save as PDF"):
-        save_as_pdf_reportlab(
+    ### Code for Save as PDF
+    # Generate the PDF in-memory
+    pdf_file = save_as_pdf_reportlab(
             coords['x1'], coords['y1'], coords['x2'], coords['y2'], coords['x3'], coords['y3'],
             a, b, c, A, B, C, area, perimeter, I_x, I_y, U_x, U_y, G_x, G_y,
             midpoints, radius_in, radius_circum, fig
-        )
-        st.success("Results saved as PDF!")
-        with open("Triangle_solver_results.pdf", "rb") as pdf_file:
-            st.download_button("Download PDF", pdf_file, file_name="Triangle_solver_results.pdf")
+    )
+
+    # Allow user to download the generated PDF file
+    st.download_button("Download PDF", pdf_file, file_name="Triangle_solver_results.pdf", mime="application/pdf")
+
+    ### Old code for Save as PDF
+    # if st.button("Save as PDF"):
+    #     save_as_pdf_reportlab(
+    #         coords['x1'], coords['y1'], coords['x2'], coords['y2'], coords['x3'], coords['y3'],
+    #         a, b, c, A, B, C, area, perimeter, I_x, I_y, U_x, U_y, G_x, G_y,
+    #         midpoints, radius_in, radius_circum, fig
+    #     )
+    #     st.success("Results saved as PDF!")
+    #     with open("Triangle_solver_results.pdf", "rb") as pdf_file:
+    #         st.download_button("Download PDF", pdf_file, file_name="Triangle_solver_results.pdf")
 
-    # New code for Save as PDF
+    ### New code for Save as PDF
     # if st.button("Save as PDF"):
     #     try:
     #         # Add debug message