Update app.py

app.py

@@ -78,8 +78,8 @@ def main():
         project_view_page()
     elif st.session_state.page == "generate_documentation":
         generate_documentation_page()
-    elif st.session_state.page == "view_documentation":
-        view_documentation_page()
+    elif st.session_state.page == "saved_documentation":
+        saved_documentation_page()
 
 def login_page():
     st.subheader("Please Log In or Register to Continue")
@@ -590,7 +590,7 @@ def generate_markdown_file(documentation, output_path):
 
 
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-def view_documentation_page():
+def saved_documentation_page():
     # Sidebar with "Back to Project" and "Log Out" buttons
     st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Project"):
@@ -602,8 +602,8 @@ def view_documentation_page():
         st.session_state.page = "login"
         st.rerun()
 
-    st.subheader(f"View Documentation for {st.session_state.current_project}")
-    st.write("Below is a list of generated documentation for this project. Click to view or download.")
+    st.subheader(f"Saved Documentation for {st.session_state.current_project}")
+    st.write("Below is a list of saved documentation for this project. Click to download.")
 
     # Define paths for generated files
     user_folder = os.path.join("user_projects", st.session_state.username)
@@ -620,33 +620,21 @@ def view_documentation_page():
     else:
         if pdf_exists:
             with open(pdf_path, "rb") as pdf_file:
-                if st.button("View PDF Documentation"):
-                    st.subheader("PDF Documentation")
-                    st.write("The generated PDF documentation is displayed below:")
-                    st.download_button(
-                        label="Download PDF",
-                        data=pdf_file.read(),
-                        file_name=f"{st.session_state.current_project}_Documentation.pdf",
-                        mime="application/pdf",
-                    )
+                st.download_button(
+                    label="Download PDF Documentation",
+                    data=pdf_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.pdf",
+                    mime="application/pdf",
+                )
 
         if markdown_exists:
-            with open(markdown_path, "r") as markdown_file:
-                markdown_content = markdown_file.read()
-                if st.button("View Markdown Documentation"):
-                    st.subheader("Markdown Documentation")
-                    st.text_area(
-                        "Generated Markdown Documentation",
-                        value=markdown_content,
-                        height=600,
-                        disabled=True,
-                    )
-                    st.download_button(
-                        label="Download Markdown",
-                        data=markdown_content,
-                        file_name=f"{st.session_state.current_project}_Documentation.md",
-                        mime="text/markdown",
-                    )
+            with open(markdown_path, "rb") as markdown_file:
+                st.download_button(
+                    label="Download Markdown Documentation",
+                    data=markdown_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.md",
+                    mime="text/markdown",
+                )
 
 
 
@@ -671,8 +659,8 @@ def project_view_page():
         st.session_state.page = "generate_documentation"
         st.rerun()
 
-    if st.button("View Documentation"):
-        st.session_state.page = "view_documentation"
+    if st.button("Saved Documentation"):
+        st.session_state.page = "saved_documentation"
         st.rerun()
 
     # Toggle file structure display (if required)