Update app.py

app.py

@@ -504,10 +504,12 @@ def generate_documentation_page():
         markdown_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.md")
 
         # Generate PDF Button
-        if not os.path.exists(pdf_path):
+        if not os.path.exists(pdf_path) and not pdf_generated:
             if st.button("Generate PDF"):
                 try:
                     generate_pdf(documentation, pdf_path)
+                    st.session_state.pdf_generated = True
+                    st.rerun()
                     st.success("PDF Available in the 'Saved Documentation' Page.")
                 except Exception as e:
                     st.error(f"Failed to generate PDF: {e}")
@@ -515,10 +517,12 @@ def generate_documentation_page():
             st.success("PDF has already been generated! Check the 'Saved Documentation' Page.")
 
         # Generate Markdown Button
-        if not os.path.exists(markdown_path):
+        if not os.path.exists(markdown_path) and not markdown_generated:
             if st.button("Generate Markdown File"):
                 try:
                     generate_markdown_file(documentation, markdown_path)
+                    st.session_state.markdown_generated = True  # Set a flag for Markdown generation
+                    st.rerun()  # Trigger a rerun to show the download button
                     st.success("Markdown Available in the 'Saved Documentation' Page.")
                 except Exception as e:
                     st.error(f"Failed to generate Markdown file: {e}")