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")
@@ -585,44 +585,51 @@ def generate_markdown_file(documentation, output_path):
 
 
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-def view_documentation_page():
-    st.subheader(f"View Documentation for {st.session_state.current_project}")
-    st.write("This page displays the generated documentation for the selected project.")
-
-    # Check if files are available in session state
-    pdf_path = st.session_state.get("pdf_file_path")
-    markdown_path = st.session_state.get("markdown_file_path")
-
-    # Display PDF file if it exists
-    if pdf_path and os.path.exists(pdf_path):
-        st.write("### PDF Documentation")
-        with open(pdf_path, "rb") as pdf_file:
-            pdf_data = pdf_file.read()
-            st.download_button(
-                label="Download PDF",
-                data=pdf_data,
-                file_name=os.path.basename(pdf_path),
-                mime="application/pdf",
-            )
-        st.write("View PDF Documentation below:")
-        st.pdf_viewer(pdf_path)
-
-    # Display Markdown file if it exists
-    if markdown_path and os.path.exists(markdown_path):
-        st.write("### Markdown Documentation")
-        with open(markdown_path, "r") as md_file:
-            markdown_content = md_file.read()
-            st.download_button(
-                label="Download Markdown",
-                data=markdown_content,
-                file_name=os.path.basename(markdown_path),
-                mime="text/markdown",
-            )
-        st.markdown(f"```\n{markdown_content}\n```", unsafe_allow_html=True)
-
-    if st.button("Back to Project"):
+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"):
         st.session_state.page = "project_view"
         st.rerun()
+    if st.sidebar.button("Log Out"):
+        st.session_state.authenticated = False
+        st.session_state.username = None
+        st.session_state.page = "login"
+        st.rerun()
+
+    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)
+    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")
+
+    # Check if files exist
+    pdf_exists = os.path.exists(pdf_path)
+    markdown_exists = os.path.exists(markdown_path)
+
+    # Display available documentation
+    if not pdf_exists and not markdown_exists:
+        st.info("No documentation has been generated yet. Please go to the 'Generate Documentation' page.")
+    else:
+        if pdf_exists:
+            with open(pdf_path, "rb") as pdf_file:
+                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, "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",
+                )
 
 
 
@@ -647,8 +654,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)