Update app.py

app.py

@@ -505,7 +505,6 @@ def generate_documentation_page():
                 md_file.write(documentation)
             st.success(f"Markdown file generated at: {markdown_file_path}")
 
-
 # Helper function to generate PDF
 def generate_pdf(documentation):
     pdf = FPDF()
@@ -519,7 +518,7 @@ 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:
@@ -527,12 +526,14 @@ def generate_pdf(documentation):
             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="documentation.pdf",
+        file_name=pdf_file_name,
         mime="application/pdf",
     )
     os.unlink(pdf_file.name)
@@ -540,16 +541,32 @@ def generate_pdf(documentation):
 
 # Helper function to generate Markdown file
 def generate_markdown_file(documentation):
+    project_name = st.session_state.current_project
+    markdown_file_name = f"{project_name} Documentation.md"
+
+    # Format documentation with markdown-specific rules
+    formatted_lines = []
+    for line in documentation.splitlines():
+        if line.startswith("- '") and line.endswith("':"):
+            formatted_lines.append(f"**{line}**")  # Bold specific lines
+        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
+        else:
+            formatted_lines.append(line)
+
+    formatted_documentation = "\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(documentation)
+        f.write(formatted_documentation)
 
     # Download the Markdown file
     st.download_button(
         label="Download Markdown File",
         data=open(markdown_file.name, "rb").read(),
-        file_name="documentation.md",
+        file_name=markdown_file_name,
         mime="text/markdown",
     )
     os.unlink(markdown_file.name)