Update app.py

app.py

@@ -438,7 +438,7 @@ def generate_documentation_page():
     st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Project"):
         st.session_state.page = "project_view"
-        st.experimental_rerun()
+        st.rerun()
     if st.sidebar.button("Log Out"):
         logout_user()
         st.rerun()
@@ -504,49 +504,54 @@ def generate_documentation_page():
         pdf_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.pdf")
         markdown_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.md")
 
+        col1, col2 = st.columns(2)
+
         # PDF generation and download logic
-        if not st.session_state.get("pdf_generated", False):
-            if st.button("Generate PDF"):
+        with col1:
+            if not st.session_state.get("pdf_generated", False):
+                if st.button("Generate PDF"):
+                    try:
+                        generate_pdf(documentation, pdf_path)
+                        st.session_state.pdf_generated = True
+                        st.success("PDF Available in the 'Saved Documentation' Page")
+                        st.rerun()
+                    except Exception as e:
+                        st.error(f"Failed to generate PDF: {e}")
+            else:
                 try:
-                    generate_pdf(documentation, pdf_path)
-                    st.session_state.pdf_generated = True
-                    st.success("PDF Available in the 'Saved Documentation' Page")
-                    st.rerun()
-                except Exception as e:
-                    st.error(f"Failed to generate PDF: {e}")
-        else:
-            try:
-                with open(pdf_path, "rb") as pdf_file:
-                    st.download_button(
-                        label="Download PDF",
-                        data=pdf_file.read(),
-                        file_name=f"{st.session_state.current_project}_Documentation.pdf",
-                        mime="application/pdf",
-                    )
-            except FileNotFoundError:
-                st.error("PDF file not found. Try generating it again.")
+                    with open(pdf_path, "rb") as pdf_file:
+                        st.download_button(
+                            label="Download PDF",
+                            data=pdf_file.read(),
+                            file_name=f"{st.session_state.current_project}_Documentation.pdf",
+                            mime="application/pdf",
+                        )
+                except FileNotFoundError:
+                    st.error("PDF file not found. Try generating it again.")
 
         # Markdown generation and download logic
-        if not st.session_state.get("markdown_generated", False):
-            if st.button("Generate Markdown File"):
+        with col2:
+            if not st.session_state.get("markdown_generated", False):
+                if st.button("Generate Markdown File"):
+                    try:
+                        generate_markdown_file(documentation, markdown_path)
+                        st.session_state.markdown_generated = True
+                        st.success("Markdown Available in the 'Saved Documentation' Page")
+                        st.rerun()
+                    except Exception as e:
+                        st.error(f"Failed to generate Markdown file: {e}")
+            else:
                 try:
-                    generate_markdown_file(documentation, markdown_path)
-                    st.session_state.markdown_generated = True
-                    st.success("Markdown Available in the 'Saved Documentation' Page")
-                    st.rerun()
-                except Exception as e:
-                    st.error(f"Failed to generate Markdown file: {e}")
-        else:
-            try:
-                with open(markdown_path, "rb") as markdown_file:
-                    st.download_button(
-                        label="Download Markdown File",
-                        data=markdown_file.read(),
-                        file_name=f"{st.session_state.current_project}_Documentation.md",
-                        mime="text/markdown",
-                    )
-            except FileNotFoundError:
-                st.error("Markdown file not found. Try generating it again.")
+                    with open(markdown_path, "rb") as markdown_file:
+                        st.download_button(
+                            label="Download Markdown File",
+                            data=markdown_file.read(),
+                            file_name=f"{st.session_state.current_project}_Documentation.md",
+                            mime="text/markdown",
+                        )
+                except FileNotFoundError:
+                    st.error("Markdown file not found. Try generating it again.")
+