Update app.py

app.py

@@ -470,55 +470,29 @@ def generate_documentation_page():
 
         # Button to generate PDF
         if st.button("Generate PDF"):
-            pdf = FPDF()
-            pdf.set_auto_page_break(auto=True, margin=15)
-            pdf.add_page()
-            pdf.set_font("Courier", size=12)
-
-            # Write documentation to PDF with headers
-            for line in documentation.split("\n"):
-                if line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
-                   line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
-                    pdf.set_font("Courier", style="B", size=14)
-                elif line.startswith("- '"):
-                    pdf.set_font("Courier", style="B", size=12)
-                else:
-                    pdf.set_font("Courier", size=12)
-                pdf.multi_cell(0, 10, line)
-
-            # Save and allow download of the PDF
-            pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-            pdf.output(pdf_file.name)
-            st.download_button(
-                label="Download PDF",
-                data=open(pdf_file.name, "rb").read(),
-                file_name=f"{st.session_state.current_project} Documentation.pdf",
-                mime="application/pdf",
-            )
-            os.unlink(pdf_file.name)
+            pdf_file_path = os.path.join(project_folder, f"{st.session_state.current_project} Documentation.pdf")
+            generate_pdf(documentation, pdf_file_path)
+
+            # Save the path in session state for later viewing
+            st.session_state.pdf_file_path = pdf_file_path
+            st.success("PDF file generated successfully! It will now appear in the 'View Documentation' page.")
 
         # Button to generate Markdown file
         if st.button("Generate Markdown File"):
-            markdown_file_path = os.path.join(user_folder, f"{st.session_state.current_project} Documentation.md")
-            with open(markdown_file_path, "w") as md_file:
-                md_file.write(f"# Documentation for {st.session_state.current_project}\n\n")
-                md_file.write(documentation)
-
-            # Save and allow download of the Markdown file
-            with open(markdown_file_path, "rb") as md_file:
-                st.download_button(
-                    label="Download Markdown File",
-                    data=md_file.read(),
-                    file_name=f"{st.session_state.current_project} Documentation.md",
-                    mime="text/markdown",
-                )
+            markdown_file_path = os.path.join(project_folder, f"{st.session_state.current_project} Documentation.md")
+            generate_markdown_file(documentation, markdown_file_path)
+
+            # Save the path in session state for later viewing
+            st.session_state.markdown_file_path = markdown_file_path
+            st.success("Markdown file generated successfully! It will now appear in the 'View Documentation' page.")
+
 
 # Helper function to generate PDF
-def generate_pdf(documentation):
+def generate_pdf(documentation, output_path):
     pdf = FPDF()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
-    pdf.set_font("Courier", size=12)  # Use Courier for an IDE-like font
+    pdf.set_font("Courier", size=12)
 
     # Add headers and content
     for line in documentation.splitlines():
@@ -526,75 +500,88 @@ def generate_pdf(documentation):
             pdf.set_font("Courier", style="B", size=12)  # Bold for specific lines
             pdf.multi_cell(0, 10, line)
         elif line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
-             line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
+           line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
             pdf.set_font("Courier", style="B", size=14)  # Bold larger headers
             pdf.multi_cell(0, 10, line)
         else:
             pdf.set_font("Courier", size=12)  # Regular for other lines
             pdf.multi_cell(0, 10, line)
 
-    # Save and download the PDF
-    project_name = st.session_state.current_project
-    pdf_file_name = f"{project_name} Documentation.pdf"
-    pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-    pdf.output(pdf_file.name)
-    st.download_button(
-        label="Download PDF",
-        data=open(pdf_file.name, "rb").read(),
-        file_name=pdf_file_name,
-        mime="application/pdf",
-    )
-    os.unlink(pdf_file.name)
+    # Save the PDF
+    pdf.output(output_path)
 
 
-# Helper function to generate Markdown file
-def generate_markdown_file(documentation):
-    # Ensure the user folder is defined
-    user_folder = os.path.join("user_projects", st.session_state.username)
-    os.makedirs(user_folder, exist_ok=True)
 
-    # Generate the Markdown file name
-    project_name = st.session_state.current_project
-    markdown_file_name = f"{project_name} Documentation.md"
+# Helper function to generate Markdown file
+def generate_markdown_file(documentation, output_path):
+    """
+    Generates a markdown file from the documentation with structured formatting.
 
-    # Format documentation with markdown-specific rules
+    Args:
+        documentation (str): The documentation content to format and save.
+        output_path (str): Path to save the generated markdown file.
+    """
     formatted_lines = []
     for line in documentation.splitlines():
+        # Add bold formatting to specific lines (e.g., function names)
         if line.startswith("- '") and line.endswith("':"):
-            formatted_lines.append(f"**{line}**")  # Bold specific lines
+            formatted_lines.append(f"**{line.strip()}**")
+        # Add section headers with markdown syntax
         elif line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
              line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
-            formatted_lines.append(f"# {line}")  # Header for large sections
+            formatted_lines.append(f"# {line.strip()}")
         else:
+            # Regular text as-is
             formatted_lines.append(line)
 
-    formatted_documentation = "\n".join(formatted_lines)
+    # Write the formatted lines to a markdown file
+    with open(output_path, "w") as f:
+        f.write("\n".join(formatted_lines))
 
-    # Save as a temporary Markdown file
-    markdown_file = tempfile.NamedTemporaryFile(delete=False, suffix=".md")
-    with open(markdown_file.name, "w") as f:
-        f.write(formatted_documentation)
-
-    # Provide download button for the Markdown file
-    st.download_button(
-        label="Download Markdown File",
-        data=open(markdown_file.name, "rb").read(),
-        file_name=markdown_file_name,
-        mime="text/markdown",
-    )
-    os.unlink(markdown_file.name)
 
 
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-
-
 def view_documentation_page():
     st.subheader(f"View Documentation for {st.session_state.current_project}")
-    st.write("This page will display the generated documentation for the selected 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"):
         st.session_state.page = "project_view"
         st.rerun()
 
+
+
 def project_view_page():
     # Sidebar with logout and return buttons
     st.sidebar.title(f"Project: {st.session_state.current_project}")