Update app.py

app.py

@@ -591,43 +591,62 @@ 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"):
+    # 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"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.")
+
+    # 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:
+                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",
+                    )
+
+        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",
+                    )